1 files changed, 6 insertions, 2 deletions

diff --git a/setup.sh b/setup.sh
index 64d5142..6d51389 100755
--- a/setup.sh
+++ b/setup.sh
@@ -78,7 +78,6 @@ protocol static {
 EOF
     if [ -n "$ospf" ]; then
       interfacelist=$(
-        echo "  area 0 {"
         echo "    interface \"lo\" { stub; };"
         getval interface | while read -r inter; do
           val="$(getval "interface $inter" OSPF)"
@@ -91,7 +90,6 @@ EOF
             echo "    };"
           fi
         done
-        echo "  };"
       )
 
       cat <<EOF
@@ -100,14 +98,20 @@ protocol ospf v3 ospf4 {
     import all;
     export filter { if source ~ [RTS_DEVICE, RTS_INHERIT] && net ~ [$(getval "Subnet" | grep -v ':' | while read -r line; do printf "%s+," "$line"; done | sed 's/,$//')] then accept; else reject; };
   };
+  area 0 {
+$(getval "Stubnet" | grep -v ':' | while read -r net; do echo "    stubnet $net {};"; done)
 $interfacelist
+  };
 }
 protocol ospf v3 ospf6 {
   ipv6 {
     import all;
     export filter { if source ~ [RTS_DEVICE, RTS_INHERIT] && net ~ [$(getval "Subnet" | grep ':' | while read -r line; do printf "%s+," "$line"; done | sed 's/,$//')] then accept; else reject; };
   };
+  area 0 {
+$(getval "Stubnet" | grep ':' | while read -r net; do echo "    stubnet $net {};"; done)
 $interfacelist
+  };
 }
 EOF
     fi