firewall.core.fw_ifcfg: New module to search for and change ifcfg files

src/firewall/core/fw_ifcfg.py

@@ -0,0 +1,67 @@
+#!/usr/bin/python
+# -*- coding: utf-8 -*-
+#
+# Copyright (C) 2010-2016 Red Hat, Inc.
+#
+# Authors:
+# Thomas Woerner <twoerner@redhat.com>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; either version 2 of the License, or
+# (at your option) any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <http://www.gnu.org/licenses/>.
+#
+
+"""Functions to search for and change ifcfg files"""
+
+__all__ = [ "search_ifcfg_of_interface", "ifcfg_set_zone_of_interface" ]
+
+import os, os.path
+
+from firewall import config
+from firewall.core.logger import log
+from firewall.core.io.ifcfg import ifcfg
+
+def search_ifcfg_of_interface(interface):
+    """search ifcfg file for the interface in config.IFCFGDIR"""
+
+    filename = "%s/ifcfg-%s" % (config.IFCFGDIR, interface)
+    if os.path.exists(filename):
+        ifcfg_file = ifcfg(filename)
+        ifcfg_file.read()
+        if ifcfg_file.get("DEVICE") == interface:
+            return ifcfg_file
+
+    for filename in sorted(os.listdir(config.IFCFGDIR)):
+        if not filename.startswith("ifcfg-"):
+            continue
+        for ignored in [ ".bak", ".orig", ".rpmnew", ".rpmorig", ".rpmsave",
+                         "-range" ]:
+            if filename.endswith(ignored):
+                continue
+        if "." in filename:
+            continue
+        ifcfg_file = ifcfg("%s/%s" % (config.IFCFGDIR, filename))
+        ifcfg_file.read()
+        if ifcfg_file.get("DEVICE") == interface:
+            return ifcfg_file
+
+    return None
+
+def ifcfg_set_zone_of_interface(zone, interface):
+    """Set zone (ZONE=<zone>) in the ifcfg file that uses the interface
+    (DEVICE=<interface>)"""
+
+    ifcfg_file = search_ifcfg_of_interface(interface)
+    if ifcfg_file is not None and ifcfg_file.get("ZONE") != zone:
+        log.debug1("Setting ZONE=%s in '%s'" % (zone, ifcfg_file.filename))
+        ifcfg_file.set("ZONE", zone)
+        ifcfg_file.write()