#!/bin/sh /etc/rc.common

conlog() {
    msg="$@"
    test -z "$msg" && return 0
    logger -t "dnsmasq[${$}]" "$msg"
    #DATE="$(date)"
    #echo "$DATE dnsmasq[${$}]: $msg" >>/dev/console 2>/dev/null
    return 0
}

syslog() {
    msg="$@"
    test -z "$msg" && return 0
    logger -p9 -t "dnsmasq[${$}]" "$msg"
    return 0
}

xappend() {
    local value="$1"

    echo "${value#--}" >>$CONFIGFILE
}

dhcp_calc() {
    local ip="$1"
    local res=0

    while [ -n "$ip" ]; do
        part="${ip%%.*}"
        res="$(($res * 256))"
        res="$(($res + $part))"
        [ "${ip%.*}" != "$ip" ] && ip="${ip#*.}" || ip=
    done
    echo "$res"
}

append_bool() {
    local section="$1"
    local option="$2"
    local value="$3"
    local _loctmp
    config_get_bool _loctmp "$section" "$option" 0
    [ $_loctmp -gt 0 ] && xappend "$value"
}

append_parm() {
    local section="$1"
    local option="$2"
    local switch="$3"
    local _loctmp
    config_get _loctmp "$section" "$option"
    [ -z "$_loctmp" ] && return 0
    xappend "$switch=$_loctmp"
}

append_server() {
    xappend "--server=$1"
}

append_address() {
    xappend "--address=$1"
}

append_interface() {
    local ifname=$(uci_get_state network "$1" ifname "$1")
    xappend "--interface=$ifname"
}

append_notinterface() {
    local ifname=$(uci_get_state network "$1" ifname "$1")
    xappend "--except-interface=$ifname"
}

append_addnhosts() {
    xappend "--addn-hosts=$1"
}

append_bogusnxdomain() {
    xappend "--bogus-nxdomain=$1"
}

dnsmasq() {
    local cfg="$1"
    append_bool "$cfg" authoritative "--dhcp-authoritative"
    append_bool "$cfg" nodaemon "--no-daemon"
    append_bool "$cfg" domainneeded "--domain-needed"
    append_bool "$cfg" filterwin2k "--filterwin2k"
    append_bool "$cfg" nohosts "--no-hosts"
    append_bool "$cfg" nonegcache "--no-negcache"
    append_bool "$cfg" strictorder "--strict-order"
    append_bool "$cfg" logqueries "--log-queries"
    append_bool "$cfg" logdhcp "--log-dhcp"
    append_bool "$cfg" clearonreload "--clear-on-reload"
    append_bool "$cfg" noresolv "--no-resolv"
    append_bool "$cfg" localise_queries "--localise-queries"
    append_bool "$cfg" readethers "--read-ethers"
    append_bool "$cfg" dbus "--enable-dbus"
    append_bool "$cfg" boguspriv "--bogus-priv"
    append_bool "$cfg" expandhosts "--expand-hosts"
    append_bool "$cfg" enable_tftp "--enable-tftp"
    append_bool "$cfg" nonwildcard "--bind-interfaces"
    append_bool "$cfg" fqdn "--dhcp-fqdn"
    append_bool "$cfg" allservers "--all-servers"
    append_bool "$cfg" client_update_ddns "--dhcp-client-update"

    #--dhcp-option-force=[tag:<tag>,[tag:<tag>,]][encap:<opt>,][vi-encap:<enterprise>,][vendor:[<vendor-class>],]<opt>,[<value>[,<value>]]
    #This works in exactly the same way as --dhcp-option except that the option will always be sent, even if the client does not ask for
    #it in the parameter request list. This is sometimes needed, for example when sending options to PXELinux.
    #
    #--dhcp-option-force=43,ANDROID_METERED
    #
    append_parm "$cfg" negttl "--neg-ttl"
    append_parm "$cfg" maxttl "--max-ttl"
    append_parm "$cfg" maxcachettl "--max-cache-ttl"
    append_parm "$cfg" dhcpscript "--dhcp-script"
    append_parm "$cfg" cachesize "--cache-size"
    append_parm "$cfg" dnsforwardmax "--dns-forward-max"
    append_parm "$cfg" port "--port"
    append_parm "$cfg" ednspacket_max "--edns-packet-max"
    append_parm "$cfg" dhcpleasemax "--dhcp-lease-max"
    append_parm "$cfg" "queryport" "--query-port"
    append_parm "$cfg" "domain" "--domain"
    append_parm "$cfg" "local" "--server"
    config_list_foreach "$cfg" "server" append_server
    config_list_foreach "$cfg" "address" append_address
    config_list_foreach "$cfg" "interface" append_interface
    config_list_foreach "$cfg" "notinterface" append_notinterface
    config_list_foreach "$cfg" "addnhosts" append_addnhosts
    config_list_foreach "$cfg" "bogusnxdomain" append_bogusnxdomain
    append_parm "$cfg" "leasefile" "--dhcp-leasefile"
    append_parm "$cfg" "resolvfile" "--resolv-file"
    append_parm "$cfg" "tftp_root" "--tftp-root"
    append_parm "$cfg" "dhcp_boot" "--dhcp-boot"
    append_parm "$cfg" "local_ttl" "--local-ttl"

    config_get DOMAIN "$cfg" domain

    config_get_bool ADD_LOCAL_DOMAIN "$cfg" add_local_domain 1
    config_get_bool ADD_LOCAL_HOSTNAME "$cfg" add_local_hostname 1

    config_get_bool readethers "$cfg" readethers
    [ "$readethers" = "1" -a \! -e "/etc/ethers" ] && touch /etc/ethers

    config_get leasefile $cfg leasefile
    [ -n "$leasefile" -a \! -e "$leasefile" ] && touch "$leasefile"
    config_get_bool cachelocal "$cfg" cachelocal 1

    config_get hostsfile "$cfg" dhcphostsfile
    [ -e "$hostsfile" ] && xappend "--dhcp-hostsfile=$hostsfile"

    mkdir -p /tmp/hosts
    xappend "--addn-hosts=/tmp/hosts"

    local rebind
    config_get_bool rebind "$cfg" rebind_protection 1
    [ $rebind -gt 0 ] && {
        logger -t dnsmasq \
            "DNS rebinding protection is active," \
            "will discard upstream RFC1918 responses!"
        xappend "--stop-dns-rebind"

        local rebind_localhost
        config_get_bool rebind_localhost "$cfg" rebind_localhost 0
        [ $rebind_localhost -gt 0 ] && {
            logger -t dnsmasq "Allowing 127.0.0.0/8 responses"
            xappend "--rebind-localhost-ok"
        }

        append_rebind_domain() {
            logger -t dnsmasq "Allowing RFC1918 responses for domain $1"
            xappend "--rebind-domain-ok=$1"
        }

        config_list_foreach "$cfg" rebind_domain append_rebind_domain
    }

    [ -f "/www/xiaoqiang/diagnosis/index.html" ] && {
        # network troubleshoot intercept
        local intercept="$(uci -q get network.lan.ipaddr)"
        [ -n "$intercept" ] && {
            xappend "--intercept-ip-address=$intercept"
            [ -f "/etc/intercept_wlist" ] && xappend "--intercept-white-list=/etc/intercept_wlist"
        }
    }

    local is_nxdomain=$(uci -q get http_status_stat.settings.enabled)
    is_nxdomain=${is_nxdomain:-"0"}
    config_get_bool enable_nxdomain_resolve "$cfg" nxdomain_helper 1
    local cc=$(bdata get CountryCode)
    cc=${cc:-"CN"}
    local nxdomain_ipaddr=$(uci -q get firewall.nxdomain.src_dip)
    [ $is_nxdomain -eq "1" -a $cc = "CN" -a "$enable_nxdomain_resolve" = "1" -a -n "$nxdomain_ipaddr" ] && {
        # replace nxdomain
        xappend "--nxdomain-ip-address=$nxdomain_ipaddr"
    }

    # dnsmasq script
    [ -d "/lib/dnsmasq/lease.d" ] && {
        xappend "--dhcp-script=/lib/dnsmasq/dhcp_lease"
    }
    dhcp_option_add "$cfg" "" 0

    echo >>$CONFIGFILE
}

dhcp_subscrid_add() {
    local cfg="$1"

    config_get networkid "$cfg" networkid
    [ -n "$networkid" ] || return 0

    config_get subscriberid "$cfg" subscriberid
    [ -n "$subscriberid" ] || return 0

    xappend "--dhcp-subscrid=$networkid,$subscriberid"

    config_get_bool force "$cfg" force 0

    dhcp_option_add "$cfg" "$networkid" "$force"
}

dhcp_remoteid_add() {
    local cfg="$1"

    config_get networkid "$cfg" networkid
    [ -n "$networkid" ] || return 0

    config_get remoteid "$cfg" remoteid
    [ -n "$remoteid" ] || return 0

    xappend "--dhcp-remoteid=$networkid,$remoteid"

    config_get_bool force "$cfg" force 0

    dhcp_option_add "$cfg" "$networkid" "$force"
}

dhcp_circuitid_add() {
    local cfg="$1"

    config_get networkid "$cfg" networkid
    [ -n "$networkid" ] || return 0

    config_get circuitid "$cfg" circuitid
    [ -n "$circuitid" ] || return 0

    xappend "--dhcp-circuitid=$networkid,$circuitid"

    config_get_bool force "$cfg" force 0

    dhcp_option_add "$cfg" "$networkid" "$force"
}

dhcp_userclass_add() {
    local cfg="$1"

    config_get networkid "$cfg" networkid
    [ -n "$networkid" ] || return 0

    config_get userclass "$cfg" userclass
    [ -n "$userclass" ] || return 0

    xappend "--dhcp-userclass=$networkid,$userclass"

    config_get_bool force "$cfg" force 0

    dhcp_option_add "$cfg" "$networkid" "$force"
}

dhcp_vendorclass_add() {
    local cfg="$1"

    config_get networkid "$cfg" networkid
    [ -n "$networkid" ] || return 0

    config_get vendorclass "$cfg" vendorclass
    [ -n "$vendorclass" ] || return 0

    xappend "--dhcp-vendorclass=$networkid,$vendorclass"

    config_get_bool force "$cfg" force 0

    dhcp_option_add "$cfg" "$networkid" "$force"
}

dhcp_host_add() {
    local cfg="$1"

    config_get_bool force "$cfg" force 0

    config_get networkid "$cfg" networkid
    [ -n "$networkid" ] && dhcp_option_add "$cfg" "$networkid" "$force"

    config_get ip "$cfg" ip
    [ -n "$ip" ] || return 0

    macs=""
    config_get mac "$cfg" mac
    for m in $mac; do append macs "$m" ","; done
    [ -n "$macs" ] || return 0

    config_get tag "$cfg" tag

    xappend "--dhcp-host=$macs${networkid:+,net:$networkid}${tag:+,set:$tag}${ip:+,$ip}"
}

dhcp_tag_add() {
    local cfg="$1"

    tag="$cfg"

    [ -n "$tag" ] || return 0

    config_get_bool force "$cfg" force 0
    [ "$force" = "0" ] && force=

    config_get option "$cfg" dhcp_option
    for o in $option; do
        xappend "--dhcp-option${force:+-force}=tag:$tag,$o"
    done
}

dhcp_mac_add() {
    local cfg="$1"

    config_get networkid "$cfg" networkid
    [ -n "$networkid" ] || return 0

    config_get mac "$cfg" mac
    [ -n "$mac" ] || return 0

    xappend "--dhcp-mac=$networkid,$mac"

    dhcp_option_add "$cfg" "$networkid"
}

dhcp_boot_add() {
    local cfg="$1"

    config_get networkid "$cfg" networkid

    config_get filename "$cfg" filename
    [ -n "$filename" ] || return 0

    config_get servername "$cfg" servername
    [ -n "$servername" ] || return 0

    config_get serveraddress "$cfg" serveraddress
    [ -n "$serveraddress" ] || return 0

    xappend "--dhcp-boot=${networkid:+net:$networkid,}$filename,$servername,$serveraddress"

    config_get_bool force "$cfg" force 0

    dhcp_option_add "$cfg" "$networkid" "$force"
}

dhcp_add() {
    local cfg="$1"
    config_get net "$cfg" interface
    [ -n "$net" ] || return 0

    config_get networkid "$cfg" networkid
    [ -n "$networkid" ] || networkid="$net"

    config_get ifname "$net" ifname

    [ "$cachelocal" = "0" ] && network_get_dnsserver dnsserver "$net" && {
        DNS_SERVERS="$DNS_SERVERS $dnsserver"
    }

    append_bool "$cfg" ignore "--no-dhcp-interface=$ifname" && return 0

    config_get proto "$net" proto
    [ static = "$proto" ] || return 0

    config_get ipaddr "$net" ipaddr
    config_get netmask "$cfg" netmask
    [ -n "$netmask" ] || config_get netmask "$net" netmask

    #check for an already active dhcp server on the interface, unless 'force' is set
    config_get_bool force "$cfg" force 0
    [ $force -gt 0 -o -z "$ifname" ] || {
        udhcpc -n -q -s /bin/true -t 1 -i $ifname >&- && {
            conlog "found already running DHCP-server on interface '$ifname'" \
                "refusing to start, use 'option force 1' to override"
            return 0
        }
    }

    config_get start "$cfg" start
    config_get limit "$cfg" limit
    config_get leasetime "$cfg" leasetime
    config_get options "$cfg" options
    config_get_bool dynamicdhcp "$cfg" dynamicdhcp 1

    leasetime="${leasetime:-12h}"
    start="$(dhcp_calc "${start:-100}")"
    limit="${limit:-150}"
    [ "$limit" -gt 0 ] && limit=$((limit - 1))
    eval "$(ipcalc.sh $ipaddr $netmask $start $limit)"
    if [ "$dynamicdhcp" = "0" ]; then END="static"; fi
    xappend "--dhcp-range=$networkid,$START,$END,$NETMASK,$leasetime${options:+ $options}"

    dhcp_option_add "$cfg" "$networkid"
    dhcp_option_force_add "$cfg" "$networkid"
}

dhcp_option_add() {
    local cfg="$1"
    local networkid="$2"
    local force="$3"

    [ "$force" = "0" ] && force=

    config_get dhcp_option "$cfg" dhcp_option
    for o in $dhcp_option; do
        xappend "--dhcp-option${force:+-force}=${networkid:+$networkid,}$o"
    done
}

dhcp_option_force_add() {
    local cfg="$1"
    local networkid="$2"
    local force="1"

    config_get dhcp_option_force "$cfg" dhcp_option_force
    for o in $dhcp_option_force; do
        xappend "--dhcp-option${force:+-force}=${networkid:+$networkid,}$o"
    done

    local model=$(uci -q get misc.hardware.model)
    [ -z "$model" ] && model=$(cat /proc/xiaoqiang/model)
    xappend "--dhcp-option-force=12,MiWiFi-$model-srv"

    local color=$(bdata get color)
    local version=$(uci -q -c /usr/share/xiaoqiang get xiaoqiang_version.version.ROM)
    # add vendor info options
    xappend "dhcp-option-force=43,miwifi-$model-$version${color:+-$color}"
}

dhcp_domain_add() {
    local cfg="$1"
    local ip name names

    config_get names "$cfg" name "$2"
    [ -n "$names" ] || return 0

    config_get ip "$cfg" ip "$3"
    [ -n "$ip" ] || return 0

    local oIFS="$IFS"
    IFS="."
    set -- $ip
    IFS="$oIFS"
    local raddr="${4:+$4.$3.$2.$1.in-addr.arpa}"

    for name in $names; do
        local fqdn="$name"

        [ "${fqdn%.*}" == "$fqdn" ] &&
            fqdn="$fqdn${DOMAIN:+.$DOMAIN}"

        xappend "--address=/$fqdn/$ip"

        [ -n "$raddr" ] && {
            xappend "--ptr-record=$raddr,$fqdn"
            raddr=""
        }
    done
}

dhcp_srv_add() {
    local cfg="$1"

    config_get srv "$cfg" srv
    [ -n "$srv" ] || return 0

    config_get target "$cfg" target
    [ -n "$target" ] || return 0

    config_get port "$cfg" port
    [ -n "$port" ] || return 0

    config_get class "$cfg" class
    config_get weight "$cfg" weight

    local service="$srv,$target,$port${class:+,$class${weight:+,$weight}}"

    xappend "--srv-host=$service"
}

dhcp_mx_add() {
    local cfg="$1"
    local domain relay pref

    config_get domain "$cfg" domain
    [ -n "$domain" ] || return 0

    config_get relay "$cfg" relay
    [ -n "$relay" ] || return 0

    config_get pref "$cfg" pref 0

    local service="$domain,$relay,$pref"

    xappend "--mx-host=$service"
}

dhcp_cname_add() {
    local cfg="$1"
    local cname target

    config_get cname "$cfg" cname
    [ -n "$cname" ] || return 0

    config_get target "$cfg" target
    [ -n "$target" ] || return 0

    xappend "--cname=${cname},${target}"
}

killonepid() {
    local onepid ksig timelimit wcnt
    onepid=$1
    ksig=$2
    if [ -z "$onepid" ]; then
        return 0
    fi
    test -z "$ksig" && ksig=15
    #
    if [ $onepid -le 100 ]; then
        return 0
    fi
    timelimit=5
    wcnt=0
    while [ : ]; do
        if [ "$ksig" = '0' ]; then
            kill -${ksig} $onepid 2>/dev/null
            return $?
        fi
        kill -${ksig} $onepid 2>/dev/null
        test $? -ne 0 && return 0
        let wcnt=$wcnt+1
        test $wcnt -gt $timelimit && break
        sleep 1
    done
    kill -9 $onepid 2>/dev/null
    test $? -eq 0 && return 1
    return 0
    #
}
