#!/usr/bin/perl ########################################################################### # # kpcli - KeePass Command Line Interface # # Author: Lester Hightower # # This program was inspired by "kedpm -c" and resulted despite illness # (or more likely because of it) over the USA Thanksgiving holiday in # late November of 2010. As a long-time user of the Ked Password Manager # I really missed a command line interface after getting an Android cell # phone and switching to KeePass, so that I could access my password # database on my phone. This program scratches that itch. # ########################################################################### use strict; # core use version; # core use Clone; # core use FileHandle; # core use Getopt::Long; # core use File::Basename; # core use Digest::file; # core use Digest::SHA qw(sha256); # core use Data::Dumper qw(Dumper); # core use Term::ANSIColor; # core use Crypt::Rijndael; # non-core, libcrypt-rijndael-perl on Ubuntu use Sort::Naturally; # non-core, libsort-naturally-perl on Ubuntu use Term::ReadKey; # non-core, libterm-readkey-perl on Ubuntu use Term::ShellUI; # non-core, libterm-shellui-perl on Ubuntu # - add Term::ReadLine::Gnu for cli history use File::KeePass 0.03; # non-core, libfile-keepass-perl on Ubuntu # - >=v0.03 needed due critical bug fixes $|=1; my $DEBUG=0; my $DEFAULT_ENTRY_ICON = 0; # In keepassx, icon 0 is a golden key my $DEfAULT_GROUP_ICON = 49; # In keepassx, icon 49 is an opened file folder my $FOUND_DIR = '_found'; # The find command's results go in /_found/ my $APP_NAME = basename($0); $APP_NAME =~ s/\.pl$//; my $VERSION = "1.7"; our $HISTORY_FILE = ""; # Gets set in the MyGetOpts() function my $opts=MyGetOpts(); # Will only return with options we think we can use # Setup our Term::ShellUI object my $term = new Term::ShellUI( app => $APP_NAME, history_file => $HISTORY_FILE, keep_quotes => 0, commands => { #"" => { args => sub { shift->complete_history(@_) } }, "history" => { desc => "Prints the command history", doc => "\nSpecify a number to list the last N lines of history" . "Pass -c to clear the command history, " . "-d NUM to delete a single item\n", args => "[-c] [-d] [number]", method => sub { shift->history_call(@_) }, exclude_from_history => 1, }, "version" => { desc => "Print the version of this program", method => sub { print "$VERSION\n"; }, exclude_from_history => 1, }, "ver" => { alias => "version", exclude_from_completion=>1, exclude_from_history => 1,}, "help" => { desc => "Print helpful information", args => sub { shift->help_args(undef, @_); }, method => sub { my_help_call(@_); }, exclude_from_history => 1, #method => sub { shift->help_call(undef, @_); } }, "h" => { alias => "help", exclude_from_completion=>1, exclude_from_history => 1,}, "?" => { alias => "help", exclude_from_completion=>1, exclude_from_history => 1,}, "cl" => { desc => "Change directory and list entries (cd+ls)", doc => "\n" . "Change the pwd to an absolute or relative path\n" . "and list the entries there. This is a useful way\n" . "to quickly navigate to a path and have the entries\n" . "listed in preparation to run the show command.\n", maxargs => 1, args => \&complete_groups, method => sub { if(cli_cd(@_) == 0) { cli_ls() } }, }, "cls" => { desc => 'Clear screen ("clear" command also works)', doc => "\n" . "Clear the screen, which is useful when guests arrive.\n", maxargs => 0, method => sub { print "\033[2J\033[0;0H"; }, exclude_from_history => 1, }, "clear" => { alias => "cls", exclude_from_history => 1, }, "cd" => { desc => "Change directory (path to a group)", doc => "\n" . "Change the pwd to an absolute or relative path.\n" . "Slashes in names are escaped with backslashes:\n" . "(i.e. \"cd /personal/Comcast\\/Xfinity\").\n", maxargs => 1, args => \&complete_groups, method => \&cli_cd, }, "chdir" => { alias => 'cd' }, "saveas" => { desc => "Save to a specific filename " . "(saveas [])", minargs => 1, maxargs => 2, args => [\&Term::ShellUI::complete_files, \&Term::ShellUI::complete_files], proc => \&cli_saveas, }, "export" => { desc => "Export entries to a new KeePass DB " . "(export [])", doc => "\n" . "Use this command to export the full tree of groups\n" . "and entries to another KeePass database file on disk,\n" . "starting at your current path (pwd).\n" . "\n" . "This is also a \"safer\" way to change your database\n" . "password. Export from /, verify that the new file is\n" . "good, and then remove your original file.\n", minargs => 1, maxargs => 2, args => [\&Term::ShellUI::complete_files, \&Term::ShellUI::complete_files], proc => \&cli_export, }, "import" => { desc => "Import another KeePass DB " . "(import [])", doc => "\n" . "Use this command to import the entire KeePass DB\n" . "specified by into a new group at .\n", minargs => 2, maxargs => 3, args => [\&Term::ShellUI::complete_files,\&complete_groups, \&Term::ShellUI::complete_files], proc => \&cli_import, }, "open" => { desc => "Open a KeePass database file " . "(open [])", minargs => 1, maxargs => 2, args => [\&Term::ShellUI::complete_files, \&Term::ShellUI::complete_files], proc => \&cli_open, }, "mkdir" => { desc => "Create a new group (mkdir )", minargs => 1, maxargs => 1, args => \&complete_groups, method => \&cli_mkdir, }, "rmdir" => { desc => "Delete a group (rmdir )", minargs => 1, maxargs => 1, args => \&complete_groups, method => \&cli_rmdir, }, "ls" => { desc => "Lists entries in pwd or in the specified path", minargs => 0, maxargs => 1, args => \&complete_groups, method => \&cli_ls, }, "new" => { desc => "Create a new entry in the current group (pwd)", minargs => 0, maxargs => 0, args => "", method => \&cli_new, }, "rm" => { desc => "Remove an entry: rm ", minargs => 1, maxargs => 1, args => \&complete_groups_and_entries, method => \&cli_rm, }, "show" => { desc => "Show an entry: show [-f] [-a] ", doc => "\n" . "The show command tries to intelligently determine\n" . "what you want to see and to make it easy to display.\n" . "Show can take a path to an entry as its argument or\n" . "an entry number as shown by the ls command.\n" . "\n" . "When using entry numbers, they will refer to the last\n" . "path when an ls was performed or pwd if ls has not\n" . "yet been run.\n" . "\n" . "By default, passwords are \"hidden\" by being displayed as\n" . "\"red on red\" where they can be copied to the clip board\n" . "but not seen. Provide the -f option to show passwords.\n" . "Use the -a option to see create and modified times, and\n" . "the index of the icon set for the entry.\n" . "", minargs => 1, maxargs => 3, args => \&complete_groups_and_entries, method => \&cli_show, }, "edit" => { desc => "Edit an entry: edit ", minargs => 1, maxargs => 1, args => \&complete_groups_and_entries, method => \&cli_edit, }, "mv" => { desc => "Move an entry: mv ", minargs => 2, maxargs => 2, args => [\&complete_groups_and_entries, \&complete_groups], method => \&cli_mv, }, "rename" => { desc => "Rename a group: rename ", minargs => 1, maxargs => 1, args => \&complete_groups, method => \&cli_rename, }, "save" => { desc => "Save the database to disk", minargs => 0, maxargs => 0, args => "", method => \&cli_save, }, "close" => { desc => "Close the currently opened database", minargs => 0, maxargs => 0, args => "", method => \&cli_close, }, "find" => { desc => "Finds entries by Title", doc => "\n" . "Searches for entries with the given search term\n" . "in their title and places matches into \"/$FOUND_DIR/\".\n", minargs => 1, maxargs => 1, args => "", method => \&cli_find, }, "pwd" => { desc => "Print the current working directory", maxargs => 0, proc => \&cli_pwd, }, "icons" => { desc => "Change group or entry icons in the database", maxargs => 0, proc => \&cli_icons, }, "quit" => { desc => "Quit this program (EOF and exit also work)", maxargs => 0, method => \&cli_quit, exclude_from_history => 1, }, "exit" => { alias => "quit", exclude_from_history => 1,} }, ); $term->prompt(\&term_set_prompt); # Seed our state global variable our $state={ 'appname' => $APP_NAME, 'term' => $term, 'kdb_has_changed' => 0, 'last_ls_path' => '', 'put_master_passwd' => \&put_master_passwd, 'get_master_passwd' => \&get_master_passwd, }; # If given --kdb=, open that file if (length($opts->{kdb})) { my $err = open_kdb($opts->{kdb}, $opts->{key}); # Sets $state->{'kdb'} if (length($err)) { print "Error opening file: $err\n"; } } else { new_kdb($state); } # Enter the interative kpcli shell session print "\n" . "KeePass CLI ($APP_NAME) v$VERSION is ready for operation.\n" . "Type 'help' for a description of available commands.\n" . "Type 'help ' for details on individual commands.\n"; if ($DEBUG) {print 'Using '.$term->{term}->ReadLine." for readline.\n"; } if (! $DEBUG && $term->{term}->ReadLine ne 'Term::ReadLine::Gnu') { warn "* Please install Term::ReadLine::Gnu for better functionality!\n"; } # My patch made it into Term::ShellUI v0.9, but I still chose not to make # this script demand >=0.9 and instead look for the add_eof_exit_hook() and # use it if it exists and warn if not. if (Term::ShellUI->can('add_eof_exit_hook')) { $term->add_eof_exit_hook(\&eof_exit_hook); } else { warn "* Please upgrade Term::ShellUI to version 0.9 or newer.\n"; } print "\n"; $term->run(); exit; ############################################################################ ############################################################################ ############################################################################ sub open_kdb($$) { my $file=shift @_; my $key_file=shift @_; our $state; # Make sure the file exists and is readable if (! -f $file) { return "File does not exist: $file"; } if (! -r $file) { return "File is not readable: $file"; } # Look for lock file and warn if it is found my $lock_file = $file . '.lock'; # KeePassX style if (-f $lock_file) { my $bold="\e[1m"; my $red="\e[31m"; my $yellow="\e[33m"; my $clear="\e[0m"; print $bold . $yellow . "WARNING:" . $clear . $red . " A KeePassX-style lock file is in place for this file.\n" . " It may be opened elsewhere." . "$yellow$bold Be careful of saving!\n" . $clear; } else { $state->{placed_lock_file} = $lock_file; } # Ask the user for the master password and then open the kdb my $master_pass=GetMasterPasswd(); $state->{kdb} = File::KeePass->new; if (! eval { $state->{kdb}->load_db($file, composite_master_pass($master_pass, $key_file)) }) { die "Couldn't load the file $file: $@"; } if ($state->{placed_lock_file}) { touch_file($state->{placed_lock_file}); } $state->{kdb_file} = $file; $state->{key_file} = $key_file; $state->{put_master_passwd}($master_pass); $state->{kdb_has_changed}=0; $master_pass="\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0"; # Build the %all_grp_paths_fwd and %all_grp_paths_rev structures refresh_state_all_paths(); # Store the md5sum of the file so we can watch for unexpected changes $state->{kdb_file_md5} = Digest::file::digest_file_hex($file, "MD5"); # Initialize our state into "/" cli_cd($term, {'args' => ["/"]}); return ''; # If we return anything else it is an error message } # Called by Term::ShellUI to get the user prompt sub term_set_prompt($$) { my $term=shift @_; my $raw_cmd=shift @_; our $state; my $pwd=$state->{all_grp_paths_rev}->{$state->{path}->{id}}; $pwd =~ s%/%\\/%g; $pwd =~ s/\0/\//g; my $app=$state->{appname}; my $pwd=get_pwd(); return "$app:$pwd> "; } # Walks through a tree of groups building a flat hash of NULL-separated # paths to group IDs. Called on the root to build a full path tree. sub build_all_group_paths { my $hash = shift @_; my $g = shift @_; my $root_path = shift @_ || []; foreach my $me (@{$g}) { my @path_to_me = @{$root_path}; push @path_to_me, $me->{title}; my $path=join("\0",@path_to_me); $hash->{$path}=$me->{id}; if (defined($me->{groups})) { build_all_group_paths($hash,$me->{groups},\@path_to_me); } } return (scalar(keys(%{$hash}))); } # Walks through a tree of groups building a flat hash of NULL-separated # paths to entry IDs. Called on the root to build a full path tree. sub build_all_entry_paths { my $hash = shift @_; my $g = shift @_; my $root_path = shift @_ || []; foreach my $me (@{$g}) { my @path_to_me = @{$root_path}; push @path_to_me, $me->{title}; if (defined($me->{entries})) { foreach my $ent (@{$me->{entries}}) { my $path=join( "\0", (@path_to_me, $ent->{title}) ); $hash->{$path}=$ent->{id}; } } if (defined($me->{groups})) { build_all_entry_paths($hash,$me->{groups},\@path_to_me); } } return (scalar(keys(%{$hash}))); } # Returns the current path the user is sitting in. sub get_pwd { my $pwd=''; if (defined($state->{all_grp_paths_rev}->{$state->{path}->{id}})) { $pwd=$state->{all_grp_paths_rev}->{$state->{path}->{id}}; } $pwd =~ s%/%\\/%g; $pwd =~ s/\0/\//g; $pwd = '/' . $pwd; return $pwd; } # Destroys our /_found group (where we place search results) sub destroy_found { our $state; # Look for an exising /_found and kill it if it exists my $k=$state->{kdb}; my $found_group=$k->find_group({level=>0,title=>$FOUND_DIR}); if (defined($found_group)) { my @oldents = $k->find_entries({group=>$found_group->{id}}); foreach my $ent (@oldents) { $k->delete_entry({id => $ent->{id}}); } $k->delete_group({level=>0,title=>$FOUND_DIR}); # Because we destroyed /_found we must refresh our $state paths refresh_state_all_paths(); } } # Refreshes $state->{all_grp_paths_fwd} and $state->{all_grp_paths_rev} sub refresh_state_all_paths() { our $state; my %all_grp_paths_fwd; build_all_group_paths(\%all_grp_paths_fwd,$state->{kdb}->groups); my %all_grp_paths_rev = reverse %all_grp_paths_fwd; $state->{all_grp_paths_fwd}=\%all_grp_paths_fwd; $state->{all_grp_paths_rev}=\%all_grp_paths_rev; my %all_ent_paths_fwd; build_all_entry_paths(\%all_ent_paths_fwd,$state->{kdb}->groups); my %all_ent_paths_rev = reverse %all_ent_paths_fwd; $state->{all_ent_paths_fwd}=\%all_ent_paths_fwd; $state->{all_ent_paths_rev}=\%all_ent_paths_rev; } # Gathers the list of groups and entries for the pwd we're sitting in sub get_current_groups_and_entries { return get_groups_and_entries(get_pwd()); } sub get_groups_and_entries { my $path=shift @_; our $state; my $k=$state->{kdb}; my @groups=(); my @entries=(); my $norm_path = normalize_path_string($path); if (length($norm_path) < 1) { @groups = $k->find_groups({level=>0}); @entries = $k->find_entries({level => 0}); } else { my $id=$state->{all_grp_paths_fwd}->{$norm_path}; my ($this_grp,@trash) = $k->find_groups({id=>$id}); if (defined($this_grp->{groups})) { # subgroups @groups = @{$this_grp->{groups}}; } @entries = $k->find_entries({group_id => $id}); } @groups = sort group_sort @groups; @entries = sort { ncmp($a->{title},$b->{title}); } @entries; return (\@groups,\@entries); } # This function takes a group ID and returns all of the child # groups of that group, flattened. sub all_child_groups_flattened($) { my $group_id=shift @_; our $state; my $k=$state->{kdb}; my @groups=(); my ($this_grp,@trash) = $k->find_groups({id=>$group_id}); if (defined($this_grp->{groups})) { # subgroups @groups = @{$this_grp->{groups}}; foreach my $child_group (@groups) { push @groups, all_child_groups_flattened($child_group->{id}); } } return @groups; } # A function to properly sort groups by title sub group_sort($$) { my $a=shift @_; my $b=shift @_; # _found at level 0 is a special case (from our find command). if ($a->{title} eq $FOUND_DIR && $a->{level} == 0) { return 1; } elsif ($b->{title} eq $FOUND_DIR && $b->{level} == 0) { return -1; # Backup at level=0 is a special case (KeePassX's Backup group). } elsif ($a->{title} eq 'Backup' && $a->{level} == 0) { return 1; } elsif ($b->{title} eq 'Backup' && $b->{level} == 0) { return -1; # Sort everything else naturally (Sort::Naturally::ncmp). } else { return ncmp($a->{title},$b->{title}); # Natural sort } } # ------------------------------------------------------------------------- # All of the cli_*() functions are below here # ------------------------------------------------------------------------- sub cli_pwd { print get_pwd() . "\n"; } sub cli_cd { my $self = shift @_; my $params = shift @_; our $state; my $raw_pathstr = $params->{args}->[0]; # "cd ." if ($raw_pathstr =~ m/^[.]$/) { return; # nothing to do } # "cd -" if ($raw_pathstr =~ m/^[-]$/) { return cli_cd($self, {'args' => [$state->{oldpwd}]}); } # Everything else is handled by helpers return cli_cd_helper($state,normalize_path_string($raw_pathstr)); } # Takes a possibly wacky path with ".."s and such and normalizes it into a # NULL-separated path that can be used as a key into $state->{all_grp_paths_fwd} sub normalize_path_string($) { my $path_string = shift @_; our $state; # Split the path into @path # http://efreedom.com/Question/1-3588341/Implement-Escape-Sequence-Using-Split-Perl my $delim="/"; my $escape="\\"; my @path = $path_string =~ /(?:\Q$delim\E|^)((?>(?:\Q$escape\E.|(?!\Q$delim\E).)*))/gs; #/(?:\Q$delim\E|^)((?:\Q$escape\E.|(?!\Q$delim\E).)*+)/gs; # perl 5.10+ s/\Q$escape$delim\E/$delim/g for @path; @path=grep(!/^$/, @path); # Drop meaningless (excess) deimiters (/foo//bar) # This block handles absolute and relative paths my $path_str=''; if ($path_string =~ m%^/%) { # Absolute path $path_str=join("\0",@path); } else { # Relative path my $pwd=$state->{all_grp_paths_rev}->{$state->{path}->{id}}; my @nwd=split("\0", $pwd); push @nwd, @path; $path_str=join("\0", @nwd); } # We should now have a NULL-separated, fully qualified path and # we just need to work on cleaning up things like "." and ".." # Squash single dots while ($path_str=~s/\0[.]\0|^[.]\0|\0[.]$/\0/) {} # Kill ".." and their parents while ($path_str=~s/[^\0]+\0+[.][.]//) {} $path_str=~s/\0\0+/\0/g; # squash any adjacent delimeters $path_str=~s/^\0+//; # squash any leading delimeters $path_str=~s/\0+$//; # squash any trailing delimeters return $path_str; } sub cli_cd_helper($$) { my $state=shift @_; my $path_str=shift @_; if (defined($state->{all_grp_paths_fwd}->{$path_str})) { $state->{oldpwd}=get_pwd(); my $id=$state->{all_grp_paths_fwd}->{$path_str}; $state->{path}->{id}=$id; return 0; } elsif ($path_str eq '') { # cd / $state->{oldpwd}=get_pwd(); delete $state->{path}; return 0; } else { print "Invalid path\n"; return -1; } } sub cli_find($) { my $self = shift @_; my $params = shift @_; our $state; destroy_found(); # Now do the Title search my $k=$state->{kdb}; my $search_str = $params->{args}->[0]; print "Searching for \"$search_str\" ...\n"; # Make $search_str a case-insensitive regex my @letters=split(//, $search_str); foreach my $l (@letters) { if (uc($l) ne lc($l)) { $l='[' . uc($l) . lc($l) . ']'; } } $search_str=join('', @letters); # Search entries by title, skipping the KeePassX /Backup group if it exists my $search_params = { 'title =~' => $search_str }; my $backup_dir_normalized=normalize_path_string("/Backup"); # /Backup if (defined($state->{all_grp_paths_fwd}->{$backup_dir_normalized})) { $search_params->{'group_id !'} = $state->{all_grp_paths_fwd}->{$backup_dir_normalized}; } my @e = $k->find_entries($search_params); if ( scalar(@e) < 1) { print "No matches.\n"; return; } # If we get this far we have results to add to a new /_found my $found_group = $k->add_group({title => $FOUND_DIR}); # root level group my $found_gid = $found_group->{'id'}; $k->unlock; my @matches=(); FINDS: foreach my $ent (@e) { my %new_ent = %{$ent}; # Clone the entity $new_ent{id} = int(rand(1000000000000000)); # A random new id $new_ent{group} = $found_gid; # Place this entry clone into /_found # $new_ent{path} is _NOT_ a normal key for File::KeePass but this is # safe because we are adding it to entries in the /_found group which # will not be saved to a file. my $nulled_path=$state->{all_ent_paths_rev}->{$ent->{id}}; $new_ent{path} = '/' . dirname(humanize_path($nulled_path)) . '/'; $new_ent{full_path} = '/' . humanize_path($nulled_path); $k->add_entry(\%new_ent); push(@matches, \%new_ent); } $k->lock; # Because we added a new /_found we must refresh our $state paths refresh_state_all_paths(); # Tell the user what we found print " - ".scalar(@matches)." matches found and placed into /$FOUND_DIR/\n"; # If we only found one, ask the user if they want to see it if (scalar(@matches) == 1) { print "Would you like to show this entry? [y/N] "; my $key=get_single_key(); print "\n"; if (lc($key) eq 'y') { my $search_params = { 'group_id =' => $found_gid }; my ($e,@empty) = $k->find_entries($search_params); my $full_path="/$FOUND_DIR/" . $e->{title}; cli_show($self, { args => [ $full_path ] }); } } } # Something is going wrong between KeePassX and File::KeePass related to # the unknown values read/written by File::KeePass from/to files written # by KeePassX. Commenting out line 378 of File/KeePass.pm is one fix, # this prevents me from needing to do that by just removing the unknown # values before saving. If there is a downside to this on the KeePassX # side I've not found it yet. I do have an email out to Paul, the author # of File::KeePass, requesting some assistance in grokking the problem. # # NOTE: I thought that this should not be needed for File::Keepass >= 0.3, # but on 2011-02-02 I discovered that creating new groups with # File::Keepass and not scrubbing on save created corrupt files: # "Group header offset is out of range" errors. # Sourceforge bug# 3187054 demonstrated the problem as well. sub scrub_unknown_values_from_all_groups { our $state; # No need to do this with newer versions of File::KeePass (fixed in 2.01) if (version->parse($File::KeePass::VERSION) >= version->parse('2.03')) { return; } my $k=$state->{kdb}; my @all_groups_flattened = $k->find_groups({}); my @unkown_field_groups=(); foreach my $g (@all_groups_flattened) { if (defined($g->{unknown})) { #warn "LHHD: " . &Dumper($g->{unknown}) . "\n"; delete $g->{unknown}; push @unkown_field_groups, $g->{title}; } } my $count = scalar(@unkown_field_groups); if ($count > 0) { warn "Deleted unknown fields from these $count groups: " . join(", ", @unkown_field_groups) . "\n"; } } sub deny_if_readonly { our %opts; if (defined($opts->{readonly}) && int($opts->{readonly})) { print "Function not available with --readonly set.\n"; return 1; } return 0; } sub cli_save($) { my $self = shift @_; my $params = shift @_; if (deny_if_readonly()) { return; } our $state; if (! length($state->{kdb_file})) { print "Please use the saveas command for new files.\n"; return; } # If the user has asked for a *.kdbx file, check the File::KeePass version if (version->parse($File::KeePass::VERSION) < version->parse('2.03')) { if ($state->{kdb_file} =~ m/\.kdbx$/i) { print "KeePass v2 (*.kdbx files) require File::KeePass >= v2.03\n"; return; } } if (warn_if_file_changed()) { return; } # Check for a lock file that we did not place there my $lock_file = $state->{kdb_file} . '.lock'; # KeePassX style if (-f $lock_file && $state->{placed_lock_file} ne $lock_file) { my $bold="\e[1m"; my $red="\e[31m"; my $yellow="\e[33m"; my $clear="\e[0m"; print $bold . $yellow . "WARNING:" . $clear . $red . " A KeePassX-style lock file is in place for this file.\n" . " It may be opened elsewhere. Save anyway? [y/N] " . $clear; my $key=get_single_key(); print "\n"; if (lc($key) ne 'y') { return; } } # If we got this far the user is OK with us locking the file even # if we had a contention. touch_file($lock_file); $state->{placed_lock_file} = $lock_file; # Scrub the data and write the file destroy_found(); scrub_unknown_values_from_all_groups(); # TODO - remove later my $k=$state->{kdb}; $k->unlock; my $master_pass= composite_master_pass($state->{get_master_passwd}(),$state->{key_file}); $k->save_db($state->{kdb_file},$master_pass); $state->{kdb_has_changed}=0; # set our state to no change since last save $master_pass="\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0"; print "Saved to $state->{kdb_file}\n"; $k->lock; # Update the md5sum of the file after we just saved it my $file = $state->{kdb_file}; $state->{kdb_file_md5} = Digest::file::digest_file_hex($file, "MD5"); } sub cli_rm($) { my $self = shift @_; my $params = shift @_; our $state; if (deny_if_readonly() || warn_if_file_changed()) { return; } my $target = $params->{args}->[0]; my $ent=find_target_entity_by_number_or_path($target); if (! defined($ent)) { print "Don't see an entry at path: $target\n"; return -1; } my $ent_id=$ent->{id}; $state->{kdb}->delete_entry({ id => $ent_id }); $state->{kdb_has_changed}=1; RequestSaveOnDBChange(); } sub validate_entry_number($$) { my $item_number=shift @_; my $path=shift @_; our $state; if (normalize_path_string($path) eq '') { print "Entries cannot exist in the root path (/).\n"; return -1; } if ($item_number !~ m/^[0-9]+$/) { print "Invalid item number (must be an integer).\n"; return -1; } my ($rGrps,$rEnts) = get_groups_and_entries($path); my $entry_max=scalar(@{$rEnts}) - 1; if ($item_number > $entry_max) { print "Invalid item number. Valid entries are 0-$entry_max.\n"; return -1; } return 0; } # This routine takes one parameter that will be either a path # to an entity or an entity number as shown my the ls command # and will use $state information such as last_ls_path to # return a reference to that entity in the $state-{kdb} database, # if possible (valid input). sub find_target_entity_by_number_or_path($) { my $target=shift @_; our $state; my $ent=undef; # hope to populate this in a second... # This section looks for an entity by an "ls" number if ($target =~ m/^[0-9]+$/) { my $path=$state->{last_ls_path}; if (!length($path)) { $path=get_pwd(); } if (! validate_entry_number($target,$path)) { my ($rGrps,$rEnts) = get_groups_and_entries($path); $ent=$rEnts->[$target]; } } # This section looks by a path name if (defined $state->{all_ent_paths_fwd}->{normalize_path_string($target)}) { my $entry_id=$state->{all_ent_paths_fwd}->{normalize_path_string($target)}; $ent = $state->{kdb}->find_entry( {id=>$entry_id} ); } return $ent; } sub cli_rename($$) { my $self = shift @_; my $params = shift @_; our $state; if (deny_if_readonly() || warn_if_file_changed()) { return; } my $target_dir = $params->{args}->[0]; my $dir_normalized=normalize_path_string($target_dir); my $grp=undef; if (defined($state->{all_grp_paths_fwd}->{$dir_normalized})) { my $grp_id = $state->{all_grp_paths_fwd}->{$dir_normalized}; $grp=$state->{kdb}->find_group( { id => $grp_id } ); } if (! defined($grp)) { print "Unknown group: $target_dir\n"; return -1; } print "Enter the groups new Title: "; my $new_title = ReadLine(0); chomp($new_title); if (length($new_title)) { $state->{kdb}->unlock; $grp->{title} = $new_title; $state->{kdb}->lock; } # Because we renamed a group we must refresh our $state paths refresh_state_all_paths(); $state->{kdb_has_changed}=1; RequestSaveOnDBChange(); } sub cli_mv($$) { my $self = shift @_; my $params = shift @_; our $state; if (deny_if_readonly() || warn_if_file_changed()) { return; } my $target_ent = $params->{args}->[0]; my $ent=find_target_entity_by_number_or_path($target_ent); if (! defined($ent)) { print "Unknown entry: $target_ent\n"; return -1; } my $target_dir = $params->{args}->[1]; my $dir_normalized=normalize_path_string($target_dir); my $grp=undef; if (defined($state->{all_grp_paths_fwd}->{$dir_normalized})) { my $grp_id = $state->{all_grp_paths_fwd}->{$dir_normalized}; $grp=$state->{kdb}->find_group( { id => $grp_id } ); } if (! defined($grp)) { print "Unknown group: $target_dir\n"; return -1; } # Verify no entry title conflict at the new location my $new_entry_path=$dir_normalized . "\0" . $ent->{title}; if (defined($state->{all_ent_paths_fwd}->{$new_entry_path})) { print "There is already and entry named \"$ent->{title}\" there.\n"; } # Unlock the kdb, clone the entry, remove its ID and set its new group, # add it to the kdb, delete the old entry, then lock the kdb... $state->{kdb}->unlock; my %ent_copy = %{$ent}; delete $ent_copy{id}; $ent_copy{group} = $grp; if ($state->{kdb}->add_entry(\%ent_copy)) { $state->{kdb}->delete_entry({ id=>$ent->{id} }); } $state->{kdb}->lock; # Because we moved an entry we must refresh our $state paths refresh_state_all_paths(); $state->{kdb_has_changed}=1; RequestSaveOnDBChange(); } sub cli_show($$) { my $self = shift @_; my $params = shift @_; our $state; # Users can provide a -f option to show the password. We use GetOptions # to parse this command line, and $target holds that target. my $target=''; my %opts=(); { local @ARGV = @{$params->{args}}; my $result = &GetOptions(\%opts, 'f', 'a'); if (scalar(@ARGV) != 1) { return -1; } $target = $ARGV[0]; } my $ent=find_target_entity_by_number_or_path($target); if (! defined($ent)) { return -1; } $state->{kdb}->unlock; print "\n"; if (defined($ent->{path})) { print show_format("Path",$ent->{path}) . "\n"; } # Unless -f is specified, we "hide" the password as red-on-red. my $password = $ent->{password}; if (! defined($opts{f})) { $password = colored(['red on_red'], $password); } print show_format("Title",$ent->{title}) . "\n" . show_format("Uname",$ent->{username}) . "\n" . show_format("Pass",$password) . "\n" . show_format("URL",$ent->{url}) . "\n" . show_format("Notes",$ent->{comment}) . "\n" . ($DEBUG ? show_format("ID",$ent->{id}) . "\n" : ''); if (defined($opts{a})) { print show_format("Icon#",$ent->{icon}) . "\n" . show_format("Creat",$ent->{created}) . "\n" . show_format("Modif",$ent->{modified}) . "\n"; } print "\n"; print &Dumper($ent) . "\n" if ($DEBUG > 2); $state->{kdb}->lock; } sub cli_edit($) { my $self = shift @_; my $params = shift @_; our $state; if (deny_if_readonly() || warn_if_file_changed()) { return; } my $target = $params->{args}->[0]; my $ent=find_target_entity_by_number_or_path($target); if (! defined($ent)) { print "Don't see an entry at path: $target\n"; return -1; } # Loop through the fields taking edits the user wants to make my @fields = get_entry_fields(); my $had_changes=0; foreach my $input (@fields) { if ($input->{hide_entry}) { print $input->{txt} . ": "; } else { my $val = $ent->{$input->{key}}; if ($val =~ m/\r|\n/) { $val = "\n$val\n"; } print $input->{txt} . " (\"".$val."\"): "; } if ($input->{genpasswd}) { print " "x25 . '("g" to generate a password)' . "\r"; } if ($input->{hide_entry}) { ReadMode(2); # Hide typing } my $val = ''; if ($input->{multiline}) { $val = new_edit_multiline_input($input); } else { $val = new_edit_single_line_input($input); } # If the field was not empty, change it to the new $val if (length($val)) { $state->{kdb}->unlock; $ent->{$input->{key}} = $val; $state->{kdb}->lock; $had_changes=1; } ReadMode(0); # Return to normal } # If the use made changes, update modify time and prompt them to save if ($had_changes) { $state->{kdb}->unlock; $ent->{modified} = $state->{kdb}->now; $state->{kdb}->lock; refresh_state_all_paths(); $state->{kdb_has_changed}=1; RequestSaveOnDBChange(); } return 0; } # Single line input helper function for cli_new and cli_edit. sub new_edit_single_line_input($) { my $input = shift @_; my $val = ReadLine(0); if ($input->{hide_entry}) { print "\n"; } chomp $val; if ($input->{genpasswd} && $val eq 'g') { $val=generatePassword(20); } elsif (length($val) && $input->{double_entry_verify}) { print "Retype to verify: "; my $checkval = ReadLine(0); if ($input->{hide_entry}) { print "\n"; } chomp $checkval; if ($checkval ne $val) { print "Entries mismatched. Please try again.\n"; redo; } } return $val; } # Multi-line input helper function for cli_new and cli_edit. sub new_edit_multiline_input($) { my $input = shift @_; my $bold="\e[1m"; my $red="\e[31m"; my $yellow="\e[33m"; my $clear="\e[0m"; print "\n$yellow(end multi-line input with a single \".\" on a line)$clear\n"; my $val = ''; my $unfinished = 1; while ($unfinished) { my $line = ReadLine(0); if ($line =~ m/^\.[\r\n]*$/) { # a lone "." ends our input $unfinished = 0; } else { $val .= $line; if ($val =~ m/^[\r\n]*$/) { $val = ''; $unfinished = 0; } } } chomp($val); # Remove extra line at the end return $val; } # Formats an entry for display for cli_show() sub show_format($$) { my $title=shift @_; my $value=shift @_; my $val=$value; if ($val =~ m/\n/) { my @val_lines=split(/\n/, $val); $val=join("\n" . " "x(5+2), @val_lines); } return sprintf("%5s: %s", $title,$val); } sub get_entry_fields { my @fields = ( { key=>'title', txt=>'Title' }, { key=>'username', txt=>'Username' }, { key=>'password', txt=>'Password', hide_entry => 1, double_entry_verify => 1, genpasswd => 1 }, { key=>'url', txt=>'URL' }, { key=>'comment', txt=>'Notes/Comments', 'multiline' => 1 }, ); return @fields; } sub cli_icons($) { my $self = shift @_; my $params = shift @_; our $state; print "Change icons on Groups or Entries (g/e/Cancel)? "; my $groups_or_entries=lc(get_single_key()); print "\n"; if ($groups_or_entries !~ m/^[ge]$/) { return; } print "Change icons Here, Below here, or Globally (h/b/g/Cancel)? "; my $glob_or_rel=lc(get_single_key()); print "\n"; if ($glob_or_rel !~ m/^[hgb]$/) { return; } print "What would you line the new icon to be (0-64/Cancel)? "; my $val = ReadLine(0); chomp($val); if ($val !~ m/^[0-9]+$/ || $val < 0 || $val > 64) { print "Invalid icon number.\n"; return; } # This code fills @{$groups} or @{$entries} with the items that the # user wants to change the icons on. my ($groups,$entries) = ([],[]); if ($glob_or_rel eq 'h') { # "Here" is easy -- we have a function for that. ($groups,$entries) = get_current_groups_and_entries(); if ($groups_or_entries eq 'e') { $groups = []; } else { $entries = []; } } else { if ($glob_or_rel eq 'g') { # Globally is easy, it's all groups my $k=$state->{kdb}; @{$groups} = $k->find_groups({}); } else { my $id=$state->{path}->{id}; @{$groups} = all_child_groups_flattened($id); # *only child groups* } # If the user wanted to operate on entries, collect all the entries # in the @{$groups} and then empty @{$groups}. if ($groups_or_entries eq 'e') { foreach my $group (@{$groups}) { if (defined($group->{entries})) { push @{$entries}, @{$group->{entries}}; } } $groups = []; } } # Change the items, recording the number of changes. my $items_changed=0; foreach my $item (@{$groups}, @{$entries}) { $item->{icon} = $val; $items_changed++; } # Tell the user what we did. print "The icon value was set to $val on $items_changed records.\n"; # If we changed anything, ask the user if they want to save if ($items_changed > 0) { $state->{kdb_has_changed}=1; RequestSaveOnDBChange(); } return 0; } sub cli_new($) { my $self = shift @_; my $params = shift @_; our $state; if (deny_if_readonly() || warn_if_file_changed()) { return; } my $pwd=get_pwd(); if ($pwd =~ m/^\/+$/) { print "Entries cannot be made in this path ($pwd).\n"; return -1; } print "Adding new entry to \"$pwd\"\n"; # Grab the entries at this $id (pwd) so we can check for conflicts my $k=$state->{kdb}; my $id=$state->{path}->{id}; my ($this_grp,@trash) = $k->find_groups({id=>$id}); my @entries = $k->find_entries({group_id => $id}); my $new_entry = { 'group' => $id, }; my @fields = get_entry_fields(); foreach my $input (@fields) { if ($input->{genpasswd}) { print " "x25 . '("g" to generate a password)' . "\r"; } print $input->{txt} . ": "; if ($input->{hide_entry}) { ReadMode(2); # Hide typing } my $val = ''; if ($input->{multiline}) { $val = new_edit_multiline_input($input); } else { $val = new_edit_single_line_input($input); } # If the user gave us an empty title, abort the new entry if ($input->{key} eq 'title' && length($val) == 0) { return; } $new_entry->{$input->{key}} = $val; ReadMode(0); # Return to normal } $new_entry->{icon} = $DEFAULT_ENTRY_ICON; $k->unlock; my $new_entry_ref = $k->add_entry($new_entry); $k->lock; if ($new_entry_ref->{id}) { refresh_state_all_paths(); $state->{kdb_has_changed}=1; RequestSaveOnDBChange(); } else { print "Failed to add new entry.\n"; } return 0; } sub cli_import($$) { my $file=shift @_; my $new_group=shift @_; my $key_file=shift @_; our $state; if (deny_if_readonly()) { return; } # If the user gave us a bogus file there's nothing to do if (! -f ($file)) { print "File does not exist: $file\n"; return -1; } # If the $new_group path is relative, make it absolute if ($new_group !~ m/^\//) { $new_group = get_pwd() . "/$new_group"; } # We won't import into an existing group my $full_path=normalize_path_string($new_group); if (defined($state->{all_grp_paths_fwd}->{$full_path})) { print "You must specify a _new_ group to import into.\n"; return -1; } # Make sure the new group's parent exists my ($grp_path,$grp_name)=normalize_and_split_raw_path($new_group); if ($grp_path != '' && ! defined($state->{all_grp_paths_fwd}->{$grp_path})) { print "Path does not exist: /" . humanize_path($grp_path) . "\n"; return -1; } # Set the $parent_group value appropriately my $parent_group = undef; # Root by default if (length($grp_path)) { $parent_group = $state->{all_grp_paths_fwd}->{$grp_path}; } # Ask the user for the master password and then open the kdb my $master_pass=GetMasterPasswd(); my $iKDB = File::KeePass->new; if (! eval { $iKDB->load_db($file, composite_master_pass($master_pass, $key_file)) }) { print "Couldn't load the file $file: $@\n"; return -1; } # Add the new group, to its parent or to root if $parent_group==undef my $k=$state->{kdb}; my $new_group=$k->add_group({ title => $grp_name, group => $parent_group, }); # Copy the $iKDB into our $k at $new_group $iKDB->unlock(); $k->unlock(); my @root_groups = $iKDB->find_groups({level=>0}); foreach my $i_root_grp (@root_groups) { copy_kdb_group_tree($k,$i_root_grp,$new_group); } $k->lock(); $iKDB->lock(); $iKDB=undef; # Refresh all paths and mark state as changed refresh_state_all_paths(); $state->{kdb_has_changed}=1; RequestSaveOnDBChange(); } sub cli_export($$) { my $file=shift @_; my $key_file=shift @_; our $state; # Warn is we are being asked to overwrite a file if (-e $file) { print "WARNING: $file already exists.\n" . "Overwrite it? [y/N] "; my $key=get_single_key(); print "\n"; if (lc($key) ne 'y') { return -1; } } # Get the master password for the exported file my $master_pass=GetMasterPasswd(); if (length($master_pass) == 0) { print "For your safety, empty passwords are not allowed...\n"; return; } print "Retype to verify: "; ReadMode('noecho'); my $checkval = ReadLine(0); ReadMode('normal'); chomp $checkval; print "\n"; if ($master_pass ne $checkval) { print "Passwords did not match...\n"; return; } # Build the new kdb in RAM my $k=$state->{kdb}; my $new_kdb=new File::KeePass; $k->unlock; # Required so that we can copy the passwords if (get_pwd() ne '/') { # Grab the root group's $id at our pwd my $pwd_group_id=$state->{path}->{id}; my ($root_grp,@trash) = $k->find_groups({id=>$pwd_group_id}); copy_kdb_group_tree($new_kdb,$root_grp,undef); } else { # Put all of the root groups into the new file (entire file copy) my @root_groups = $k->find_groups({level=>0}); foreach my $root_grp (@root_groups) { copy_kdb_group_tree($new_kdb,$root_grp,undef); } } $k->lock; $new_kdb->unlock; my $new_db_bin = $new_kdb->gen_db(composite_master_pass($master_pass,$key_file)); $new_kdb->lock; # Test parsing the kdb from RAM (we'll most likely die if this fails) my $new_db=new File::KeePass; $new_db->parse_db($new_db_bin,composite_master_pass($master_pass,$key_file)); # Now write the new kdb to disk my $fh=new FileHandle; if (open($fh,'>',$file)) { print $fh $new_db_bin; close $fh; print "Exported to $file\n"; } else { print "Could not open \"$file\" for writing.\n"; } return 0; } # A helper function for cli_export() and cli_import(). It takes a kdb object, # a group as a starting point to copy from, and optionally a parent_group to # copy to. It copies everything from the source group's root downward. In our # use cases, the _target_ $kdb object passed in here is typically a different # one than the _source_ $group is from. sub copy_kdb_group_tree($$$) { my $kdb=shift @_; my $group=shift @_; my $parent_group=shift @_ || undef; # When undef, it writes to the root # Add the new group, to it's parent or root if $parent_group==undef my $new_group=$kdb->add_group({ title => $group->{title}, icon => $group->{icon}, id => $group->{id}, group => $parent_group, }); # Add the new_group's entries if (ref($group->{entries}) eq 'ARRAY') { foreach my $entry (@{$group->{entries}}) { $entry->{group} = $new_group; $kdb->add_entry($entry); } } # Add the new_group's child groups if (ref($group->{groups}) eq 'ARRAY') { foreach my $child_grp (@{$group->{groups}}) { copy_kdb_group_tree($kdb,$child_grp,$new_group); } } } sub cli_saveas($) { my $file=shift @_; my $key_file=shift @_; our $state; # If the user has asked for a *.kdbx file, check the File::KeePass version if (version->parse($File::KeePass::VERSION) < version->parse('2.03')) { if ($file =~ m/\.kdbx$/i) { print "KeePass v2 (*.kdbx files) require File::KeePass >= v2.03\n"; return; } } my $master_pass=GetMasterPasswd(); print "Retype to verify: "; ReadMode('noecho'); my $checkval = ReadLine(0); ReadMode('normal'); chomp $checkval; print "\n"; if ($master_pass ne $checkval) { print "Passwords did not match...\n"; return; } destroy_found(); scrub_unknown_values_from_all_groups(); # TODO - remove later $state->{kdb}->unlock; $state->{kdb}->save_db($file,composite_master_pass($master_pass,$key_file)); $state->{kdb}->lock; $state->{kdb}= File::KeePass->new; if (! eval { $state->{kdb}->load_db($file, composite_master_pass($master_pass,$key_file)) }) { die "Couldn't load the file $file: $@"; } $state->{kdb_has_changed}=0; $state->{kdb_file} = $file; $state->{key_file} = $key_file; $state->{put_master_passwd}($master_pass); $master_pass="\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0\0"; return 0; } # This routine takes a raw (directly from user input) path string # and "normalizes" it (converts it to NULL-separated w/out escapes) # and splits it into its dirname and basename components. sub normalize_and_split_raw_path($) { my $raw_pathstr=shift @_; my $path=normalize_path_string($raw_pathstr); my $basename=''; if ($path =~ m/\0/) { # case of at least one path delimeter $path =~ s/\0([^\0]+)$//; $basename=$1; } elsif ($raw_pathstr =~ m/^\//) { # case of simple "/foo" (no subdir) $basename=$path; $path=''; } else { # case of simple "foo" (no path delims) $path = normalize_path_string(get_pwd()); $basename=$raw_pathstr; } return($path,$basename); } sub cli_rmdir($) { my $self = shift @_; my $params = shift @_; our $state; if (deny_if_readonly() || warn_if_file_changed()) { return; } my $raw_pathstr=$params->{'args'}->[0]; my ($path,$grp_name) = normalize_and_split_raw_path($raw_pathstr); # Make sure the group exists. my $grp_path="$path\0$grp_name"; $grp_path=~s/^\0+//; if (! defined($state->{all_grp_paths_fwd}->{$grp_path})) { print "Path does not exist: /" . humanize_path($grp_path) . "\n"; return -1; } my $group_id = $state->{all_grp_paths_fwd}->{$grp_path}; my $group = $state->{kdb}->find_group({ id => $group_id }); my $entry_cnt=0; if (defined($group->{entries})) { $entry_cnt= scalar(grep(m/^$grp_path\0/,keys %{$state->{all_ent_paths_fwd}})); } my $group_cnt=0; if (defined($group->{entries})) { $group_cnt= scalar(grep(m/^$grp_path\0/,keys %{$state->{all_grp_paths_fwd}})); } my $child_cnt=$entry_cnt + $group_cnt; if ( $child_cnt > 0) { print "WARNING: This group has $child_cnt child groups and/or entries.\n" . "Really remove it!? [y/N] "; my $key=get_single_key(); print "\n"; if (lc($key) ne 'y') { return -1; } } my $deleted_group = $state->{kdb}->delete_group({ id => $group_id }); # Because we removed a group we need to refresh our state paths refresh_state_all_paths(); $state->{kdb_has_changed}=1; RequestSaveOnDBChange(); return 0; } sub cli_mkdir($) { my $self = shift @_; my $params = shift @_; our $state; if (deny_if_readonly() || warn_if_file_changed()) { return; } my $raw_pathstr = $params->{args}->[0]; my ($path,$newdir) = normalize_and_split_raw_path($raw_pathstr); # Make sure the group doesn't already exist. my $newdir_path="$path\0$newdir"; $newdir_path=~s/^\0+//; if (defined($state->{all_grp_paths_fwd}->{$newdir_path})) { print "Path already exists: /" . humanize_path($newdir_path) . "\n"; return -1; } # Create the group my $group=''; if ($path eq '') { $group = $state->{kdb}->add_group({ title => $newdir, icon => $DEfAULT_GROUP_ICON, }); # root level group } elsif (defined($state->{all_grp_paths_fwd}->{$path})) { my $group_id=$state->{all_grp_paths_fwd}->{$path}; $group = $state->{kdb}->add_group({ title => $newdir, group => $group_id, icon => $DEfAULT_GROUP_ICON, }); } else { print "Cannot make directory at path " . humanize_path($path) . "\n"; return -1; } # Because we created a new group we need to refresh ours state paths refresh_state_all_paths(); $state->{kdb_has_changed}=1; RequestSaveOnDBChange(); return 0; } sub humanize_path($) { my $path=shift @_; $path =~ s/\//\\\//g; $path =~ s/\0/\//g; return $path; } sub cli_open($) { my $path=shift @_; my $key=shift @_; our $state; # If cli_close() does not return 0 the user decided not to close the file if (cli_close() != 0) { return -1; } if ( -f $path ) { my $err = open_kdb($path, $key); if (length($err)) { print "Error opening file: $err\n"; } } else { print "Cannot open: $path\n"; } } # Get a single keypress from the user sub get_single_key { my $key=''; ReadMode('raw'); # Turn off controls keys while (not defined ($key = ReadKey(-1))) { # No key yet } ReadMode('restore'); return $key; } sub cli_close { our $state; if ($state->{kdb_has_changed}) { print "WARNING: The database has changed and was not saved.\n" . "Really close it? [y/N] "; my $key=get_single_key(); print "\n"; if (lc($key) ne 'y') { return -1; } } new_kdb($state); return 0; } # This sets $state to a brand new, KeePassX-style, empty, unsaved database sub new_kdb($) { my $state=shift @_; $state->{kdb_has_changed}=0; $state->{'kdb'} = File::KeePass->new; # To be compatible with KeePassX $state->{'kdb'}->add_group({ title => 'eMail' }); $state->{'kdb'}->add_group({ title => 'Internet' }); refresh_state_all_paths(); if (-f $state->{placed_lock_file}) { unlink($state->{placed_lock_file}); } delete($state->{placed_lock_file}); delete($state->{kdb_file}); delete($state->{key_file}); delete($state->{master_pass}); cli_cd($term, {'args' => ["/"]}); } sub cli_ls($) { my $self = shift @_; my $params = shift @_; our $state; my $path=$params->{'args'}->[0]; # If we were given a path, use cli_cd() to go there temporarily... my $old_path=''; if (length($path)) { $old_path=get_pwd(); if (cli_cd($term, {'args' => [$path]})) { return -1; # If cli_cd() returned non-zero it failed } } # List the pwd $state->{last_ls_path} = get_pwd(); my ($rGrps,$rEnts) = get_current_groups_and_entries(); if (scalar(@{$rGrps}) > 0) { print "=== Groups ===\n"; print join("\n", @{get_human_group_list($rGrps)}) . "\n"; } if (scalar(@{$rEnts}) > 0) { print "=== Entries ===\n"; print join("\n", @{get_human_entry_list($rEnts)}) . "\n"; } # If we temporarily cd'ed, cd back. if (length($old_path)) { cli_cd($term, {'args' => [$old_path]}); } return 0; } # Helper function for cli_ls() sub get_human_group_list($) { my $rGroup=shift @_; my @list=(); foreach my $grp (@{$rGroup}) { #push (@list, sprintf("%15d %s/", $grp->{id}, $grp->{title})); push (@list, "$grp->{title}/"); push (@list, &Dumper($grp)) if ($DEBUG > 2); } return \@list; } # Helper function for cli_ls() sub get_human_entry_list($) { my $rEntries=shift @_; my @list=(); my $i=0; my $d_len=int((scalar(@{$rEntries}) - 1) / 10) + 1; foreach my $ent (@{$rEntries}) { my $url=$ent->{url}; $url=~s/^https?:\/\///i; $url=~s/\/+$//; push (@list, sprintf("%".$d_len."d. %-40.40s %30.30s", $i, $ent->{title}, $url)); $i++; } return \@list; } # Routine to hook into Term::ShellUI's exit on Ctrl-D functionality sub eof_exit_hook { our $state; # We need a newline if cli_quit() will talk tothe user about saving if ($state->{kdb_has_changed}) { print "\n"; } # cli_quit() will handle user interaction and return a value for # the exit_hook of Term::ShellUI. return cli_quit($state->{term},undef); } sub cli_quit($$) { my $self = shift @_; my $params = shift @_; our $state; if ($state->{kdb_has_changed}) { print "WARNING: The database has changed and was not saved.\n" . "Really quit? [y/N] "; my $key=get_single_key(); if (lc($key) ne 'y') { print "\n"; return -1; # It is not OK to quit } } if (-f $state->{placed_lock_file}) { unlink($state->{placed_lock_file}); } delete($state->{placed_lock_file}); $self->exit_requested(1); return 0; # It's OK to quit } # Function to nag the user about saving each time the DB is modified sub RequestSaveOnDBChange { our $state; # If the db hasn't changed don't bother the user if (! $state->{kdb_has_changed}) { return -1; } # If this is a newly created file we don't bother the user with # asking to save after every change. if (! length($state->{kdb_file})) { return -1; } print "Database was modified. Do you want to save it now? [y/N]: "; my $key=get_single_key(); print "\n"; if (lc($key) ne 'y') { return; } # Calling cli_save() should be silent and safe at this point. return cli_save(undef); } sub GetMasterPasswd { print "Please provide the master password: "; ReadMode('noecho'); my $master_pass = ReadLine(0); chomp $master_pass; ReadMode('normal'); print "\n"; return $master_pass; } sub MyGetOpts { my %opts=(); my $result = &GetOptions(\%opts, "kdb=s", "key=s", "histfile=s", "help", "h", "readonly"); # If the user asked for help or GetOptions complained, give help and exit if ($opts{help} || $opts{h} || (! int($result))) { print GetUsageMessage(); exit; } # Allow the user to override the history file if (defined($opts{histfile}) && length($opts{histfile})) { our $HISTORY_FILE = $opts{histfile}; } else { our $HISTORY_FILE = "~/.$APP_NAME-history"; } my @errs=(); if ((length($opts{kdb}) && (! -e $opts{kdb}))) { push @errs, "for option --kdb=, the file must exist."; } if ((length($opts{key}) && (! -e $opts{key}))) { push @errs, "for option --key=, the file must exist."; } if (scalar(@errs)) { warn "There were errors:\n" . " " . join("\n ", @errs) . "\n\n"; die &GetUsageMessage(); } return \%opts; } sub GetUsageMessage { my $t="Usage: $APP_NAME [--kdb=] [--key=]\n" . "\n" . " --help\tThis message.\n" . " --kdb\tOptional KeePass database file to open (must exist).\n" . " --key\tOptional KeePass key file (must exist).\n" . " --histfile\tSpecify your history file (or perhaps /dev/null).\n" . " --readonly\tRun in read-only mode; no changes will be allowed.\n" . "\n" . "Run kpcli with no options and type 'help' at its command prompt to learn\n" . "about kpcli's commands.\n"; "\n"; return $t; } # Because Term::ShellUI has a fixed width (%20s) for the command length # and we don't need nearly that much, we had to implement our own help # function instead of using the built-in help_call() method. sub my_help_call($) { my $term = shift @_; # @_ now holds: [Term::ShellUI->{commands}, ] # If the user is asking for detailed help on a specific command, do that if (scalar(@_) > 1) { $term->help_call(undef, @_); return; } # If no specific command was requested, show the command summaries my $help = $term->get_all_cmd_summaries($term->commands()); $help =~ s/^ {12}//gm; # Trim some leading spaces off of each line of output print $help; print "\n" . "Type \"help \" for more detailed help on a command.\n"; return 0; } ######################################################################## # Command Completion Routines ########################################## ######################################################################## sub complete_groups { my $self = shift; my $cmpl = shift; our $state; $self->suppress_completion_append_character(); my $path = $cmpl->{str} || "."; # Tack on trailing slashes for user convenience... if ($path =~ m/(^|\/)[.][.]$/) { $path .= "/"; } # onto "..../.." if ($path eq '.') { $path .= "/"; } # onto a single "." my $srch_path=normalize_path_string($path); my @possibles = grep(/^$srch_path\0?[^\0]+$/, sort keys %{$state->{all_grp_paths_fwd}}); my @results=(); foreach my $opt (@possibles) { $opt=humanize_path($opt); if ($path =~ m/^(\/+)/) { # Absolute path (easy case!) $opt=$1.$opt; } else { # Path relative to pwd my $user_new_path=get_pwd() . "/" . $path; my $new_dir=normalize_path_string($user_new_path); # If the user's input does not resolve to a fully qualified # path then we need to pop off the last bit to get to that. if (! defined($state->{all_grp_paths_fwd}->{$new_dir})) { my @path=split(/\0/, $new_dir); pop @path; $new_dir = join("\0", @path); } $new_dir = humanize_path($new_dir); # Dirname requires "../." (Trailing dot) to give ".." in those cases! my $dirname_path=$path; if ($path =~ m/\/$/) { $dirname_path .= "."; } my $path_to_put_back=dirname($dirname_path) . "/"; if ($new_dir eq '') { # All the way at the root level $opt = $path_to_put_back . $opt; } else { # Some non-root level (deeper) $opt=~s/^$new_dir\//$path_to_put_back/; } # Lop the leading "./" off the head if it was not user supplied if ($path !~ m/^[.]\// && $opt =~ m/^[.]\//) { $opt =~ s/^[.]\///; } # If the user did supply a leading "./" and we missed it, add it if ($path =~ m/^[.]\// && $opt !~ m/^[.]\//) { $opt = "./$opt"; } } push @results, "$opt/"; } return \@results; } sub complete_groups_and_entries { my $self = shift; my $cmpl = shift; our $state; # Grab the groups for this path my $groups=complete_groups($self,$cmpl); # Now gather up the entries (code very similar to complete_groups() my $path = $cmpl->{str} || "."; my $srch_path=normalize_path_string($path); my @entries = grep(/^$srch_path\0?[^\0]*$/, sort keys %{$state->{all_ent_paths_fwd}}); # This loop is modifying @entries values in place! foreach my $opt (@entries) { $opt=humanize_path($opt); if ($path =~ m/^(\/+)/) { $opt=$1.$opt; } else { my $pwd=get_pwd(); $pwd=~s/^\///; $opt=~s/^$pwd\/+//; if ($path =~ m/^[\/.]+/) { $opt=$1.$opt; } } # Lop the leading "./" off the head if it was not user supplied if ($path !~ m/^[.]\// && $opt =~ m/^[.]\//) { $opt =~ s/^[.]\///; } # If the user did supply a leading "./" and we missed it, add it if ($path =~ m/^[.]\// && $opt !~ m/^[.]\//) { $opt = "./$opt"; } } # Merge the groups and entries my @possibles = sort (@{$groups}, @entries); return \@possibles; } ######################################################################## # Rijndael encrypt/decrypt routines - borrowed from File::KeePass ###### ######################################################################## sub decrypt_rijndael_cbc { my ($buffer, $key, $enc_iv) = @_; my $cipher = Crypt::Rijndael->new($key, Crypt::Rijndael::MODE_CBC()); $cipher->set_iv($enc_iv); $buffer = $cipher->decrypt($buffer); my $extra = ord(substr $buffer, -1, 1); substr($buffer, length($buffer) - $extra, $extra, ''); return $buffer; } sub encrypt_rijndael_cbc { my ($buffer, $key, $enc_iv) = @_; my $cipher = Crypt::Rijndael->new($key, Crypt::Rijndael::MODE_CBC()); $cipher->set_iv($enc_iv); my $extra = (16 - length($buffer) % 16) || 16; # pad so we can trim $buffer .= chr($extra) for 1 .. $extra; return $cipher->encrypt($buffer); } sub composite_master_pass($$) { my ($pass, $key_file) = @_; # composite password in case of key file if (defined $key_file and length($key_file) and -f $key_file) { # KeePass v2.03 and higher has native key file support, and so we # use that if we have it. if (version->parse($File::KeePass::VERSION) >= version->parse('2.03')) { return [$pass, $key_file]; } # TODO - at some point, when File::KeePass v2.03 is very mainstream, # the code to the end of this if block should be removed. It allowed # support for key files for *.kdb files before File::KeePass supported # that natively. File::KeePass now also supports that for *.kdbx. open(my $fh,'<',$key_file) || die "Couldn't open key file $key_file: $!\n"; my $size = -s $key_file; read($fh, my $buffer, $size); close $fh; if (length($buffer) != $size) { die "Couldn't read entire key file contents of $key_file.\n"; } $pass = substr(sha256($pass),0,32); if ($size == 32) { $pass .= $buffer; } elsif ($size == 64) { for (my $i = 0; $i < 64; $i += 2) { $pass .= chr(hex(substr($buffer,$i,2))); } } else { $pass .= substr(sha256($buffer),0,32); } } return $pass; } sub put_master_passwd($) { my $master_pass = shift @_; our $state; $state->{'master_pass_key'}=''; $state->{'master_pass_key'} .= chr(int(255 * rand())) for 1..16; $state->{'master_pass_enc_iv'}=''; $state->{'master_pass_enc_iv'} .= chr(int(255 * rand())) for 1..16; $master_pass='CLEAR:' . $master_pass; $state->{'master_pass'}=encrypt_rijndael_cbc($master_pass, $state->{'master_pass_key'}, $state->{'master_pass_enc_iv'}); return 0; } sub get_master_passwd() { our $state; my $master_pass=decrypt_rijndael_cbc($state->{master_pass}, $state->{'master_pass_key'}, $state->{'master_pass_enc_iv'}); if ($master_pass=~s/^CLEAR://) { return $master_pass; } else { die "Failed to properly decrypt my copy of the master password.\n"; } } # This routine checks to see if the file has changed on disk and warns if so sub warn_if_file_changed { our $state; my $file = $state->{kdb_file}; if (! length($file)) { return 0; } # If no file was opened, don't warn my $file_md5 = Digest::file::digest_file_hex($file, "MD5"); if ($state->{kdb_file_md5} ne $file_md5) { my $bold="\e[1m"; my $red="\e[31m"; my $yellow="\e[33m"; my $clear="\e[0m"; print $bold . $yellow . "WARNING:" . $clear . $red . " The file has changed on disk since kpcli opened it!\n" . " It may be opened elsewhere. Continue anyway? [y/N] " . $clear; my $key=get_single_key(); print "\n"; if (lc($key) ne 'y') { return -1; } } return 0; } sub generatePassword { my $length = shift; my @normal_chars=('a'..'z','A'..'Z',0..9); my @special_chars=qw(_); my $charset = join('', (@normal_chars,@special_chars)); # Generate the password my $password = ''; while (length($password) < $length) { $password .= substr($charset, (int(rand(length($charset)))), 1); } # Make sure that at least one special character appears my $sccc=join('', @special_chars); if ($password !~ m/[\Q$sccc\E]/) { my $sc=$special_chars[int(rand(length($sccc)))]; substr($password,int(rand(length($password))), 1, $sc); } return $password } ######################################################################## # Unix-style, "touch" a file ######################################################################## sub touch_file { my $filename = shift @_; if (! -f $filename) { my $fh=new FileHandle; open($fh, "> $filename"); close($fh); } my $sig_pipe_store=$SIG{'PIPE'}; $SIG{'PIPE'} = 'IGNORE'; my $now=time; my $retval=utime $now, $now, $filename; $SIG{'PIPE'} = $sig_pipe_store; return($retval); } ######################################################################## # POD ################################################################## ######################################################################## =head1 NAME kpcli - A command line interface to KeePass database files. =head1 DESCRIPTION A command line interface (interactive shell) to work with KeePass database files (http://http://en.wikipedia.org/wiki/KeePass). This program was inspired by my use of "kedpm -c" combined with my need to migrate to KeePass. The curious can read about the Ked Password Manager at http://http://kedpm.sourceforge.net/. =head1 USAGE Please run the program and type "help" to learn how to use it. =head1 PREREQUISITES This script requires these non-core modules: C - "apt-get install libcrypt-rijndael-perl" on Ubuntu 10.04 C - "apt-get install libterm-readkey-perl" on Ubuntu 10.04 C - "apt-get install libsort-naturally-perl" on Ubuntu 10.04 C - "apt-get install libfile-keepass-perl" on Ubuntu 12.04 C - "apt-get install libterm-shellui-perl" on Ubuntu 12.10 It is also recommended that you install C which will give you command history and completion functionality. That module is in the libterm-readline-gnu-perl package on Ubuntu. =head1 CAVEATS AND WORDS OF CAUTION Only interoperability with KeePassX (http://www.keepassx.org/) has been tested. File::KeePass seems to have a bug related to some "unknown" data that KeePassX stores in the *.kdb file. This program deletes those unknown data when saving. Research into libkpass http://libkpass.sourceforge.net/) has revealed what File::KeePass classifies as "unknown" are the times for created/modified/accessed/expires as well as "flags" (id=9), but only for groups -- File::KeePass seems to handle those fields just fine for entries. I have not found any ill-effect from dropping those fields when saving and so that is what kpcli does today to work around this File::KeePass bug. =head1 BUGS =head2 Tab Completion Tab completion is not perfect. It has problems with some entries that contain spaces, slashes, and/or backslashes. I don't know if I am doing something wrong or if Term::ShellUI may have some bugs in its command completion code. =head2 Using Ctrl-D to Exit Versions of Term::ShellUI prior to v0.9. do not have the ability to trap Ctrl-D exits by the client program. I submitted a patch to remedy that and it made it into Term::ShellUI v0.9. Please upgrade if kpcli asks you to. =head2 Multiple Entries or Groups With the Same Name in the Same Group This program does not support multiple entries in the same group having the exact same name, nor does it support multiple groups at the same level having the same name, and it likely never will. KeepassX does support those and this program needs to be updated to detect and alert when an opened database file has those issues, and maybe refuse to save (overwrite) a file that is opened like that. Alternatively, we could alert and rename offending groups/entries at load time, by appending "-2", "-3", etc. =head1 AUTHOR Lester Hightower =head1 LICENSE This program may be distributed under the same terms as Perl itself. =head1 CREDITS Special thanks to Paul Seamons, author of File::KeePass, and to Scott Bronson, author of Term::ShellUI. Without those two modules this work would never have been posible. =head1 CHANGELOG 2010-Nov-28 - v0.1 - Initial release. 2010-Nov-28 - v0.2 - Encrypt the master password in RAM. 2010-Nov-29 - v0.3 - Fixed master password encryption for saveas. 2010-Nov-29 - v0.4 - Fixed code to work w/out Term::ReadLine::Gnu. Documented File::KeePass v0.1 hierarchy bug. 2010-Nov-29 - v0.5 - Made find command case insensitive. Bugfix in new command (path regex problem). 2010-Nov-29 - v0.6 - Added lock file support; warn if a lock exists. 2010-Dec-01 - v0.7 - Further documented the group fields that are dropped, in the CAVEATS section of the POD. Sort group and entry titles naturally. 2010-Dec-23 - v0.8 - Worked with File::KeePass author to fix a couple of bugs and then required >=v0.03 of that module. Sorted "/_found" to last in the root group list. Fixed a "database changed" state bug in cli_save(). Made the find command ignore entries in /Backup/. Find now offers show when only one entry is found. Provided a patch to Term::ShellUI author to add eof_exit_hook and added support for it to kpcli. 2011-Feb-19 - v0.9 - Fixed bugs related to spaces in group names as reported in SourceForge bug number 3132258. The edit command now prompts to save on changes. Put scrub_unknown_values_from_all_groups() calls back into place after realizing that v0.03 of File::KeePass did not resolve all of the problems. 2011-Apr-23 - v1.0 - Changed a perl 5.10+ regex to a backward-compatable one to resolve SourceForge bug number 3192413. Modified the way that the /Backup group is ignored by the find command to stop kpcli from croaking on multiple entries with the same name in that group. - Note: There is a more general bug here that needs addressing (see BUGS section). An empty title on new entry aborts the new entry. Changed kdb file are now detected/warned about. Tested against Term::ShellUI v0.9, which has my EOF hook patch, and updated kpcli comments about it. Term::ShellUI's complete_history() method was removed between v0.86 and v0.9 and so I removed kpli's call to it (Ctrl-r works for history). Added the "icons" command. 2011-Sep-07 - v1.1 - Empty DBs are now initialized to KeePassX style. Fixed a couple of bugs in the find command. Fixed a password noecho bug in the saveas command. Fixed a kdb_has_changed bug in the saveas command. Fixed a cli_open bug where it wasn't cli_close'ing. Fixed variable init bugs in put_master_passwd(). Fixed a false warning in warn_if_file_changed(). 2011-Sep-30 - v1.2 - Added the "export" command. Added the "import" command. Command "rmdir" asks then deletes non-empty groups. Command "new" can auto-generate random passwords. 2012-Mar-03 - v1.3 - Fixed bug in cl command as reported in SourceForge bug number 3496544. 2012-Apr-17 - v1.4 - Added key file support based on a user contributed patch with SourceForge ID# 3518388. Added my_help_call() to allow for longer and more descriptive command summaries (for help command). Stopped allowing empty passwords for export. 2012-Oct-13 - v1.5 - Fixed "help " commands, that I broke in v1.4. Command "edit" can auto-generate random passwords. Added the "cls" and "clear" commands from a patch with SourceForge ID# 3573930. Tested compatibility with File::KeePass v2.03 and made minor changes that are possible with >=2.01. With File::KeePass v2.03, kpcli should now support KeePass v2 files (*.kdbx). 2012-Nov-25 - v1.6 - Hide passwords (red on red) in the show command unless the -f option is given. Added the --readonly command line option. Added support for multi-line notes/comments; input ends on a line holding a single ".". 2013-Apr-25 - v1.7 - Patched to use native File::KeePass support for key files, if the File::KeePass version is new enough. Added the "version" and "ver" commands. Updated documentation as Ubuntu 12.10 now packages all of kpcli's dependencies. Added --histfile command line option. Record modified times on edited records, from a patch with SourceForge ID# 3611713. Added the -a option to the show command. =head1 TODO ITEMS Handle Ctrl-C presses (complicated by Term::ShellUI) - http://mail.pm.org/pipermail/melbourne-pm/2007-January/002214.html =head1 OPERATING SYSTEMS AND SCRIPT CATEGORIZATION =pod OSNAMES Unix-like (written and tested on Ubuntu Linux 10.04.1 LTS). =pod SCRIPT CATEGORIES UNIX/System_administration =cut