diff options
Diffstat (limited to 'cloudinit/CloudConfig/cc_update_etc_hosts.py')
| -rw-r--r-- | cloudinit/CloudConfig/cc_update_etc_hosts.py | 34 |
1 files changed, 33 insertions, 1 deletions
diff --git a/cloudinit/CloudConfig/cc_update_etc_hosts.py b/cloudinit/CloudConfig/cc_update_etc_hosts.py index 856cbae1..10ee5435 100644 --- a/cloudinit/CloudConfig/cc_update_etc_hosts.py +++ b/cloudinit/CloudConfig/cc_update_etc_hosts.py @@ -17,12 +17,44 @@ # along with this program. If not, see <http://www.gnu.org/licenses/>. import cloudinit.util as util from cloudinit.CloudConfig import per_always +import platform +import StringIO frequency = per_always def handle(name,cfg,cloud,log,args): if not util.get_cfg_option_bool(cfg,"manage_etc_hosts",False): - log.debug("manage_etc_hosts is not set. not modifying /etc/hosts") + log.debug("manage_etc_hosts is not set, checking sanity of /etc/hosts") + with open('/etc/hosts', 'r') as etchosts: + current_hostname = platform.node() + hosts_line = "# Added by cloud-init\n127.0.1.1\t%s.localdomain %s\n" % (current_hostname, current_hostname) + need_write = False + need_change = True + new_etchosts = StringIO.StringIO() + for line in etchosts: + split_line = [s.strip() for s in line.split()] + # skip over malformed /etc/hosts entries + if len(split_line) < 2: + continue + ip, hostnames = split_line[0], split_line[1:] + if ip == "127.0.1.1": + for hostname in hostnames: + if hostname == current_hostname: + need_change = False + if need_change == True: + line = hosts_line + need_change = False + need_write = True + new_etchosts.write(line) + etchosts.close() + if need_change == True: + new_etchosts.write(hosts_line) + need_write = True + if need_write == True: + new_etcfile = open ('/etc/hosts','wb') + new_etcfile.write(new_etchosts.getvalue()) + new_etcfile.close() + new_etchosts.close() return try: |