safe 0.2

data/README

@@ -0,0 +1,154 @@
+=== Safe ===
+Safe is a command line password management program. It securely stores all your
+user IDs and passwords using a single password as the master key. It stores the
+master key as a hash (SHA2), and uses it to encrypt the rest of the data using
+Blowfish.
+
+== License ==
+Safe is licensed under the BSD License. Copyright (c) 2007, Rob Warner
+
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without modification,
+are permitted provided that the following conditions are met:
+
+    * Redistributions of source code must retain the above copyright notice,
+      this list of conditions and the following disclaimer.
+    * Redistributions in binary form must reproduce the above copyright notice,
+      this list of conditions and the following disclaimer in the documentation
+      and/or other materials provided with the distribution.
+    * Neither the name of Rob Warner nor the names of its contributors may be
+      used to endorse or promote products derived from this software without
+      specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
+CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+
+== Installation ==
+To install safe, you must have Ruby and Ruby Gems installed. From the command
+line, type:
+
+  gem install safe --include-dependencies
+
+safe depends on two gems:
+
+  1) crypt
+  2) highline
+  
+Also, you must set up an environment variable called SAFE_DIR that points to the
+directory in which your encrypted password file (called .safe.xml) will be
+stored. Examples:
+
+Linux/Mac OS X:
+  export SAFE_DIR=~
+
+Windows:
+  set SAFE_DIR=c:\data
+
+== Version History ==
+0.1 -- May 12, 2007
+0.2 -- August 19, 2007
+
+See Release-Notes.txt for more information
+
+== Usage ==
+Run safe by typing:
+
+  safe.rb
+
+The first time you run safe, you will be prompted for a password, and a new 
+password file (.safe.xml) will be created in the directory specified by
+SAFE_DIR, like this:
+
+  $ safe.rb
+  Password: ****
+  Creating new file . . .
+
+If you have not set up the SAFE_DIR environment variable, you will see this
+message:
+
+  Set environment variable SAFE_DIR to the directory for your safe file
+
+To correct, set the SAFE_DIR environment variable.
+
+If you have set up SAFE_DIR to point to a non-existent directory, you will see
+this message:
+
+  Environment variable SAFE_DIR does not point to an existing directory
+  (/home/foo)
+
+where "/home/foo" is the value of SAFE_DIR. To correct, point SAFE_DIR to an
+existing directory.
+
+If you type:
+
+  safe.rb -h
+  
+the available options display. They are:
+
+Usage: safe.rb [options]
+
+Options:
+    -a, --add NAME                   Add an entry
+    -c, --count                      Print the count of entries
+    -d, --delete NAME                Delete an entry
+    -i, --import FILE                Import a file in <name,ID,password> format
+    -l, --list NAME                  List an entry
+    -v, --version                    Print version
+    -p, --password                   Change password
+    
+For example, to add an entry to your password file for Rubyforge.org, you type:
+
+  safe.rb -a Rubyforge.org
+  
+You will be prompted for your master password, and then your Rubyforge.org
+user ID and password, like this:
+
+  Password: ****
+  Rubyforge.org User ID: myuserid
+  Rubyforge.org Password: ****
+
+NOTE: If NAME contains spaces, you must surround it in quotes, like this:
+
+  safe.rb -a "A Cool Site"
+  
+To list your Rubyforge.org password, you type:
+
+  safe.rb -l Rubyforge.org
+  
+You will be prompted for your master password, and then your Rubyforge.org
+user ID and password will display, like this:
+
+  Rubyforge.org     myuserid        mypassword
+
+NOTE: The NAME you pass to -l is not case sensitive, and will find all entries
+beginning with NAME. For example, typing:
+
+  safe.rb -l r
+  
+will list all entries beginning with "r" or "R."
+
+To delete your Rubyforge.org password, you type:
+
+  safe.rb -d Rubyforge.org
+  
+You will be prompted for your master password, and then your Rubyforge.org
+password is irretrievably deleted. You will not be asked to confirm the
+deletion.
+
+== Backups ==
+Please back up your .safe.xml file! See License section for disclaimer
+information.
+
+== Contact ==
+You can contact me, Rob Warner, at rwarner@grailbox.com.
+

data/bin/safe.rb

@@ -0,0 +1,110 @@
+#!/usr/bin/env ruby
+
+########################################################################
+# safe: Safe Storage for Passwords.
+# Stores user IDs and passwords, encrypted by a password.
+#
+# Copyright (c) 2007, Rob Warner
+#
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+#   * Redistributions of source code must retain the above copyright notice,
+#     this list of conditions and the following disclaimer.
+#   * Redistributions in binary form must reproduce the above copyright notice,
+#     this list of conditions and the following disclaimer in the documentation
+#     and/or other materials provided with the distribution.
+#   * Neither the name of Rob Warner nor the names of its contributors may be
+#     used to endorse or promote products derived from this software without
+#     specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
+# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+########################################################################
+require 'rubygems'
+require 'optparse'
+require File.join(File.dirname(__FILE__), '..', 'lib', 'safeentry')
+require File.join(File.dirname(__FILE__), '..', 'lib', 'safefile')
+require File.join(File.dirname(__FILE__), '..', 'lib', 'safeutils')
+
+# TODO support list <NAME> without -l flag
+
+class Safe
+  SAFE_VERSION = '0.2'
+  
+  # Gets the command line
+  def self.get_command_line
+    options = {}
+    options[:action] = 'list'
+    opts = OptionParser.new
+    opts.banner = 'Usage: safe.rb [options]'
+    opts.separator ''
+    opts.separator 'Options:'
+    opts.on('-a', '--add NAME', 'Add an entry') do |val|
+      options[:name] = val
+      options[:action] = 'add'
+    end
+    opts.on('-c', '--count', 'Print the count of entries') do
+      options[:action] = 'count'
+    end
+    opts.on('-d', '--delete NAME', 'Delete an entry') do |val|
+      options[:name] = val
+      options[:action] = 'delete'
+    end
+    opts.on('-i', '--import FILE', 'Import a file in <name,ID,password> format') do |val|
+      options[:name] = val
+      options[:action] = 'import'
+    end
+    opts.on('-l', '--list NAME', 'List an entry') do |val|
+      options[:name] = val
+      options[:action] = 'list'
+    end
+    opts.on('-v', '--version', 'Print version') do
+      options[:action] = 'version'
+    end
+    opts.on('-p', '--password', 'Change password') do
+      options[:action] = 'password'
+    end
+    opts.parse! rescue abort [$!.message, opts].join("\n")
+    options
+  end
+  
+  # Main
+  def self.run
+    options = get_command_line
+    if options[:action] == 'version'
+      puts "Safe version #{SAFE_VERSION}, Copyright (C) 2007 Rob Warner"
+    else
+      begin
+        dir = SafeUtils::get_safe_dir
+      rescue
+        abort $!.message
+      end
+      password = SafeUtils::get_password
+      file = SafeFile.new(password, dir)
+    
+    # Perform the requested action
+      case options[:action]
+        when 'list': file.list(options[:name])
+        when 'add': file.add(options[:name])
+        when 'delete': file.delete(options[:name])
+        when 'count': puts file.count
+        when 'password': file.change_password
+        when 'import': SafeUtils::import(file, options[:name])
+      end
+    end
+  end
+end
+
+Safe.run

data/lib/safeentry.rb

@@ -0,0 +1,53 @@
+########################################################################
+# safeentry.rb
+# Copyright (c) 2007, Rob Warner
+#
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+#   * Redistributions of source code must retain the above copyright notice,
+#     this list of conditions and the following disclaimer.
+#   * Redistributions in binary form must reproduce the above copyright notice,
+#     this list of conditions and the following disclaimer in the documentation
+#     and/or other materials provided with the distribution.
+#   * Neither the name of Rob Warner nor the names of its contributors may be
+#     used to endorse or promote products derived from this software without
+#     specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
+# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+########################################################################
+
+##
+# SafeEntry
+# A single entry (name, ID, password) in the "safe"
+##
+class SafeEntry
+  attr_accessor :name, :id, :password
+
+  def initialize(name, id, password)
+    @name = name
+    @id = id
+    @password = password
+  end
+  
+  def to_s
+    "#@name\t#@id\t#@password"
+  end
+  
+  def <=>(other)
+    self.name.upcase <=> other.name.upcase
+  end
+end
+

data/lib/safefile.rb

@@ -0,0 +1,212 @@
+########################################################################
+# safefile.rb
+# Copyright (c) 2007, Rob Warner
+#
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+#   * Redistributions of source code must retain the above copyright notice,
+#     this list of conditions and the following disclaimer.
+#   * Redistributions in binary form must reproduce the above copyright notice,
+#     this list of conditions and the following disclaimer in the documentation
+#     and/or other materials provided with the distribution.
+#   * Neither the name of Rob Warner nor the names of its contributors may be
+#     used to endorse or promote products derived from this software without
+#     specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
+# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+########################################################################
+require 'rubygems'
+require 'crypt/blowfish'
+require 'digest/sha2'
+require 'rexml/document'
+require 'fileutils'
+require 'base64'
+
+include REXML
+
+##
+# SafeFile
+# The file containing the safe entries
+##
+class SafeFile
+  attr_accessor :hash, :salt, :password, :filename, :entries
+  
+  def initialize(password, dir, filename = '.safe.xml')
+    @entries = Hash.new
+    @filename = "#{dir}/#{filename}"
+    @password = password
+    load
+  end
+
+  # Loads the file
+  def load
+    f = File.open(@filename, File::CREAT)
+    begin
+      if f.lstat.size == 0
+          create_file(f)
+      else
+          decrypt_file(f)
+      end
+      rescue
+        # TODO Determine whether error is can't create file, and print appropriate error
+        puts $!
+      ensure
+        f.close unless f.nil?
+    end
+  end
+
+  # Decrypts the file
+  def decrypt_file(file)
+    begin
+      doc = Document.new file
+      root = doc.root
+      @hash = root.elements["hash"].text
+      @salt = root.elements["salt"].text
+      if authorized?
+        crypt_key = Crypt::Blowfish.new(@password)
+        e_entries = root.elements["entries"]
+        e_entries.elements.each("entry") do |entry|
+          fields = crypt_key.decrypt_string(Base64::decode64(entry.cdatas[0].to_s)).split("\t")
+          if fields.length == 3
+            insert(fields[0], fields[1], fields[2])
+          else
+            puts "Cannot parse #{fields}, discarding."
+          end
+        end
+      else
+        abort('The password you entered is not valid')
+      end
+    rescue ParseException
+      puts "Cannot parse #{file.path}"
+    end
+  end
+
+  # Creates a new file
+  def create_file(file)
+    puts 'Creating new file . . .'
+    generate_hash_and_salt
+    save
+  end
+
+  # Saves the file
+  def save
+    crypt_key = Crypt::Blowfish.new(@password)
+    doc = Document.new
+    root = Element.new "safe"
+    doc.add_element root
+
+    root.add_element create_element("hash", @hash)
+    root.add_element create_element("salt", @salt)
+    
+    e_entries = Element.new "entries"
+    @entries.each_value do |entry|
+      e = Element.new "entry"
+      CData.new Base64.encode64(crypt_key.encrypt_string(entry.to_s)), true, e
+      e_entries.add_element e
+    end
+    root.add_element e_entries
+    
+    f = File.new(@filename, 'w')
+    doc.write f, 0
+    f.close
+  end
+
+  # Adds an entry
+  def add(name)
+    if can_insert? name
+      print "#{name} User ID: "
+      id = gets.chomp!
+      pw = SafeUtils::get_password("#{name} Password: ")
+      insert(name, id, pw)
+      save
+    end
+  end
+  
+  # Inserts an entry
+  def insert(name, id, pw)
+    @entries.delete(name)
+    @entries[name] = SafeEntry.new(name, id, pw)
+  end
+  
+  # Determines whether we can insert this entry--if it exists, we prompt for the OK
+  def can_insert?(name)
+    proceed = true
+    if @entries.has_key?(name)
+      print "Overwrite existing #{name} (y/N)? "
+      proceed = (gets.chomp == 'y')
+    end
+    proceed
+  end
+
+  # Deletes an entry
+  def delete(name)
+    if @entries.has_key?(name)
+      @entries.delete(name)
+      save
+    else
+      puts "Entry #{name} not found"
+    end
+  end
+
+  # Lists the desired entries
+  def list(name)
+    output = Array.new
+    @entries.each_value do |entry|
+      # TODO Inconsistent case handling
+      if name == nil || entry.name.upcase =~ /^#{name.upcase}/
+        output << entry
+      end
+    end
+    output.sort.each do |entry|
+      puts entry
+    end
+  end
+
+  # Returns the count of entries in the file
+  def count()
+    @entries.length
+  end
+  
+  # Changes the file's password
+  def change_password
+    new_password = SafeUtils::get_password('Enter new password: ')
+    rpt_password = SafeUtils::get_password('Confirm password: ')
+    if new_password == rpt_password
+      @password = new_password
+      generate_hash_and_salt
+      save
+    else
+      puts 'Passwords do not match'
+    end
+  end
+
+  def generate_hash_and_salt
+    @salt = [Array.new(20){rand(256).chr}.join].pack('m').chomp
+    @hash = Digest::SHA256.hexdigest(@password + @salt)
+  end
+  
+  # Returns whether the password is authorized
+  def authorized?
+    Digest::SHA256.hexdigest(@password + @salt) == @hash
+  end
+  
+  # Helper method to create an XML element with a name and text
+  def create_element(name, text)
+    e = Element.new name
+    e.text = text
+    e
+  end
+end  

data/lib/safeutils.rb

@@ -0,0 +1,76 @@
+########################################################################
+# safeutils.rb
+# Copyright (c) 2007, Rob Warner
+#
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions are met:
+#
+#   * Redistributions of source code must retain the above copyright notice,
+#     this list of conditions and the following disclaimer.
+#   * Redistributions in binary form must reproduce the above copyright notice,
+#     this list of conditions and the following disclaimer in the documentation
+#     and/or other materials provided with the distribution.
+#   * Neither the name of Rob Warner nor the names of its contributors may be
+#     used to endorse or promote products derived from this software without
+#     specific prior written permission.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
+# CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
+# EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
+# PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
+# PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
+# LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
+# NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+# SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+########################################################################
+require 'highline/import'
+
+##
+# SafeUtils
+# Utility methods for the Safe application
+##
+class SafeUtils
+  # Gets the directory for the data file
+  def SafeUtils.get_safe_dir(var = 'SAFE_DIR')
+    dir = ENV[var]
+    raise "Set environment variable #{var} to the directory for your safe file" unless dir
+    begin
+      d = Dir.new(dir)
+    rescue
+      raise "Environment variable #{var} does not point to an existing directory (#{dir})"
+    end
+    dir
+  end
+  
+  # Prompts the user for a password
+  def SafeUtils.get_password(prompt = 'Password: ', mask = '*')
+    ask(prompt) { |q| q.echo = mask }
+  end
+  
+  # Imports a file in the format:
+  # name,ID,password
+  # name,ID,password
+  # etc.
+  def SafeUtils.import(safe_file, import_file)
+    begin
+      open(import_file).each do |line|
+        puts "Importing #{line}"
+        fields = line.split(',')
+        if fields.length == 3
+          safe_file.insert(fields[0], fields[1], fields[2])
+        else
+          puts "Cannot import; number of fields = #{fields.length}"
+        end
+      end
+      safe_file.save
+    rescue
+      puts $!
+      raise "Problem importing file #{import_file}"
+    end
+  end
+end

data/test/tc_safeentry.rb

@@ -0,0 +1,37 @@
+########################################################################
+# tc_safeentry.rb
+# Copyright (c) 2007 Rob Warner. 
+# All rights reserved. This program and the accompanying materials 
+# are made available under the terms of the Eclipse Public License v1.0 
+# which accompanies this distribution, and is available at 
+# http://www.eclipse.org/legal/epl-v10.html 
+########################################################################
+
+require File.join(File.dirname(__FILE__), '..', 'lib', 'safeentry')
+require 'test/unit'
+
+class SafeDirTest < Test::Unit::TestCase
+  def test_to_s
+    se = SafeEntry.new('name', 'id', 'password')
+    assert_equal "name\tid\tpassword", se.to_s
+
+    se = SafeEntry.new('', '', '')
+    assert_equal "\t\t", se.to_s
+
+    se = SafeEntry.new(' ', ' ', ' ')
+    assert_equal " \t \t ", se.to_s
+  end
+  
+  def test_sort
+    test = Array.new
+    test << SafeEntry.new('apple', 'baker', 'charlie')
+    test << SafeEntry.new('zoo', 'yacht', 'x-ray')
+    test << SafeEntry.new('aardvark', 'kangaroo', 'wallaby')
+    test << SafeEntry.new('123', '456', '789')
+    test.sort!
+    assert_equal '123', test[0].name
+    assert_equal 'aardvark', test[1].name
+    assert_equal 'apple', test[2].name
+    assert_equal 'zoo', test[3].name
+  end
+end

data/test/tc_safefile.rb

@@ -0,0 +1,120 @@
+########################################################################
+# tc_safefile.rb
+# Copyright (c) 2007 Rob Warner. 
+# All rights reserved. This program and the accompanying materials 
+# are made available under the terms of the Eclipse Public License v1.0 
+# which accompanies this distribution, and is available at 
+# http://www.eclipse.org/legal/epl-v10.html 
+########################################################################
+
+require File.join(File.dirname(__FILE__), '..', 'lib', 'safefile')
+require File.join(File.dirname(__FILE__), '..', 'lib', 'safeentry')
+require 'test/unit'
+require 'fileutils'
+
+class SafeFileTest < Test::Unit::TestCase
+  def test_insert_and_delete
+    f = SafeFile.new("password", ".")
+    assert_equal "./.safe.xml", f.filename
+    assert_equal 0, f.entries.length
+    
+    f.insert('apple', 'baker', 'charlie')
+    assert_equal 1, f.entries.length
+
+    f.insert('apple', 'charlie', 'baker')
+    assert_equal 1, f.entries.length
+
+    f.insert('foo', 'bar', 'baz')
+    assert_equal 2, f.entries.length
+    
+    f.delete('bar')
+    assert_equal 2, f.entries.length
+    
+    f.delete('foo')
+    assert_equal 1, f.entries.length
+    
+    f.delete('apple')
+    assert_equal 0, f.entries.length
+    
+    f.delete('apple')
+    assert_equal 0, f.entries.length
+
+    FileUtils::rm("./.safe.xml")
+  end
+  
+  def test_entry
+    f = SafeFile.new("password", ".")
+    f.insert('apple', 'baker', 'charlie')
+    e = f.entries['apple']
+    assert_equal 'apple', e.name
+    assert_equal 'baker', e.id
+    assert_equal 'charlie', e.password
+    assert_equal "apple\tbaker\tcharlie", e.to_s
+
+    FileUtils::rm("./.safe.xml")
+  end
+
+  def test_modify_entry
+    f = SafeFile.new("password", ".")
+    f.insert('apple', 'baker', 'charlie')
+    e = f.entries['apple']
+    assert_equal 'apple', e.name
+    assert_equal 'baker', e.id
+    assert_equal 'charlie', e.password
+    assert_equal "apple\tbaker\tcharlie", e.to_s
+    
+    f.insert('apple', 'foo', 'bar')
+    e = f.entries['apple']
+    assert_equal 'apple', e.name
+    assert_equal 'foo', e.id
+    assert_equal 'bar', e.password
+    assert_equal "apple\tfoo\tbar", e.to_s
+    
+    FileUtils::rm("./.safe.xml")
+  end
+    
+  def test_authorized
+    f = SafeFile.new("password", ".")
+    assert_equal true, f.authorized?
+    
+    f.password = "foo"
+    assert_equal false, f.authorized?
+
+    FileUtils::rm("./.safe.xml")
+  end
+  
+  def test_count
+    f = SafeFile.new("password", ".")
+    
+    assert_equal 0, f.count
+    
+    f.insert('apple', 'baker', 'charlie')
+    assert_equal 1, f.count
+    
+    f.insert('apple1', 'baker', 'charlie')
+    assert_equal 2, f.count
+
+    f.insert('apple1', 'baker', 'charlie')
+    assert_equal 2, f.count
+    
+    f.insert('apple2', 'baker', 'charlie')
+    assert_equal 3, f.count
+    
+    f.delete('apple')
+    assert_equal 2, f.count
+    
+    f.delete('apple')
+    assert_equal 2, f.count
+
+    f.delete('apple1')
+    assert_equal 1, f.count
+
+    f.delete('apple2')
+    assert_equal 0, f.count
+
+    f.delete('apple2')
+    assert_equal 0, f.count
+
+    FileUtils::rm("./.safe.xml")
+  end
+end

data/test/tc_safeutils.rb

@@ -0,0 +1,45 @@
+########################################################################
+# tc_safeutils.rb
+# Copyright (c) 2007 Rob Warner. 
+# All rights reserved. This program and the accompanying materials 
+# are made available under the terms of the Eclipse Public License v1.0 
+# which accompanies this distribution, and is available at 
+# http://www.eclipse.org/legal/epl-v10.html 
+########################################################################
+
+require File.join(File.dirname(__FILE__), '..', 'lib', 'safeutils')
+require 'test/unit'
+require 'fileutils'
+
+class SafeUtilsTest < Test::Unit::TestCase
+  def test_get_safe_dir
+    ev = 'MY_SAFE_ENV_VAR'
+    
+    # Make sure we're not clobbering an environment variable
+    temp = ENV[ev]
+    assert_equal nil, temp
+    
+    # Test when the variable isn't set
+    begin
+      dir = SafeUtils::get_safe_dir(ev)
+    rescue
+      assert_equal "Set environment variable #{ev} to the directory for your safe file", $!.message
+    end
+
+    # Test when the directory doesn't exist
+    ENV[ev] = 'Does not exist'
+    begin
+      dir = SafeUtils::get_safe_dir(ev)
+    rescue
+      assert_equal "Environment variable #{ev} does not point to an existing directory (#{ENV[ev]})", $!.message
+    end
+
+    # Test when the directory exists
+    ENV[ev] = '.'
+    dir = SafeUtils::get_safe_dir(ev)
+    assert_equal '.', dir
+    
+    # Make sure the directory exists
+    FileUtils::cd dir
+  end
+end

metadata

@@ -0,0 +1,68 @@
+--- !ruby/object:Gem::Specification 
+rubygems_version: 0.9.4
+specification_version: 1
+name: safe
+version: !ruby/object:Gem::Version 
+  version: "0.2"
+date: 2007-08-19 00:00:00 -04:00
+summary: A command-line password storage program
+require_paths: 
+- lib
+email: rwarner@grailbox.com
+homepage: http://grailbox.com/safe
+rubyforge_project: 
+description: safe safely stores all your user IDs and passwords, encrypted by a password of your choosing.
+autorequire: 
+default_executable: 
+bindir: bin
+has_rdoc: false
+required_ruby_version: !ruby/object:Gem::Version::Requirement 
+  requirements: 
+  - - ">"
+    - !ruby/object:Gem::Version 
+      version: 0.0.0
+  version: 
+platform: ruby
+signing_key: 
+cert_chain: 
+post_install_message: 
+authors: 
+- Rob Warner
+files: 
+- README
+- lib/safeentry.rb
+- lib/safefile.rb
+- lib/safeutils.rb
+test_files: 
+- test/tc_safefile.rb
+- test/tc_safeentry.rb
+- test/tc_safeutils.rb
+rdoc_options: []
+
+extra_rdoc_files: []
+
+executables: 
+- safe.rb
+extensions: []
+
+requirements: []
+
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: crypt
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Version::Requirement 
+    requirements: 
+    - - ">"
+      - !ruby/object:Gem::Version 
+        version: 0.0.0
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: highline
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Version::Requirement 
+    requirements: 
+    - - ">"
+      - !ruby/object:Gem::Version 
+        version: 0.0.0
+    version: