rubeepass 0.1.0

data/lib/rubeepass/error/invalid_gzip_error.rb

@@ -0,0 +1,7 @@
+require "rubeepass/error"
+
+class RubeePass::Error::InvalidGzipError < RubeePass::Error
+    def initialize
+        super("Invalid gzip format!")
+    end
+end

data/lib/rubeepass/error/invalid_header_error.rb

@@ -0,0 +1,7 @@
+require "rubeepass/error"
+
+class RubeePass::Error::InvalidHeaderError < RubeePass::Error
+    def initialize
+        super("Invalid header format!")
+    end
+end

data/lib/rubeepass/error/invalid_magic_error.rb

@@ -0,0 +1,7 @@
+require "rubeepass/error"
+
+class RubeePass::Error::InvalidMagicError < RubeePass::Error
+    def initialize
+        super("Invalid magic values detected!")
+    end
+end

data/lib/rubeepass/error/invalid_password_error.rb

@@ -0,0 +1,7 @@
+require "rubeepass/error"
+
+class RubeePass::Error::InvalidPasswordError < RubeePass::Error
+    def initialize
+        super("Invalid password provided!")
+    end
+end

data/lib/rubeepass/error/invalid_protected_data_error.rb

@@ -0,0 +1,7 @@
+require "rubeepass/error"
+
+class RubeePass::Error::InvalidProtectedDataError < RubeePass::Error
+    def initialize
+        super("Invalid protected data!")
+    end
+end

data/lib/rubeepass/error/invalid_protected_stream_key_error.rb

@@ -0,0 +1,7 @@
+require "rubeepass/error"
+
+class RubeePass::Error::InvalidProtectedStreamKeyError < RubeePass::Error
+    def initialize
+        super("Invalid protected stream key!")
+    end
+end

data/lib/rubeepass/error/invalid_version_error.rb

@@ -0,0 +1,7 @@
+require "rubeepass/error"
+
+class RubeePass::Error::InvalidVersionError < RubeePass::Error
+    def initialize
+        super("Invalid version detected!")
+    end
+end

data/lib/rubeepass/error/not_aes_error.rb

@@ -0,0 +1,7 @@
+require "rubeepass/error"
+
+class RubeePass::Error::NotAESError < RubeePass::Error
+    def initialize
+        super("Not AES!")
+    end
+end

data/lib/rubeepass/error/not_salsa20_error.rb

@@ -0,0 +1,7 @@
+require "rubeepass/error"
+
+class RubeePass::Error::NotSalsa20Error < RubeePass::Error
+    def initialize
+        super("Not a Salsa20 CrsAlgorithm!")
+    end
+end

data/lib/rubeepass/error.rb

@@ -0,0 +1,12 @@
+class RubeePass::Error < RuntimeError
+end
+
+require "rubeepass/error/invalid_gzip_error"
+require "rubeepass/error/invalid_header_error"
+require "rubeepass/error/invalid_magic_error"
+require "rubeepass/error/invalid_password_error"
+require "rubeepass/error/invalid_protected_data_error"
+require "rubeepass/error/invalid_protected_stream_key_error"
+require "rubeepass/error/invalid_version_error"
+require "rubeepass/error/not_aes_error"
+require "rubeepass/error/not_salsa20_error"

data/lib/rubeepass/group.rb

@@ -0,0 +1,133 @@
+require "string"
+
+class RubeePass::Group
+    include Comparable
+
+    attr_accessor :entries
+    attr_accessor :group
+    attr_accessor :groups
+    attr_accessor :keepass
+    attr_accessor :name
+    attr_accessor :path
+    attr_accessor :uuid
+
+    def ==(other)
+        return (self.uuid == other.uuid)
+    end
+
+    def <=>(other)
+        return (self.name.downcase <=> other.name.downcase)
+    end
+
+    def details(level = 0)
+        out = Array.new
+        lvl = Array.new(level, "  ").join
+
+        group_details = [ "#{@name}".blue ]
+        group_details[0] = "#{@path}".blue if (level == 0)
+
+        group_details.each do |line|
+            out.push("#{lvl}#{line}")
+        end
+
+        @groups.values.each do |group|
+            out.push(group.details(level + 1))
+        end
+
+        @entries.values.each do |entry|
+            out.push(entry.details(level + 1))
+        end
+
+        return out.join("\n")
+    end
+
+    def entry_titles
+        return @entries.keys.sort
+    end
+
+    def find_group(path)
+        return nil if (@keepass.nil?)
+
+        path = @keepass.absolute_path(path, @path)
+        cwd = @keepass.db
+
+        path.split("/").each do |group|
+            next if (group.empty?)
+            if (cwd.has_group?(group))
+                cwd = cwd.groups[group]
+            else
+                return nil
+            end
+        end
+
+        return cwd
+    end
+
+    def fuzzy_find(input)
+        return [ [], [], [] ] if (@keepass.nil?)
+
+        input = @path if (input.nil? || input.empty?)
+        input = @keepass.absolute_path(input, @path)
+        path, target = input.rsplit("/")
+
+        new_cwd = find_group(path)
+        return [ input, [], [] ] if (new_cwd.nil?)
+
+        if (new_cwd.has_group?(target))
+            new_cwd = new_cwd.groups[target]
+            target = ""
+            input += "/"
+        end
+
+        group_completions = new_cwd.group_names
+        entry_completions = new_cwd.entry_titles
+
+        if (target.empty?)
+            return [ input, group_completions, entry_completions ]
+        end
+
+        group_completions.delete_if do |group|
+            !group.downcase.start_with?(target.downcase)
+        end
+        entry_completions.delete_if do |entry|
+            !entry.downcase.start_with?(target.downcase)
+        end
+
+        return [ input, group_completions, entry_completions ]
+    end
+
+    def group_names
+        return @groups.keys.sort
+    end
+
+    def has_entry?(title)
+        entry_titles.each do |entry|
+            return true if (title.downcase == entry.downcase)
+        end
+        return false
+    end
+
+    def has_group?(name)
+        group_names.each do |group|
+            return true if (name.downcase == group.downcase)
+        end
+        return false
+    end
+
+    def initialize(params)
+        @entries = Hash.new
+        @group = params.fetch("Group", nil)
+        @groups = Hash.new
+        @keepass = params.fetch("Keepass", nil)
+        @name = params.fetch("Name", "")
+        @uuid = params.fetch("UUID", "")
+
+        @path = @name
+        @path = "#{@group.path}/#{@name}" if (@group)
+        @path.gsub!(%r{^//}, "/")
+    end
+
+    def to_s
+        return details
+    end
+end

data/lib/rubeepass/protected_decryptor.rb

@@ -0,0 +1,30 @@
+require "salsa20"
+
+class RubeePass::ProtectedDecryptor
+    def add_to_stream(str)
+        @ciphertext.push(str)
+        return (@ciphertext.length - 1)
+    end
+
+    def get_password(index)
+        return nil if (@iv.nil? || @key.nil?)
+        return nil if ((index < 0) || (index >= @ciphertext.length))
+
+        plaintext = Salsa20.new(@key, @iv).decrypt(@ciphertext.join)
+
+        start = 0
+        index.times do |i|
+            start += @ciphertext[i].length
+        end
+
+        stop = start + @ciphertext[index].length
+
+        return plaintext[start...stop]
+    end
+
+    def initialize(key, iv)
+        @ciphertext = Array.new
+        @iv = iv
+        @key = key
+    end
+end