nuno 修订了这个 Gist 1 month ago. 转到此修订
1 file changed, 761 insertions
gistfile1.txt(文件已创建)
| @@ -0,0 +1,761 @@ | |||
| 1 | + | #!/usr/bin/env bash | |
| 2 | + | ||
| 3 | + | # ============================================ | |
| 4 | + | # Lifecycle | |
| 5 | + | # ============================================ | |
| 6 | + | ||
| 7 | + | function cmd::rule::on_load() { | |
| 8 | + | flag::register --name | |
| 9 | + | flag::register --desc | |
| 10 | + | flag::register --group | |
| 11 | + | flag::register --extends | |
| 12 | + | flag::register --remove-extends | |
| 13 | + | flag::register --block-ip | |
| 14 | + | flag::register --allow-ip | |
| 15 | + | flag::register --block-service | |
| 16 | + | flag::register --allow-service | |
| 17 | + | flag::register --block-port | |
| 18 | + | flag::register --allow-port | |
| 19 | + | flag::register --remove-block-ip | |
| 20 | + | flag::register --remove-allow-ip | |
| 21 | + | flag::register --remove-block-port | |
| 22 | + | flag::register --remove-allow-port | |
| 23 | + | flag::register --peer | |
| 24 | + | flag::register --peers | |
| 25 | + | flag::register --dns-redirect | |
| 26 | + | flag::register --color | |
| 27 | + | flag::register --base | |
| 28 | + | flag::register --no-base | |
| 29 | + | flag::register --tree | |
| 30 | + | flag::register --resolved | |
| 31 | + | flag::register --force | |
| 32 | + | flag::register --type | |
| 33 | + | flag::register --all | |
| 34 | + | } | |
| 35 | + | ||
| 36 | + | # ============================================ | |
| 37 | + | # Help | |
| 38 | + | # ============================================ | |
| 39 | + | ||
| 40 | + | function cmd::rule::help() { | |
| 41 | + | cat <<EOF | |
| 42 | + | Usage: wgctl rule <subcommand> [options] | |
| 43 | + | ||
| 44 | + | Manage firewall rules with inheritance support. | |
| 45 | + | Rules can extend base rules to compose reusable access policies. | |
| 46 | + | Service names from 'wgctl net' can be used instead of raw IPs/ports. | |
| 47 | + | ||
| 48 | + | Subcommands: | |
| 49 | + | list, ls List all rules | |
| 50 | + | show, inspect Show rule details and inheritance | |
| 51 | + | add, new, create Create a new rule | |
| 52 | + | update, edit Update a rule and re-apply to peers | |
| 53 | + | remove, rm, del Remove a rule | |
| 54 | + | assign Assign a rule to a peer | |
| 55 | + | unassign Remove rule from a peer | |
| 56 | + | reapply Re-apply rule to all assigned peers | |
| 57 | + | migrate Apply default rules to unassigned peers | |
| 58 | + | ||
| 59 | + | Options for list: | |
| 60 | + | --base Show only base rules | |
| 61 | + | --no-base Hide base rules section | |
| 62 | + | --group <name> Filter by group (case insensitive) | |
| 63 | + | --tree Show full inheritance tree inline | |
| 64 | + | ||
| 65 | + | Options for add: | |
| 66 | + | --name <name> Rule name | |
| 67 | + | --desc <description> Description | |
| 68 | + | --group <group> Display group (e.g. VM Rules, Users) | |
| 69 | + | --extends <rule,...> Inherit from base rules (comma-separated) | |
| 70 | + | --base Create as base rule (not directly assignable) | |
| 71 | + | --allow-ip <ip/cidr> Allow IP or subnet (repeatable) | |
| 72 | + | --allow-port <ip:port:proto> Allow specific port (repeatable) | |
| 73 | + | --block-ip <ip/cidr> Block IP or subnet (repeatable) | |
| 74 | + | --block-port <ip:port:proto> Block specific port (repeatable) | |
| 75 | + | --block-service <name> Block named service — resolved to IP/port at creation (repeatable) | |
| 76 | + | --allow-service <name> Allow named service — resolved to IP/port at creation (repeatable) | |
| 77 | + | --dns-redirect Force DNS through Pi-hole | |
| 78 | + | ||
| 79 | + | Options for update: | |
| 80 | + | (same as add, plus:) | |
| 81 | + | --add-extends <rule,...> Add inherited base rules | |
| 82 | + | --remove-extends <rule,...> Remove inherited base rules | |
| 83 | + | --remove-allow-ip <ip> Remove allow IP entry | |
| 84 | + | --remove-allow-port <entry> Remove allow port entry | |
| 85 | + | --remove-block-ip <ip> Remove block IP entry | |
| 86 | + | --remove-block-port <entry> Remove block port entry | |
| 87 | + | ||
| 88 | + | Options for show/inspect: | |
| 89 | + | --name <name> Rule name | |
| 90 | + | --resolved Show resolved/merged entries | |
| 91 | + | --no-peers Hide assigned peers | |
| 92 | + | ||
| 93 | + | Options for reapply: | |
| 94 | + | --name <name> Rule name | |
| 95 | + | --all Reapply all rules | |
| 96 | + | ||
| 97 | + | Examples: | |
| 98 | + | wgctl rule list | |
| 99 | + | wgctl rule list --tree | |
| 100 | + | wgctl rule list --group "VM Rules" | |
| 101 | + | wgctl rule show --name guest | |
| 102 | + | wgctl rule show --name moonlight-02 --resolved | |
| 103 | + | wgctl rule add --name no-proxmox --base --block-service proxmox | |
| 104 | + | wgctl rule add --name dev-01 --desc "Dev access" --group "Dev" --extends no-lan | |
| 105 | + | wgctl rule add --name restricted-dns --allow-service pihole:dns --block-service pihole | |
| 106 | + | wgctl rule update --name user --add-extends no-nginx | |
| 107 | + | wgctl rule assign --name dev-01 --peer laptop-nuno | |
| 108 | + | wgctl rule reapply --all | |
| 109 | + | EOF | |
| 110 | + | } | |
| 111 | + | ||
| 112 | + | # ============================================ | |
| 113 | + | # Run | |
| 114 | + | # ============================================ | |
| 115 | + | ||
| 116 | + | function cmd::rule::run() { | |
| 117 | + | local subcmd="${1:-help}" | |
| 118 | + | shift || true | |
| 119 | + | ||
| 120 | + | case "$subcmd" in | |
| 121 | + | list|ls) cmd::rule::list "$@" ;; | |
| 122 | + | show|inspect) cmd::rule::show "$@" ;; | |
| 123 | + | inspect) cmd::rule::inspect "$@" ;; | |
| 124 | + | add|new|create) cmd::rule::add "$@" ;; | |
| 125 | + | update|edit) cmd::rule::update "$@" ;; | |
| 126 | + | remove|rm|del|delete) cmd::rule::remove "$@" ;; | |
| 127 | + | assign) cmd::rule::assign "$@" ;; | |
| 128 | + | unassign) cmd::rule::unassign "$@" ;; | |
| 129 | + | migrate) cmd::rule::migrate "$@" ;; | |
| 130 | + | reapply) cmd::rule::reapply "$@" ;; | |
| 131 | + | help) cmd::rule::help ;; | |
| 132 | + | *) | |
| 133 | + | log::error "Unknown subcommand: '${subcmd}'" | |
| 134 | + | cmd::rule::help | |
| 135 | + | return 1 | |
| 136 | + | ;; | |
| 137 | + | esac | |
| 138 | + | } | |
| 139 | + | ||
| 140 | + | # ============================================ | |
| 141 | + | # List | |
| 142 | + | # ============================================ | |
| 143 | + | ||
| 144 | + | function cmd::rule::_pad() { | |
| 145 | + | local text="$1" width="$2" | |
| 146 | + | local visible | |
| 147 | + | visible=$(printf "%s" "$text" | sed 's/\x1b\[[0-9;]*m//g') | |
| 148 | + | local visible_len=${#visible} | |
| 149 | + | local byte_len=${#text} | |
| 150 | + | local extra=$(( byte_len - visible_len )) | |
| 151 | + | printf "%-$(( width + extra ))s" "$text" | |
| 152 | + | } | |
| 153 | + | ||
| 154 | + | function cmd::rule::_print_extends_tree() { | |
| 155 | + | local extends="$1" indent="${2:-2}" rules_dir="$3" | |
| 156 | + | [[ -z "$extends" ]] && return 0 | |
| 157 | + | ||
| 158 | + | local extend_list=() | |
| 159 | + | IFS=',' read -ra extend_list <<< "$extends" | |
| 160 | + | ||
| 161 | + | for base in "${extend_list[@]}"; do | |
| 162 | + | [[ -z "$base" ]] && continue | |
| 163 | + | local spaces | |
| 164 | + | spaces=$(printf '%*s' "$indent" '') | |
| 165 | + | printf " \033[0;37m%s↳ %s\033[0m\n" "$spaces" "$base" | |
| 166 | + | ||
| 167 | + | if [[ "$indent" -lt 12 ]]; then | |
| 168 | + | local sub_file="" | |
| 169 | + | if sub_file=$(json::find_rule_file "$rules_dir" "$base" 2>/dev/null); then | |
| 170 | + | local sub_extends="" | |
| 171 | + | sub_extends=$(json::get "$sub_file" "extends" 2>/dev/null \ | |
| 172 | + | | tr '\n' ',' | sed 's/,$//' || true) | |
| 173 | + | if [[ -n "$sub_extends" ]]; then | |
| 174 | + | cmd::rule::_print_extends_tree \ | |
| 175 | + | "$sub_extends" $(( indent + 4 )) "$rules_dir" | |
| 176 | + | fi | |
| 177 | + | fi | |
| 178 | + | fi | |
| 179 | + | done | |
| 180 | + | } | |
| 181 | + | ||
| 182 | + | function cmd::rule::list() { | |
| 183 | + | local rules_dir | |
| 184 | + | rules_dir="$(ctx::rules)" | |
| 185 | + | ||
| 186 | + | local show_base_only=false | |
| 187 | + | local show_base=true | |
| 188 | + | local filter_group="" | |
| 189 | + | local show_tree=false | |
| 190 | + | local found_any=false | |
| 191 | + | ||
| 192 | + | while [[ $# -gt 0 ]]; do | |
| 193 | + | case "$1" in | |
| 194 | + | --base) show_base_only=true; shift ;; | |
| 195 | + | --no-base) show_base=false; shift ;; | |
| 196 | + | --group) util::require_flag "--group" "${2:-}" || return 1 | |
| 197 | + | filter_group="${2,,}"; shift 2 ;; | |
| 198 | + | --tree) show_tree=true; shift ;; | |
| 199 | + | --help) cmd::rule::help; return ;; | |
| 200 | + | *) | |
| 201 | + | log::error "Unknown flag: $1" | |
| 202 | + | return 1 ;; | |
| 203 | + | esac | |
| 204 | + | done | |
| 205 | + | ||
| 206 | + | local rules=("${rules_dir}"/*.rule) | |
| 207 | + | if [[ ! -f "${rules[0]}" ]]; then | |
| 208 | + | log::wg "No rules configured" | |
| 209 | + | return 0 | |
| 210 | + | fi | |
| 211 | + | ||
| 212 | + | local header_printed=false | |
| 213 | + | local printing_base=false | |
| 214 | + | local current_group="" | |
| 215 | + | ||
| 216 | + | while IFS="|" read -r name desc n_allows n_blocks \ | |
| 217 | + | peer_count extends is_base group; do | |
| 218 | + | [[ -z "$name" ]] && continue | |
| 219 | + | ||
| 220 | + | # --base: show ONLY base rules | |
| 221 | + | if $show_base_only && [[ "$is_base" == "False" ]]; then | |
| 222 | + | continue | |
| 223 | + | fi | |
| 224 | + | ||
| 225 | + | # --no-base: hide base rules | |
| 226 | + | if ! $show_base && [[ "$is_base" == "True" ]]; then | |
| 227 | + | continue | |
| 228 | + | fi | |
| 229 | + | ||
| 230 | + | # --group filter (case insensitive) | |
| 231 | + | if [[ -n "$filter_group" && "${group,,}" != "$filter_group" ]]; then | |
| 232 | + | continue | |
| 233 | + | fi | |
| 234 | + | ||
| 235 | + | # Print header on first match | |
| 236 | + | if ! $header_printed; then | |
| 237 | + | log::section "Firewall Rules" | |
| 238 | + | printf "\n %-20s %-40s %-8s %-8s %s\n" \ | |
| 239 | + | "NAME" "DESCRIPTION" \ | |
| 240 | + | "$(ui::center "ALLOWS" 8)" \ | |
| 241 | + | "$(ui::center "BLOCKS" 8)" \ | |
| 242 | + | "PEERS" | |
| 243 | + | local divider | |
| 244 | + | divider=$(printf '─%.0s' {1..88}) | |
| 245 | + | printf " %s\n" "$divider" | |
| 246 | + | header_printed=true | |
| 247 | + | fi | |
| 248 | + | ||
| 249 | + | # Base rules section header | |
| 250 | + | if [[ "$is_base" == "True" && "$printing_base" == "false" ]]; then | |
| 251 | + | if ! $show_base_only; then | |
| 252 | + | local bdashes | |
| 253 | + | bdashes=$(printf '─%.0s' {1..74}) | |
| 254 | + | printf "\n \033[0;37m── Base Rules \033[0m%s\n" "$bdashes" | |
| 255 | + | fi | |
| 256 | + | printing_base=true | |
| 257 | + | current_group="" | |
| 258 | + | fi | |
| 259 | + | ||
| 260 | + | # Group header — only for non-base rules | |
| 261 | + | if [[ "$is_base" == "False" && "${group,,}" != "${current_group,,}" ]]; then | |
| 262 | + | if [[ -n "$group" ]]; then | |
| 263 | + | printf "\n \033[0;36m▸ %s\033[0m\n" "$group" | |
| 264 | + | elif [[ -n "$current_group" ]]; then | |
| 265 | + | printf "\n" | |
| 266 | + | fi | |
| 267 | + | current_group="$group" | |
| 268 | + | fi | |
| 269 | + | ||
| 270 | + | local short_desc="${desc:0:35}" | |
| 271 | + | [[ ${#desc} -gt 35 ]] && short_desc="${short_desc}..." | |
| 272 | + | ||
| 273 | + | local desc_col_width=40 | |
| 274 | + | [[ "${short_desc:-—}" == "—" ]] && desc_col_width=42 | |
| 275 | + | ||
| 276 | + | found_any=true | |
| 277 | + | ||
| 278 | + | printf " %-20s %-${desc_col_width}s %-8s %-8s %s\n" \ | |
| 279 | + | "$name" "${short_desc:-—}" \ | |
| 280 | + | "$(ui::center "$n_allows" 8)" \ | |
| 281 | + | "$(ui::center "$n_blocks" 8)" \ | |
| 282 | + | "${peer_count} peers" | |
| 283 | + | ||
| 284 | + | # Print extends | |
| 285 | + | if [[ -n "$extends" ]]; then | |
| 286 | + | if $show_tree; then | |
| 287 | + | cmd::rule::_print_extends_tree "$extends" 2 "$rules_dir" | |
| 288 | + | else | |
| 289 | + | local extend_list=() | |
| 290 | + | IFS=',' read -ra extend_list <<< "$extends" | |
| 291 | + | for base in "${extend_list[@]}"; do | |
| 292 | + | [[ -z "$base" ]] && continue | |
| 293 | + | printf " \033[0;37m ↳ %s\033[0m\n" "$base" | |
| 294 | + | done | |
| 295 | + | fi | |
| 296 | + | printf "\n" | |
| 297 | + | fi | |
| 298 | + | ||
| 299 | + | done < <(json::rule_list_data "$rules_dir" "$(ctx::meta)") | |
| 300 | + | ||
| 301 | + | if ! $found_any; then | |
| 302 | + | if [[ -n "$filter_group" ]]; then | |
| 303 | + | log::wg_warning "No rules found in group: ${filter_group}" | |
| 304 | + | else | |
| 305 | + | log::wg_warning "No rules found" | |
| 306 | + | fi | |
| 307 | + | fi | |
| 308 | + | ||
| 309 | + | printf "\n" | |
| 310 | + | } | |
| 311 | + | ||
| 312 | + | # ============================================ | |
| 313 | + | # Show | |
| 314 | + | # ============================================ | |
| 315 | + | ||
| 316 | + | function cmd::rule::show() { | |
| 317 | + | local name="" show_peers=true show_resolved=false | |
| 318 | + | ||
| 319 | + | while [[ $# -gt 0 ]]; do | |
| 320 | + | case "$1" in | |
| 321 | + | --name) util::require_flag "--name" "${2:-}" || return 1 | |
| 322 | + | name="$2"; shift 2 ;; | |
| 323 | + | --no-peers) show_peers=false; shift ;; | |
| 324 | + | --resolved) show_resolved=true; shift ;; | |
| 325 | + | --help) cmd::rule::help; return ;; | |
| 326 | + | *) log::error "Unknown flag: $1"; return 1 ;; | |
| 327 | + | esac | |
| 328 | + | done | |
| 329 | + | ||
| 330 | + | [[ -z "$name" ]] && log::error "Missing required flag: --name" && return 1 | |
| 331 | + | rule::require_exists "$name" || return 1 | |
| 332 | + | ||
| 333 | + | local rule_file | |
| 334 | + | rule_file="$(rule::path "$name")" | |
| 335 | + | ||
| 336 | + | # ── DNS display ─────────────────────────────── | |
| 337 | + | local dns_redirect resolved_dns dns_display | |
| 338 | + | dns_redirect=$(rule::get_own "$name" "dns_redirect") | |
| 339 | + | dns_redirect="${dns_redirect:-false}" | |
| 340 | + | resolved_dns=$(rule::get "$name" "dns_redirect") | |
| 341 | + | resolved_dns="${resolved_dns:-false}" | |
| 342 | + | ||
| 343 | + | if [[ "${resolved_dns,,}" == "true" && "${dns_redirect,,}" != "true" ]]; then | |
| 344 | + | dns_display="true (inherited)" | |
| 345 | + | elif [[ "${dns_redirect,,}" == "true" ]]; then | |
| 346 | + | dns_display="true" | |
| 347 | + | else | |
| 348 | + | dns_display="false" | |
| 349 | + | fi | |
| 350 | + | ||
| 351 | + | log::section "Rule: ${name}" | |
| 352 | + | printf "\n" | |
| 353 | + | ||
| 354 | + | local desc group | |
| 355 | + | desc=$(json::get "$rule_file" "desc") | |
| 356 | + | group=$(json::get "$rule_file" "group") | |
| 357 | + | ui::row "Description" "${desc:-—}" | |
| 358 | + | ui::row "Group" "${group:-—}" | |
| 359 | + | ui::row "DNS" "$dns_display" | |
| 360 | + | ||
| 361 | + | # ── Extends + own rules ──────────────────────── | |
| 362 | + | if rule::render_extends_tree "$name"; then | |
| 363 | + | # Has inheritance — tree already rendered | |
| 364 | + | printf "\n" | |
| 365 | + | else | |
| 366 | + | # No inheritance — flat view | |
| 367 | + | rule::render_flat "$name" | |
| 368 | + | fi | |
| 369 | + | ||
| 370 | + | # ── Resolved ────────────────────────────────── | |
| 371 | + | if $show_resolved; then | |
| 372 | + | cmd::rule::_show_section "Resolved (applied to peers)" | |
| 373 | + | printf "\n" | |
| 374 | + | local res_allow_ports res_allow_ips res_block_ips res_block_ports | |
| 375 | + | res_allow_ports=$(rule::get "$name" "allow_ports") | |
| 376 | + | res_allow_ips=$(rule::get "$name" "allow_ips") | |
| 377 | + | res_block_ips=$(rule::get "$name" "block_ips") | |
| 378 | + | res_block_ports=$(rule::get "$name" "block_ports") | |
| 379 | + | while IFS= read -r e; do [[ -n "$e" ]] && net::print_entry "+" "$e"; done \ | |
| 380 | + | <<< "$res_allow_ports"$'\n'"$res_allow_ips" | |
| 381 | + | while IFS= read -r e; do [[ -n "$e" ]] && net::print_entry "-" "$e"; done \ | |
| 382 | + | <<< "$res_block_ips"$'\n'"$res_block_ports" | |
| 383 | + | printf "\n" | |
| 384 | + | ||
| 385 | + | fi | |
| 386 | + | ||
| 387 | + | # ── Peers ───────────────────────────────────── | |
| 388 | + | local peer_list=() | |
| 389 | + | mapfile -t peer_list < <(peers::with_rule "$name") | |
| 390 | + | ||
| 391 | + | local peer_count=${#peer_list[@]} | |
| 392 | + | ||
| 393 | + | printf "\033[0;37m── Peers (%s) \033[0m%s\n\n" \ | |
| 394 | + | "$(color::gray "${peer_count}")" \ | |
| 395 | + | "$(printf '\033[0;37m─%.0s' {1..35})" | |
| 396 | + | ||
| 397 | + | if $show_peers && [[ $peer_count -gt 0 ]]; then | |
| 398 | + | for peer_name in "${peer_list[@]}"; do | |
| 399 | + | local ip | |
| 400 | + | ip=$(peers::get_ip "$peer_name") | |
| 401 | + | printf " %-28s %s\n" "$peer_name" "$ip" | |
| 402 | + | done | |
| 403 | + | fi | |
| 404 | + | printf "\n" | |
| 405 | + | return 0 | |
| 406 | + | } | |
| 407 | + | ||
| 408 | + | # ============================================ | |
| 409 | + | # Add | |
| 410 | + | # ============================================ | |
| 411 | + | ||
| 412 | + | function cmd::rule::add() { | |
| 413 | + | local name="" desc="" group="" | |
| 414 | + | local extends=() | |
| 415 | + | local allow_ips=() block_ips=() block_ports=() allow_ports=() | |
| 416 | + | local block_services=() allow_services=() | |
| 417 | + | local dns_redirect=false | |
| 418 | + | local is_base=false | |
| 419 | + | ||
| 420 | + | while [[ $# -gt 0 ]]; do | |
| 421 | + | case "$1" in | |
| 422 | + | --name) name="$2"; shift 2 ;; | |
| 423 | + | --desc) desc="$2"; shift 2 ;; | |
| 424 | + | --group) group="$2"; shift 2 ;; | |
| 425 | + | --extends) | |
| 426 | + | IFS=',' read -ra ext <<< "$2" | |
| 427 | + | extends+=("${ext[@]}") | |
| 428 | + | shift 2 ;; | |
| 429 | + | --base) is_base=true; shift ;; | |
| 430 | + | --allow-ip) allow_ips+=("$2"); shift 2 ;; | |
| 431 | + | --allow-port) allow_ports+=("$2"); shift 2 ;; | |
| 432 | + | --block-ip) block_ips+=("$2"); shift 2 ;; | |
| 433 | + | --block-port) block_ports+=("$2"); shift 2 ;; | |
| 434 | + | --block-service) block_services+=("$2"); shift 2 ;; | |
| 435 | + | --allow-service) allow_services+=("$2"); shift 2 ;; | |
| 436 | + | --dns-redirect) dns_redirect=true; shift ;; | |
| 437 | + | --help) cmd::rule::help; return ;; | |
| 438 | + | *) | |
| 439 | + | log::error "Unknown flag: $1" | |
| 440 | + | return 1 ;; | |
| 441 | + | esac | |
| 442 | + | done | |
| 443 | + | ||
| 444 | + | [[ -z "$name" ]] && log::error "Missing required flag: --name" && return 1 | |
| 445 | + | ||
| 446 | + | if rule::exists "$name"; then | |
| 447 | + | log::error "Rule already exists: ${name}" | |
| 448 | + | return 1 | |
| 449 | + | fi | |
| 450 | + | ||
| 451 | + | # Validate extends | |
| 452 | + | for ext in "${extends[@]}"; do | |
| 453 | + | rule::require_exists "$ext" || return 1 | |
| 454 | + | done | |
| 455 | + | ||
| 456 | + | # Determine target directory | |
| 457 | + | local rule_dir | |
| 458 | + | if $is_base; then | |
| 459 | + | rule_dir="$(ctx::rules)/base" | |
| 460 | + | mkdir -p "$rule_dir" | |
| 461 | + | else | |
| 462 | + | rule_dir="$(ctx::rules)" | |
| 463 | + | fi | |
| 464 | + | ||
| 465 | + | for svc in "${block_services[@]}"; do | |
| 466 | + | while IFS= read -r resolved; do | |
| 467 | + | if [[ "$resolved" == *:*:* ]]; then | |
| 468 | + | block_ports+=("${resolved}") | |
| 469 | + | else | |
| 470 | + | block_ips+=("${resolved}/32") | |
| 471 | + | fi | |
| 472 | + | done < <(net::resolve "$svc") | |
| 473 | + | done | |
| 474 | + | ||
| 475 | + | for svc in "${allow_services[@]}"; do | |
| 476 | + | while IFS= read -r resolved; do | |
| 477 | + | if [[ "$resolved" == *:*:* ]]; then | |
| 478 | + | allow_ports+=("${resolved}") | |
| 479 | + | else | |
| 480 | + | allow_ips+=("${resolved}/32") | |
| 481 | + | fi | |
| 482 | + | done < <(net::resolve "$svc") | |
| 483 | + | done | |
| 484 | + | ||
| 485 | + | local rule_file="${rule_dir}/${name}.rule" | |
| 486 | + | ||
| 487 | + | local allow_str block_str port_str allow_port_str extends_str | |
| 488 | + | allow_str=$(IFS=','; echo "${allow_ips[*]}") | |
| 489 | + | block_str=$(IFS=','; echo "${block_ips[*]}") | |
| 490 | + | port_str=$(IFS=','; echo "${block_ports[*]}") | |
| 491 | + | allow_port_str=$(IFS=','; echo "${allow_ports[*]}") | |
| 492 | + | extends_str=$(IFS=','; echo "${extends[*]}") | |
| 493 | + | ||
| 494 | + | json::create_rule "$rule_file" "$name" "$desc" \ | |
| 495 | + | "$($dns_redirect && echo true || echo false)" \ | |
| 496 | + | "$allow_str" "$block_str" "$port_str" \ | |
| 497 | + | "$allow_port_str" "$extends_str" "$group" || return 1 | |
| 498 | + | ||
| 499 | + | local base_label="" | |
| 500 | + | $is_base && base_label=" (base)" | |
| 501 | + | ||
| 502 | + | log::wg_success "Rule created: ${name}${base_label}" | |
| 503 | + | } | |
| 504 | + | ||
| 505 | + | # ============================================ | |
| 506 | + | # Update | |
| 507 | + | # ============================================ | |
| 508 | + | ||
| 509 | + | function cmd::rule::update() { | |
| 510 | + | local name="" desc="" group="" | |
| 511 | + | local add_extends=() rm_extends=() | |
| 512 | + | local allow_ips=() block_ips=() block_ports=() allow_ports=() | |
| 513 | + | local rm_allow_ips=() rm_block_ips=() rm_block_ports=() rm_allow_ports=() | |
| 514 | + | local dns_redirect="" | |
| 515 | + | ||
| 516 | + | while [[ $# -gt 0 ]]; do | |
| 517 | + | case "$1" in | |
| 518 | + | --name) name="$2"; shift 2 ;; | |
| 519 | + | --desc) desc="$2"; shift 2 ;; | |
| 520 | + | --group) group="$2"; shift 2 ;; | |
| 521 | + | --add-extends) | |
| 522 | + | IFS=',' read -ra ext <<< "$2" | |
| 523 | + | add_extends+=("${ext[@]}") | |
| 524 | + | shift 2 ;; | |
| 525 | + | --remove-extends) | |
| 526 | + | IFS=',' read -ra ext <<< "$2" | |
| 527 | + | rm_extends+=("${ext[@]}") | |
| 528 | + | shift 2 ;; | |
| 529 | + | --allow-ip) allow_ips+=("$2"); shift 2 ;; | |
| 530 | + | --allow-port) allow_ports+=("$2"); shift 2 ;; | |
| 531 | + | --block-ip) block_ips+=("$2"); shift 2 ;; | |
| 532 | + | --block-port) block_ports+=("$2"); shift 2 ;; | |
| 533 | + | --remove-allow-ip) rm_allow_ips+=("$2"); shift 2 ;; | |
| 534 | + | --remove-block-ip) rm_block_ips+=("$2"); shift 2 ;; | |
| 535 | + | --remove-block-port) rm_block_ports+=("$2"); shift 2 ;; | |
| 536 | + | --remove-allow-port) rm_allow_ports+=("$2"); shift 2 ;; | |
| 537 | + | --dns-redirect) dns_redirect=true; shift ;; | |
| 538 | + | --help) cmd::rule::help; return ;; | |
| 539 | + | *) | |
| 540 | + | log::error "Unknown flag: $1" | |
| 541 | + | return 1 ;; | |
| 542 | + | esac | |
| 543 | + | done | |
| 544 | + | ||
| 545 | + | [[ -z "$name" ]] && log::error "Missing required flag: --name" && return 1 | |
| 546 | + | rule::require_exists "$name" || return 1 | |
| 547 | + | ||
| 548 | + | local rule_file | |
| 549 | + | rule_file="$(rule::path "$name")" | |
| 550 | + | ||
| 551 | + | # Update simple fields | |
| 552 | + | [[ -n "$desc" ]] && json::set "$rule_file" "desc" "\"$desc\"" | |
| 553 | + | [[ -n "$group" ]] && json::set "$rule_file" "group" "\"$group\"" | |
| 554 | + | [[ -n "$dns_redirect" ]] && json::set "$rule_file" "dns_redirect" "true" | |
| 555 | + | ||
| 556 | + | # Add entries | |
| 557 | + | for ip in "${allow_ips[@]}"; do json::append "$rule_file" "allow_ips" "$ip"; done | |
| 558 | + | for ip in "${block_ips[@]}"; do json::append "$rule_file" "block_ips" "$ip"; done | |
| 559 | + | for p in "${block_ports[@]}"; do json::append "$rule_file" "block_ports" "$p"; done | |
| 560 | + | for p in "${allow_ports[@]}"; do json::append "$rule_file" "allow_ports" "$p"; done | |
| 561 | + | ||
| 562 | + | # Add/remove extends | |
| 563 | + | for ext in "${add_extends[@]}"; do | |
| 564 | + | rule::require_exists "$ext" || return 1 | |
| 565 | + | json::append "$rule_file" "extends" "$ext" | |
| 566 | + | done | |
| 567 | + | for ext in "${rm_extends[@]}"; do | |
| 568 | + | json::remove "$rule_file" "extends" "$ext" | |
| 569 | + | done | |
| 570 | + | ||
| 571 | + | # Remove entries | |
| 572 | + | for ip in "${rm_allow_ips[@]}"; do json::remove "$rule_file" "allow_ips" "$ip"; done | |
| 573 | + | for ip in "${rm_block_ips[@]}"; do json::remove "$rule_file" "block_ips" "$ip"; done | |
| 574 | + | for p in "${rm_block_ports[@]}"; do json::remove "$rule_file" "block_ports" "$p"; done | |
| 575 | + | for p in "${rm_allow_ports[@]}"; do json::remove "$rule_file" "allow_ports" "$p"; done | |
| 576 | + | ||
| 577 | + | log::wg_success "Rule updated: ${name}" | |
| 578 | + | rule::reapply_all "$name" | |
| 579 | + | } | |
| 580 | + | ||
| 581 | + | # ============================================ | |
| 582 | + | # Remove / Assign / Unassign / Migrate / Reapply | |
| 583 | + | # (unchanged from original) | |
| 584 | + | # ============================================ | |
| 585 | + | ||
| 586 | + | function cmd::rule::remove() { | |
| 587 | + | local name="" force=false | |
| 588 | + | ||
| 589 | + | while [[ $# -gt 0 ]]; do | |
| 590 | + | case "$1" in | |
| 591 | + | --name) name="$2"; shift 2 ;; | |
| 592 | + | --force) force=true; shift ;; | |
| 593 | + | --help) cmd::rule::help; return ;; | |
| 594 | + | *) log::error "Unknown flag: $1"; return 1 ;; | |
| 595 | + | esac | |
| 596 | + | done | |
| 597 | + | ||
| 598 | + | [[ -z "$name" ]] && log::error "Missing required flag: --name" && return 1 | |
| 599 | + | rule::require_exists "$name" || return 1 | |
| 600 | + | ||
| 601 | + | local peer_list=() | |
| 602 | + | mapfile -t peer_list < <(peers::with_rule "$name") | |
| 603 | + | local peer_count=${#peer_list[@]} | |
| 604 | + | ||
| 605 | + | if [[ "$peer_count" -gt 0 ]]; then | |
| 606 | + | log::error "Rule '${name}' is assigned to ${peer_count} peer(s) — unassign first or use --force" | |
| 607 | + | $force || return 1 | |
| 608 | + | for peer in "${peer_list[@]}"; do | |
| 609 | + | local ip | |
| 610 | + | ip=$(peers::get_ip "$peer") | |
| 611 | + | rule::unapply "$name" "$ip" | |
| 612 | + | done | |
| 613 | + | fi | |
| 614 | + | ||
| 615 | + | rm -f "$(rule::path "$name")" | |
| 616 | + | log::wg_success "Rule removed: ${name}" | |
| 617 | + | } | |
| 618 | + | ||
| 619 | + | function cmd::rule::assign() { | |
| 620 | + | local name="" peer="" type="" | |
| 621 | + | while [[ $# -gt 0 ]]; do | |
| 622 | + | case "$1" in | |
| 623 | + | --name) name="$2"; shift 2 ;; | |
| 624 | + | --peer) peer="$2"; shift 2 ;; | |
| 625 | + | --type) type="$2"; shift 2 ;; | |
| 626 | + | --help) cmd::rule::help; return ;; | |
| 627 | + | *) log::error "Unknown flag: $1"; return 1 ;; | |
| 628 | + | esac | |
| 629 | + | done | |
| 630 | + | ||
| 631 | + | [[ -z "$name" || -z "$peer" ]] && \ | |
| 632 | + | log::error "Missing required flags: --name and --peer" && return 1 | |
| 633 | + | ||
| 634 | + | rule::require_exists "$name" || return 1 | |
| 635 | + | rule::require_assignable "$name" || return 1 | |
| 636 | + | ||
| 637 | + | peer=$(peers::resolve_and_require "$peer" "$type") || return 1 | |
| 638 | + | ||
| 639 | + | local existing_rule | |
| 640 | + | existing_rule=$(peers::get_meta "$peer" "rule") | |
| 641 | + | ||
| 642 | + | local ip | |
| 643 | + | ip=$(peers::get_ip "$peer") | |
| 644 | + | [[ -z "$ip" ]] && log::error "Could not resolve IP for: $peer" && return 1 | |
| 645 | + | ||
| 646 | + | if [[ -n "$existing_rule" && "$existing_rule" != "$name" ]]; then | |
| 647 | + | rule::unapply "$existing_rule" "$ip" | |
| 648 | + | log::wg "Removed existing rule '${existing_rule}' from: ${peer}" | |
| 649 | + | fi | |
| 650 | + | ||
| 651 | + | rule::apply "$name" "$ip" | |
| 652 | + | log::wg_success "Assigned rule '${name}' to: ${peer}" | |
| 653 | + | } | |
| 654 | + | ||
| 655 | + | function cmd::rule::unassign() { | |
| 656 | + | local peer="" type="" | |
| 657 | + | while [[ $# -gt 0 ]]; do | |
| 658 | + | case "$1" in | |
| 659 | + | --peer) peer="$2"; shift 2 ;; | |
| 660 | + | --type) type="$2"; shift 2 ;; | |
| 661 | + | --help) cmd::rule::help; return ;; | |
| 662 | + | *) log::error "Unknown flag: $1"; return 1 ;; | |
| 663 | + | esac | |
| 664 | + | done | |
| 665 | + | ||
| 666 | + | [[ -z "$peer" ]] && log::error "Missing required flag: --peer" && return 1 | |
| 667 | + | peer=$(peers::resolve_and_require "$peer" "$type") || return 1 | |
| 668 | + | ||
| 669 | + | local existing_rule | |
| 670 | + | existing_rule=$(peers::get_meta "$peer" "rule") | |
| 671 | + | ||
| 672 | + | if [[ -z "$existing_rule" ]]; then | |
| 673 | + | log::wg_warning "Peer '${peer}' has no assigned rule" | |
| 674 | + | return 0 | |
| 675 | + | fi | |
| 676 | + | ||
| 677 | + | local ip | |
| 678 | + | ip=$(peers::get_ip "$peer") | |
| 679 | + | rule::unapply "$existing_rule" "$ip" | |
| 680 | + | log::wg_success "Unassigned rule from: ${peer}" | |
| 681 | + | } | |
| 682 | + | ||
| 683 | + | function cmd::rule::migrate() { | |
| 684 | + | log::section "Migrating peers to default rules" | |
| 685 | + | local tmp | |
| 686 | + | tmp=$(mktemp) | |
| 687 | + | ||
| 688 | + | while IFS= read -r peer_name; do | |
| 689 | + | local existing | |
| 690 | + | existing=$(peers::get_meta "$peer_name" "rule") | |
| 691 | + | [[ -n "$existing" ]] && continue | |
| 692 | + | local default_rule | |
| 693 | + | default_rule=$(peers::default_rule "$peer_name") | |
| 694 | + | rule::exists "$default_rule" || continue | |
| 695 | + | local ip | |
| 696 | + | ip=$(peers::get_ip "$peer_name") | |
| 697 | + | echo "${peer_name} ${default_rule} ${ip}" >> "$tmp" | |
| 698 | + | done < <(peers::all) | |
| 699 | + | ||
| 700 | + | local count=0 | |
| 701 | + | local lines=() | |
| 702 | + | mapfile -t lines < "$tmp" | |
| 703 | + | rm -f "$tmp" | |
| 704 | + | ||
| 705 | + | for line in "${lines[@]}"; do | |
| 706 | + | IFS=" " read -r peer_name default_rule ip <<< "$line" | |
| 707 | + | rule::apply "$default_rule" "$ip" "$peer_name" </dev/null | |
| 708 | + | (( count++ )) || true | |
| 709 | + | done | |
| 710 | + | ||
| 711 | + | log::wg_success "Migrated ${count} peers" | |
| 712 | + | } | |
| 713 | + | ||
| 714 | + | function cmd::rule::reapply() { | |
| 715 | + | local name="" all=false | |
| 716 | + | while [[ $# -gt 0 ]]; do | |
| 717 | + | case "$1" in | |
| 718 | + | --name) name="$2"; shift 2 ;; | |
| 719 | + | --all) all=true; shift ;; | |
| 720 | + | *) log::error "Unknown flag: $1"; return 1 ;; | |
| 721 | + | esac | |
| 722 | + | done | |
| 723 | + | ||
| 724 | + | if $all; then | |
| 725 | + | log::section "Reapplying all rules" | |
| 726 | + | local count=0 | |
| 727 | + | while IFS= read -r rule_file; do | |
| 728 | + | local rname | |
| 729 | + | rname=$(basename "$rule_file" .rule) | |
| 730 | + | # Skip if no peers assigned | |
| 731 | + | local peer_list=() | |
| 732 | + | mapfile -t peer_list < <(peers::with_rule "$rname") | |
| 733 | + | [[ ${#peer_list[@]} -eq 0 ]] && continue | |
| 734 | + | rule::reapply_all "$rname" | |
| 735 | + | (( count++ )) || true | |
| 736 | + | done < <(find "$(ctx::rules)" -maxdepth 1 -name "*.rule") | |
| 737 | + | log::wg_success "Reapplied ${count} assignable rules" | |
| 738 | + | return 0 | |
| 739 | + | fi | |
| 740 | + | ||
| 741 | + | [[ -z "$name" ]] && log::error "Missing --name or --all" && return 1 | |
| 742 | + | rule::require_exists "$name" || return 1 | |
| 743 | + | rule::reapply_all "$name" | |
| 744 | + | log::wg_success "Rule '${name}' reapplied" | |
| 745 | + | } | |
| 746 | + | ||
| 747 | + | # ============================================ | |
| 748 | + | # Show helpers | |
| 749 | + | # ============================================ | |
| 750 | + | ||
| 751 | + | function cmd::rule::_show_section() { | |
| 752 | + | local title="${1:-}" color="${2:-white}" use_color="${3:-false}" | |
| 753 | + | local color_code="" | |
| 754 | + | if $use_color; then | |
| 755 | + | case "$color" in | |
| 756 | + | green) color_code="\033[0;32m" ;; | |
| 757 | + | red) color_code="\033[0;31m" ;; | |
| 758 | + | esac | |
| 759 | + | fi | |
| 760 | + | printf "\n ${color_code}── %s ──────────────────────────────────\033[0m\n" "$title" | |
| 761 | + | } | |
上一页
下一页