igmpproxy: automatically add firewall rules for multicast routing based on the curren...

[openwrt/staging/chunkeey.git] / package / network / services / igmpproxy / files / igmpproxy.init

diff --git a/package/network/services/igmpproxy/files/igmpproxy.init b/package/network/services/igmpproxy/files/igmpproxy.init
index a45978f5b681abd72aa9ea0e4bbdb504f4cdc2ff..8c15814712ff4282a67b3400f8cf7254b2b96d65 100644 (file)
--- a/package/network/services/igmpproxy/files/igmpproxy.init
+++ b/package/network/services/igmpproxy/files/igmpproxy.init
@@ -1,11 +1,10 @@
 #!/bin/sh /etc/rc.common
-# Copyright (C) 2010-2012 OpenWrt.org
+# Copyright (C) 2010-2014 OpenWrt.org
 
 START=99
-STOP=10
-
-SERVICE_DAEMONIZE=1
-SERVICE_WRITE_PID=1
+USE_PROCD=1
+PROG=/usr/sbin/igmpproxy
+CONFIGFILE=/var/etc/igmpproxy.conf
 
 # igmpproxy supports both a debug mode and verbosity, which are very useful
 # when something isn't working.
@@ -34,30 +33,102 @@ igmp_header() {
 }
 
 igmp_add_phyint() {
-        local network direction altnets
+        local network direction altnets device up
 
         config_get network $1 network
         config_get direction $1 direction
         config_get altnets $1 altnet
 
-        device=$(uci_get_state network "$network" ifname "$network")
-        echo -e "\nphyint $device $direction ratelimit 0 threshold 1" >> /var/etc/igmpproxy.conf
+       json_load "$(ifstatus $network)"
+       json_get_var device l3_device
+       json_get_var up up
+
+       [ -n "$device" -a "$up" = "1" ] || {
+               procd_append_param error "$network is not up"
+               return;
+       }
+
+       echo -e "\nphyint $device $direction ratelimit 0 threshold 1" >> /var/etc/igmpproxy.conf
+
+       if [ -n "$altnets" ]; then
+               local altnet
+               for altnet in $altnets; do
+                       echo -e "\taltnet $altnet" >> /var/etc/igmpproxy.conf
+               done
+       fi
+}
+
+igmp_add_network() {
+       local network
 
-        if [ -n "$altnets" ]; then
-                local altnet
-                for altnet in $altnets; do
-                        echo -e "\taltnet $altnet" >> /var/etc/igmpproxy.conf
-                done
-        fi
+       config_get network $1 network
+       procd_add_interface_trigger "interface.*" $network /etc/init.d/igmpproxy restart
 }
 
-start() {
+igmp_add_firewall_routing() {
+       config_get network $1 network
+       config_get direction $1 direction
+
+       [[ "$direction" = "downstream" ]] || return 0
+
+       json_add_object ""
+       json_add_string type rule
+       json_add_string src "$upstream"
+       json_add_string dest "$network"
+       json_add_string family ipv4
+       json_add_string proto udp
+       json_add_string dest_ip "224.0.0.0/4"
+       json_add_string target ACCEPT
+       json_close_object
+}
+
+igmp_add_firewall_network() {
+       config_get network $1 network
+       config_get direction $1 direction
+
+       json_add_object ""
+       json_add_string type rule
+       json_add_string src "$network"
+       json_add_string proto igmp
+       json_add_string target ACCEPT
+       json_close_object
+
+       [[ "$direction" = "upstream" ]] && {
+               upstream="$network"
+               config_foreach igmp_add_firewall_routing phyint
+       }
+}
+
+service_triggers() {
+       procd_add_reload_trigger "igmpproxy"
+}
+
+start_service() {
        config_load igmpproxy
+
+       procd_open_instance
        config_foreach igmp_header igmpproxy
        config_foreach igmp_add_phyint phyint
-       service_start /usr/sbin/igmpproxy $OPTIONS /etc/igmpproxy.conf
+       procd_set_param command $PROG
+       [ -n "$OPTIONS" ] && procd_append_param $OPTIONS
+       procd_append_param command $CONFIGFILE
+       procd_set_param file $CONFIGFILE
+       procd_set_param respawn
+       procd_open_trigger
+       config_foreach igmp_add_network phyint
+       procd_close_trigger
+
+       procd_open_data
+
+       json_add_array firewall
+       config_foreach igmp_add_firewall_network phyint
+       json_close_array
+
+       procd_close_data
+
+       procd_close_instance
 }
 
-stop() {
-       service_stop /usr/sbin/igmpproxy
+service_started() {
+       procd_set_config_changed firewall
 }