diff options
Diffstat (limited to 'lib')
-rw-r--r-- | lib/cgi/session.rb | 16 |
1 files changed, 12 insertions, 4 deletions
diff --git a/lib/cgi/session.rb b/lib/cgi/session.rb index c6145d9368..c693e06df2 100644 --- a/lib/cgi/session.rb +++ b/lib/cgi/session.rb @@ -397,8 +397,9 @@ class CGI unless @hash @hash = {} begin + lockf = File.open(@path+".lock", "r") + lockf.flock File::LOCK_SH f = File.open(@path, 'r') - f.flock File::LOCK_SH for line in f line.chomp! k, v = line.split('=',2) @@ -406,6 +407,7 @@ class CGI end ensure f.close unless f.nil? + lockf.close if lockf end end @hash @@ -415,13 +417,17 @@ class CGI def update return unless @hash begin - f = File.open(@path, File::CREAT|File::TRUNC|File::RDWR, 0600) - f.flock File::LOCK_EX + lockf = File.open(@path+".lock", File::CREAT|File::RDWR, 0600) + lockf.flock File::LOCK_EX + f = File.open(@path+".new", File::CREAT|File::TRUNC|File::WRONLY, 0600) for k,v in @hash f.printf "%s=%s\n", CGI::escape(k), CGI::escape(String(v)) end + f.close + File.rename @path+".new", @path ensure - f.close unless f.nil? + f.close if f and !f.closed? + lockf.close if lockf end end @@ -432,6 +438,8 @@ class CGI # Close and delete the session's FileStore file. def delete + File::unlink @path+".lock" rescue nil + File::unlink @path+".new" rescue nil File::unlink @path rescue Errno::ENOENT end |