1  #
   2  # httpauth/htdigest.rb -- Apache compatible htdigest file
   3  #
   4  # Author: IPR -- Internet Programming with Ruby -- writers
   5  # Copyright (c) 2003 Internet Programming with Ruby writers. All rights
   6  # reserved.
   7  #
   8  # $IPR: htdigest.rb,v 1.4 2003/07/22 19:20:45 gotoyuzo Exp $
   9 
  10  require 'webrick/httpauth/userdb'
  11  require 'webrick/httpauth/digestauth'
  12  require 'tempfile'
  13 
  14  module WEBrick
  15    module HTTPAuth
  16      class Htdigest
  17        include UserDB
  18 
  19        def initialize(path)
  20          @path = path
  21          @mtime = Time.at(0)
  22          @digest = Hash.new
  23          @mutex = Mutex::new
  24          @auth_type = DigestAuth
  25          open(@path,"a").close unless File::exist?(@path)
  26          reload
  27        end
  28 
  29        def reload
  30          mtime = File::mtime(@path)
  31          if mtime > @mtime
  32            @digest.clear
  33            open(@path){|io|
  34              while line = io.gets
  35                line.chomp!
  36                user, realm, pass = line.split(/:/, 3)
  37                unless @digest[realm]
  38                  @digest[realm] = Hash.new
  39                end
  40                @digest[realm][user] = pass
  41              end
  42            }
  43            @mtime = mtime
  44          end
  45        end
  46 
  47        def flush(output=nil)
  48          output ||= @path
  49          tmp = Tempfile.new("htpasswd", File::dirname(output))
  50          begin
  51            each{|item| tmp.puts(item.join(":")) }
  52            tmp.close
  53            File::rename(tmp.path, output)
  54          rescue
  55            tmp.close(true)
  56          end
  57        end
  58 
  59        def get_passwd(realm, user, reload_db)
  60          reload() if reload_db
  61          if hash = @digest[realm]
  62            hash[user]
  63          end
  64        end
  65 
  66        def set_passwd(realm, user, pass)
  67          @mutex.synchronize{
  68            unless @digest[realm]
  69              @digest[realm] = Hash.new
  70            end
  71            @digest[realm][user] = make_passwd(realm, user, pass)
  72          }
  73        end
  74 
  75        def delete_passwd(realm, user)
  76          if hash = @digest[realm]
  77            hash.delete(user)
  78          end
  79        end
  80 
  81        def each
  82          @digest.keys.sort.each{|realm|
  83            hash = @digest[realm]
  84            hash.keys.sort.each{|user|
  85              yield([user, realm, hash[user]])
  86            }
  87          }
  88        end
  89      end
  90    end
  91  end