commit 170d8dae2793c7d397d1145255993b561bb30f05
author George Burgess IV <gbiv@google.com> Wed Feb 24 15:55:55 2016 -0800
committer George Burgess IV <gbiv@google.com> Fri Feb 26 11:48:57 2016 -0800
tree c363b3d72f0c9c3979dc1e1041a5c7b6304b325c
parent 8f1913d47990cf828d138b6624b0b286e6e31add

Add mode flags to open calls with O_CREAT.

Change-Id: Ib9fb61155d15ed9eb54942959af965df2fe21b71

bench/benchgen.py

diff --git a/bench/benchgen.py b/bench/benchgen.py
index d529332..ec14aef 100644
--- a/bench/benchgen.py
+++ b/bench/benchgen.py
@@ -183,7 +183,12 @@
                 if handle not in defined:
                     print >>bench, "int ",
                     defined.add(handle)
-                print >>bench, '%s = TEMP_FAILURE_RETRY(open("file%s", %s));' % (handle, f.ident, e.args[2])
+                create_mode = ''
+                if 'O_CREAT' in e.args[2]:
+                    assert len(e.args) > 3, 'File creation lacks a mode?'
+                    create_mode = ', ' + e.args[3]
+                print >>bench, '%s = TEMP_FAILURE_RETRY(open("file%s", %s%s));' \
+                        % (handle, f.ident, e.args[2], create_mode)
 
         elif e.call == "close":
             fd, f, handle = extract_file(e, e.args[0])