ruby-list

こんにちは。

pstoreは，ファイルを書き換えるときにバックアップファイルを作っています
が，タイミングによっては安全ではありません。

ファイルを開くときには、まず"rb+"で開こうとし，失敗したときに"wb+"で開
きなおしていますが，タイミングによってはファイルが切り詰められてしまい
ます。

バックアップファイルが存在するときに，
　　・元ファイルが正しくてバックアップファイルが不正
　　・バックアップファイルが正しくて元ファイルが不正

の両方の可能性があるために，どちらも信頼することができません。というわ
けで，パッチを書きました。

.tmp -> .new -> 元ファイルという順でファイルを変更します。

--- pstore.rb.orig	2004-02-08 20:30:25.200663384 +0900
+++ pstore.rb	2004-02-08 21:01:29.396262576 +0900
@@ -35,7 +35,11 @@
   def in_transaction
     raise PStore::Error, "not in transaction" unless @transaction
   end
-  private :in_transaction
+  def in_transaction_wr()
+    in_transaction()
+    raise PStore::Error, "in read-only transaction" if @rdonly
+  end
+  private :in_transaction, :in_transaction_wr
 
   def [](name)
     in_transaction
@@ -52,11 +56,11 @@
     self[name]
   end
   def []=(name, value)
-    in_transaction
+    in_transaction_wr()
     @table[name] = value
   end
   def delete(name)
-    in_transaction
+    in_transaction_wr()
     @table.delete name
   end
 
@@ -86,27 +90,26 @@
   def transaction(read_only=false)
     raise PStore::Error, "nested transaction" if @transaction
     begin
+      @rdonly = read_only
+      @abort = false
       @transaction = true
       value = nil
-      backup = @filename+"~"
-      begin
-	file = File::open(@filename, read_only ? "rb" : "rb+")
-	orig = true
-      rescue Errno::ENOENT
-	raise if read_only
-	file = File::open(@filename, "wb+")
-      end
-      file.flock(read_only ? File::LOCK_SH : File::LOCK_EX)
-      if read_only
-	@table = Marshal::load(file)
-      elsif orig and (content = file.read) != ""
+      file = File.open(@filename, File::RDWR | File::CREAT)
+      file.flock(File::LOCK_EX)
+      new_file = @filename + ".new"
+      commit_new() if FileTest.exist?(new_file)
+
+      if (content = file.read) != ""
 	@table = Marshal::load(content)
-	size = content.size
-	md5 = Digest::MD5.digest(content)
-	content = nil		# unreference huge data
+        if !read_only
+          size = content.size
+          md5 = Digest::MD5.digest(content)
+        end
       else
 	@table = {}
       end
+      content = nil		# unreference huge data
+
       begin
 	catch(:pstore_abort_transaction) do
 	  value = yield(self)
@@ -116,24 +119,17 @@
 	raise
       ensure
 	if !read_only and !@abort
-	  file.rewind
+          tmp_file = @filename + ".tmp"
 	  content = Marshal::dump(@table)
 	  if !md5 || size != content.size || md5 != Digest::MD5.digest(content)
-	    File::copy @filename, backup
-	    begin
-	      file.write(content)
-	      file.truncate(file.pos)
-	      content = nil		# unreference huge data
-	    rescue
-	      File::rename backup, @filename if File::exist?(backup)
-	      raise
-	    end
-	  end
-	end
-	if @abort and !orig
-	  File.unlink(@filename)
+            File.open(tmp_file, "w") {|t|
+              t.write(content)
+            }
+            File.rename(tmp_file, new_file)
+            commit_new()
+          end
+          content = nil		# unreference huge data
 	end
-	@abort = false
       end
     ensure
       @table = nil
@@ -142,6 +138,15 @@
     end
     value
   end
+
+  private
+  def commit_new()
+    new_file = @filename + ".new"
+    if !File.copy(new_file, @filename)
+      raise IOError
+    end
+    File.unlink(new_file)
+  end
 end
 
 if __FILE__ == $0

-- 
HORIKAWA Hisashi (in Kanji: 堀川 久)
Netsphere Laboratories  http://www.nslabs.jp/

Thread

Prev Next

In This Thread

Prev Next

[#39102] pstoreのバックアップファイルについて — 堀川 久 <vzw00011@...>

[#39104] Internet Explorer で印刷 — Shin-ichiro HARA <sinara@...>

[#39106] What processor do you run ruby on? — "K.Sasada" <ko1@...>

[#39111] *a=aの挙動の違い — Tamiyasu Murakami <tamiyasu_m@...>

[#39118] YAML without Syck — IWATSUKI Hiroyuki <don@...>

[#39129] InternetExplorer ってインターフェースとして使える？ — Shin-ichiro HARA <sinara@...>

[#39142] WORD Graphのデータシートにデータを貼り付けようとしています — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39148] [BUG] ruby-mode indentation bug — Kenta MURATA <muraken2@...>

[#39154] Re: WORD Graphのデータシートにデータを貼り付けようとしています — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39158] setup.rb on mswin32 — KUBO Takehiro <kubo@...>

[#39170] [ANN] Kwartz -- a template system, available in Ruby, PHP and Java — "maco@..." <kwa@...>

[#39188] debugger on RubyUnit — URA Takefumi <hage@...>

[#39189] MinGW — EMOTO Masahiko <emo@...>

[#39192] Re: WORD Graph のデータシートにデータを貼り付けようとしています — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39197] [ANN] Nora 0.0.20040216 released — MoonWolf <moonwolf@...>

[#39198] RHG 全章公開 — Minero Aoki <aamine@...>

[#39201] undefined reference to `__imp__ruby_safe_level' — EMOTO Masahiko <emo@...>

[#39202] Re: WORD Graph のデータシートにデータを貼り付けようとしています — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39204] Re: WORD Graph のデータシートにデータを貼り付けようとしています — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39212] 特異メソッドでのクラス変数 — "eclipse" <eclipse@...>

[#39213] IconvでUTF-16を変換 — Takashi & Kayoko Sano <tksano@...3.kcn.ne.jp>

[#39232] クラス変数をモジュールのメソッドから参照したい — ruby@...

[#39233] MinGW / socket.so — EMOTO Masahiko <emo@...>

[#39235] Zlib memory leak — sheepman <sheepman@...>

[#39236] uriってどう使うんでしょうか？ — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39237] erb コマンドが compile error になる — Kouhei Yanagita <sugi@...>

[#39239] Re: uriってどう使うんでしょうか？ — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39243] How about 'irb -I path'? — IKEGAMI Daisuke <ikegami@...>

[#39245] Re: uriってどう使うんでしょうか — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39248] Re: uriってどう使うんでしょうか — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39250] 気象庁の．．． Re: uriってどう使うんでしょうか — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39251] ipv6 patch for native win32 — "U.Nakamura" <usa@...>

[#39252] ruby-mode.el のインデントの問題 — Kouhei Yanagita <sugi@...>

[#39254] [ANN] Nora 0.0.20040222 released — MoonWolf <moonwolf@...>

[#39255] 気象庁の．．． Re: uriってどう使うんでしょうか — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39257] Re: 気象庁の．．． Re: uriってどう使うんでしょうか — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39260] File#path.tainted? — 堀川 久 <vzw00011@...>

[#39265] 1.8.1 の File.basename — FUJIMOTO Hisakuni <hisa@...>

[#39269] Ruby 0.95 — Masayoshi Takahashi <maki@...>

[#39270] OSX + REXML — Mitsuru Koyama <mit@...>

[#39273] Re: 気象庁の．．． Re: uriってどう使うんでしょうか — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39275] DnD on win32 — Shinichiro HIDA <shinichiro@...>

[#39286] Re: [ruby-dev:16312] Re: ライブラリ拡大計画 — "NAKAMURA, Hiroshi" <nakahiro@...>

[#39288] 固有値、固有ベクトルの計算 — Masahiro Sato <msato@...>

[ruby-list:39102] pstoreのバックアップファイルについて

Thread

In This Thread

[#39102] pstoreのバックアップファイルについて — 堀川久 <vzw00011@...>

[#39142] WORD　Graphのデータシートにデータを貼り付けようとしています — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39154] Re: WORD　Graphのデータシートにデータを貼り付けようとしています — Hideyuki Oya <h_oya@...7.dion.ne.jp>

[#39201] undefined reference to `impruby_safe_level' — EMOTO Masahiko <emo@...>

[#39260] File#path.tainted? — 堀川久 <vzw00011@...>