From 3d568db0c9837ed017348df5c7be34f7ecfa9111 Mon Sep 17 00:00:00 2001
From: Brian Coca <briancoca+dev@gmail.com>
Date: Tue, 7 May 2013 22:08:29 -0400
Subject: [PATCH] changed lininfile to be safer by writing to temp file and
 moving into place instead of writing directly to original Signed-off-by:
 Brian Coca <briancoca+dev@gmail.com>

---
 library/files/lineinfile | 29 +++++++++++++++++------------
 1 file changed, 17 insertions(+), 12 deletions(-)

diff --git a/library/files/lineinfile b/library/files/lineinfile
index b01a8faaf5..e23748a0e2 100644
--- a/library/files/lineinfile
+++ b/library/files/lineinfile
@@ -20,6 +20,7 @@
 
 import re
 import os
+import tempfile
 
 DOCUMENTATION = """
 ---
@@ -128,6 +129,14 @@ EXAMPLES = r"""
    lineinfile: dest=/opt/jboss-as/bin/standalone.conf regexp='^(.*)Xms(\d+)m(.*)$' line='\\1Xms${xms}m\\3' backrefs=yes
 """
 
+def write_changes(module,lines,dest):
+
+    tmpfd, tmpfile = tempfile.mkstemp()
+    f = os.fdopen(tmpfd,'wb')
+    f.writelines(lines)
+    f.close()
+
+    module.atomic_move(tmpfile, dest)
 
 def check_file_attrs(module, changed, message):
 
@@ -145,9 +154,7 @@ def check_file_attrs(module, changed, message):
 def present(module, dest, regexp, line, insertafter, insertbefore, create,
             backup, backrefs):
 
-    if os.path.isdir(dest):
-        module.fail_json(rc=256, msg='Destination %s is a directory !' % dest)
-    elif not os.path.exists(dest):
+    if not os.path.exists(dest):
         if not create:
             module.fail_json(rc=257, msg='Destination %s does not exist !' % dest)
         destpath = os.path.dirname(dest)
@@ -229,9 +236,7 @@ def present(module, dest, regexp, line, insertafter, insertbefore, create,
     if changed and not module.check_mode:
         if backup and os.path.exists(dest):
             module.backup_local(dest)
-        f = open(dest, 'wb')
-        f.writelines(lines)
-        f.close()
+        write_changes(module, lines, dest)
 
     msg, changed = check_file_attrs(module, changed, msg)
     module.exit_json(changed=changed, msg=msg)
@@ -239,9 +244,7 @@ def present(module, dest, regexp, line, insertafter, insertbefore, create,
 
 def absent(module, dest, regexp, backup):
 
-    if os.path.isdir(dest):
-        module.fail_json(rc=256, msg='Destination %s is a directory !' % dest)
-    elif not os.path.exists(dest):
+    if not os.path.exists(dest):
         module.exit_json(changed=False, msg="file not present")
 
     msg = ""
@@ -264,9 +267,7 @@ def absent(module, dest, regexp, backup):
     if changed and not module.check_mode:
         if backup:
             module.backup_local(dest)
-        f = open(dest, 'wb')
-        f.writelines(lines)
-        f.close()
+        write_changes(module, lines, dest)
 
     if changed:
         msg = "%s line(s) removed" % len(found)
@@ -299,6 +300,10 @@ def main():
     backrefs = module.params['backrefs']
     dest = os.path.expanduser(params['dest'])
 
+
+    if os.path.isdir(dest):
+        module.fail_json(rc=256, msg='Destination %s is a directory !' % dest)
+
     if params['state'] == 'present':
         if 'line' not in params:
             module.fail_json(msg='line= is required with state=present')