Initial commit

.gitignore

@@ -0,0 +1,7 @@
+.gradle/
+.idea/
+build/
+out/
+*.iml
+*.ipr
+*.iws

.travis.yml

@@ -0,0 +1,7 @@
+language: java
+
+jdk:
+  - oraclejdk8
+
+os:
+- linux

LICENSE.txt

@@ -0,0 +1,202 @@
+
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship, whether in Source or
+      Object form, made available under the License, as indicated by a
+      copyright notice that is included in or attached to the work
+      (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other modifications
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean any work of authorship, including
+      the original version of the Work and any modifications or additions
+      to that Work or Derivative Works thereof, that is intentionally
+      submitted to Licensor for inclusion in the Work by the copyright owner
+      or by an individual or Legal Entity authorized to submit on behalf of
+      the copyright owner. For the purposes of this definition, "submitted"
+      means any form of electronic, verbal, or written communication sent
+      to the Licensor or its representatives, including but not limited to
+      communication on electronic mailing lists, source code control systems,
+      and issue tracking systems that are managed by, or on behalf of, the
+      Licensor for the purpose of discussing and improving the Work, but
+      excluding communication that is conspicuously marked or otherwise
+      designated in writing by the copyright owner as "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any individual or Legal Entity
+      on behalf of whom a Contribution has been received by Licensor and
+      subsequently incorporated within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a
+      cross-claim or counterclaim in a lawsuit) alleging that the Work
+      or a Contribution incorporated within the Work constitutes direct
+      or contributory patent infringement, then any patent licenses
+      granted to You under this License for that Work shall terminate
+      as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or
+          Derivative Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, then any Derivative Works that You distribute must
+          include a readable copy of the attribution notices contained
+          within such NOTICE file, excluding those notices that do not
+          pertain to any part of the Derivative Works, in at least one
+          of the following places: within a NOTICE text file distributed
+          as part of the Derivative Works; within the Source form or
+          documentation, if provided along with the Derivative Works; or,
+          within a display generated by the Derivative Works, if and
+          wherever such third-party notices normally appear. The contents
+          of the NOTICE file are for informational purposes only and
+          do not modify the License. You may add Your own attribution
+          notices within Derivative Works that You distribute, alongside
+          or as an addendum to the NOTICE text from the Work, provided
+          that such additional attribution notices cannot be construed
+          as modifying the License.
+
+      You may add Your own copyright statement to Your modifications and
+      may provide additional or different license terms and conditions
+      for use, reproduction, or distribution of Your modifications, or
+      for any such Derivative Works as a whole, provided Your use,
+      reproduction, and distribution of the Work otherwise complies with
+      the conditions stated in this License.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or redistributing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or consequential damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or any and all
+      other commercial damages or losses), even if such Contributor
+      has been advised of the possibility of such damages.
+
+   9. Accepting Warranty or Additional Liability. While redistributing
+      the Work or Derivative Works thereof, You may choose to offer,
+      and charge a fee for, acceptance of support, warranty, indemnity,
+      or other liability obligations and/or rights consistent with this
+      License. However, in accepting such obligations, You may act only
+      on Your own behalf and on Your sole responsibility, not on behalf
+      of any other Contributor, and only if You agree to indemnify,
+      defend, and hold each Contributor harmless for any liability
+      incurred by, or claims asserted against, such Contributor by reason
+      of your accepting any such warranty or additional liability.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the file format. We also recommend that a
+      file or class name and description of purpose be included on the
+      same "printed page" as the copyright notice for easier
+      identification within third-party archives.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

README.md

@@ -0,0 +1,43 @@
+# KeeCrack [![Build Status](https://travis-ci.org/wbrawner/keecrack.svg?branch=master)](https://travis-ci.org/wbrawner/keecrack)
+[![Maintainability](https://api.codeclimate.com/v1/badges/a35dff49221e36abf189/maintainability)](https://codeclimate.com/github/wbrawner/keecrack/maintainability)
+[![Test Coverage](https://api.codeclimate.com/v1/badges/a35dff49221e36abf189/test_coverage)](https://codeclimate.com/github/wbrawner/keecrack/test_coverage)
+
+KeeCrack is a Java program used for brute-forcing KeePass database file master passwords. This should go without saying
+but use of this application is prohibited without the express consent of the owner of the database file.
+
+## Usage
+
+You can download a DEB, JAR, or RPM from the releases page. Each build contains both a <abbr title="graphical user
+interface">GUI</abbr> and <abbr title="command line interface">CLI</abbr>. For the GUI, download and double-click the
+JAR file
+
+## Building
+
+KeeCrack makes use of Gradle, so to build it yourself, you can just run
+
+    ./gradlew jfxJar
+
+## Contributing
+
+If you'd like to contribute, please fork the repository, make your changes, squash your commits, and send a pull request
+.
+
+## License
+
+KeeCrack is licensed under the Apache 2.0 license
+
+```
+Copyright 2018 William Brawner
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+```

build.gradle

@@ -0,0 +1,37 @@
+buildscript {
+    repositories {
+        jcenter()
+        mavenCentral()
+    }
+    dependencies {
+        classpath 'com.github.jengelman.gradle.plugins:shadow:2.0.3'
+        classpath group: 'de.dynamicfiles.projects.gradle.plugins', name: 'javafx-gradle-plugin', version: '8.8.2'
+    }
+}
+
+allprojects {
+    apply plugin: 'idea'
+}
+
+subprojects {
+    group 'com.wbrawner'
+    version '1.0-SNAPSHOT'
+
+    apply plugin: 'com.github.johnrengelman.shadow'
+    apply plugin: 'java'
+
+    sourceCompatibility = 1.8
+
+    repositories {
+        mavenCentral()
+    }
+
+    ext {
+        mockitoVersion = "2.18.0"
+    }
+
+    dependencies {
+        testImplementation group: 'junit', name: 'junit', version: '4.12'
+        testImplementation group: 'org.mockito', name: 'mockito-core', version: "$mockitoVersion"
+    }
+}

gradle/wrapper/gradle-wrapper.properties

@@ -0,0 +1,6 @@
+#Sat Apr 07 16:39:20 CDT 2018
+distributionBase=GRADLE_USER_HOME
+distributionPath=wrapper/dists
+zipStoreBase=GRADLE_USER_HOME
+zipStorePath=wrapper/dists
+distributionUrl=https\://services.gradle.org/distributions/gradle-4.0-all.zip

gradlew

@@ -0,0 +1,172 @@
+#!/usr/bin/env sh
+
+##############################################################################
+##
+##  Gradle start up script for UN*X
+##
+##############################################################################
+
+# Attempt to set APP_HOME
+# Resolve links: $0 may be a link
+PRG="$0"
+# Need this for relative symlinks.
+while [ -h "$PRG" ] ; do
+    ls=`ls -ld "$PRG"`
+    link=`expr "$ls" : '.*-> \(.*\)$'`
+    if expr "$link" : '/.*' > /dev/null; then
+        PRG="$link"
+    else
+        PRG=`dirname "$PRG"`"/$link"
+    fi
+done
+SAVED="`pwd`"
+cd "`dirname \"$PRG\"`/" >/dev/null
+APP_HOME="`pwd -P`"
+cd "$SAVED" >/dev/null
+
+APP_NAME="Gradle"
+APP_BASE_NAME=`basename "$0"`
+
+# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
+DEFAULT_JVM_OPTS=""
+
+# Use the maximum available, or set MAX_FD != -1 to use that value.
+MAX_FD="maximum"
+
+warn () {
+    echo "$*"
+}
+
+die () {
+    echo
+    echo "$*"
+    echo
+    exit 1
+}
+
+# OS specific support (must be 'true' or 'false').
+cygwin=false
+msys=false
+darwin=false
+nonstop=false
+case "`uname`" in
+  CYGWIN* )
+    cygwin=true
+    ;;
+  Darwin* )
+    darwin=true
+    ;;
+  MINGW* )
+    msys=true
+    ;;
+  NONSTOP* )
+    nonstop=true
+    ;;
+esac
+
+CLASSPATH=$APP_HOME/gradle/wrapper/gradle-wrapper.jar
+
+# Determine the Java command to use to start the JVM.
+if [ -n "$JAVA_HOME" ] ; then
+    if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
+        # IBM's JDK on AIX uses strange locations for the executables
+        JAVACMD="$JAVA_HOME/jre/sh/java"
+    else
+        JAVACMD="$JAVA_HOME/bin/java"
+    fi
+    if [ ! -x "$JAVACMD" ] ; then
+        die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
+
+Please set the JAVA_HOME variable in your environment to match the
+location of your Java installation."
+    fi
+else
+    JAVACMD="java"
+    which java >/dev/null 2>&1 || die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
+
+Please set the JAVA_HOME variable in your environment to match the
+location of your Java installation."
+fi
+
+# Increase the maximum file descriptors if we can.
+if [ "$cygwin" = "false" -a "$darwin" = "false" -a "$nonstop" = "false" ] ; then
+    MAX_FD_LIMIT=`ulimit -H -n`
+    if [ $? -eq 0 ] ; then
+        if [ "$MAX_FD" = "maximum" -o "$MAX_FD" = "max" ] ; then
+            MAX_FD="$MAX_FD_LIMIT"
+        fi
+        ulimit -n $MAX_FD
+        if [ $? -ne 0 ] ; then
+            warn "Could not set maximum file descriptor limit: $MAX_FD"
+        fi
+    else
+        warn "Could not query maximum file descriptor limit: $MAX_FD_LIMIT"
+    fi
+fi
+
+# For Darwin, add options to specify how the application appears in the dock
+if $darwin; then
+    GRADLE_OPTS="$GRADLE_OPTS \"-Xdock:name=$APP_NAME\" \"-Xdock:icon=$APP_HOME/media/gradle.icns\""
+fi
+
+# For Cygwin, switch paths to Windows format before running java
+if $cygwin ; then
+    APP_HOME=`cygpath --path --mixed "$APP_HOME"`
+    CLASSPATH=`cygpath --path --mixed "$CLASSPATH"`
+    JAVACMD=`cygpath --unix "$JAVACMD"`
+
+    # We build the pattern for arguments to be converted via cygpath
+    ROOTDIRSRAW=`find -L / -maxdepth 1 -mindepth 1 -type d 2>/dev/null`
+    SEP=""
+    for dir in $ROOTDIRSRAW ; do
+        ROOTDIRS="$ROOTDIRS$SEP$dir"
+        SEP="|"
+    done
+    OURCYGPATTERN="(^($ROOTDIRS))"
+    # Add a user-defined pattern to the cygpath arguments
+    if [ "$GRADLE_CYGPATTERN" != "" ] ; then
+        OURCYGPATTERN="$OURCYGPATTERN|($GRADLE_CYGPATTERN)"
+    fi
+    # Now convert the arguments - kludge to limit ourselves to /bin/sh
+    i=0
+    for arg in "$@" ; do
+        CHECK=`echo "$arg"|egrep -c "$OURCYGPATTERN" -`
+        CHECK2=`echo "$arg"|egrep -c "^-"`                                 ### Determine if an option
+
+        if [ $CHECK -ne 0 ] && [ $CHECK2 -eq 0 ] ; then                    ### Added a condition
+            eval `echo args$i`=`cygpath --path --ignore --mixed "$arg"`
+        else
+            eval `echo args$i`="\"$arg\""
+        fi
+        i=$((i+1))
+    done
+    case $i in
+        (0) set -- ;;
+        (1) set -- "$args0" ;;
+        (2) set -- "$args0" "$args1" ;;
+        (3) set -- "$args0" "$args1" "$args2" ;;
+        (4) set -- "$args0" "$args1" "$args2" "$args3" ;;
+        (5) set -- "$args0" "$args1" "$args2" "$args3" "$args4" ;;
+        (6) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" ;;
+        (7) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" ;;
+        (8) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" ;;
+        (9) set -- "$args0" "$args1" "$args2" "$args3" "$args4" "$args5" "$args6" "$args7" "$args8" ;;
+    esac
+fi
+
+# Escape application args
+save () {
+    for i do printf %s\\n "$i" | sed "s/'/'\\\\''/g;1s/^/'/;\$s/\$/' \\\\/" ; done
+    echo " "
+}
+APP_ARGS=$(save "$@")
+
+# Collect all arguments for the java command, following the shell quoting and substitution rules
+eval set -- $DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS "\"-Dorg.gradle.appname=$APP_BASE_NAME\"" -classpath "\"$CLASSPATH\"" org.gradle.wrapper.GradleWrapperMain "$APP_ARGS"
+
+# by default we should be in the correct project dir, but when run from Finder on Mac, the cwd is wrong
+if [ "$(uname)" = "Darwin" ] && [ "$HOME" = "$PWD" ]; then
+  cd "$(dirname "$0")"
+fi
+
+exec "$JAVACMD" "$@"

gradlew.bat

@@ -0,0 +1,84 @@
+@if "%DEBUG%" == "" @echo off
+@rem ##########################################################################
+@rem
+@rem  Gradle startup script for Windows
+@rem
+@rem ##########################################################################
+
+@rem Set local scope for the variables with windows NT shell
+if "%OS%"=="Windows_NT" setlocal
+
+set DIRNAME=%~dp0
+if "%DIRNAME%" == "" set DIRNAME=.
+set APP_BASE_NAME=%~n0
+set APP_HOME=%DIRNAME%
+
+@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
+set DEFAULT_JVM_OPTS=
+
+@rem Find java.exe
+if defined JAVA_HOME goto findJavaFromJavaHome
+
+set JAVA_EXE=java.exe
+%JAVA_EXE% -version >NUL 2>&1
+if "%ERRORLEVEL%" == "0" goto init
+
+echo.
+echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
+echo.
+echo Please set the JAVA_HOME variable in your environment to match the
+echo location of your Java installation.
+
+goto fail
+
+:findJavaFromJavaHome
+set JAVA_HOME=%JAVA_HOME:"=%
+set JAVA_EXE=%JAVA_HOME%/bin/java.exe
+
+if exist "%JAVA_EXE%" goto init
+
+echo.
+echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME%
+echo.
+echo Please set the JAVA_HOME variable in your environment to match the
+echo location of your Java installation.
+
+goto fail
+
+:init
+@rem Get command-line arguments, handling Windows variants
+
+if not "%OS%" == "Windows_NT" goto win9xME_args
+
+:win9xME_args
+@rem Slurp the command line arguments.
+set CMD_LINE_ARGS=
+set _SKIP=2
+
+:win9xME_args_slurp
+if "x%~1" == "x" goto execute
+
+set CMD_LINE_ARGS=%*
+
+:execute
+@rem Setup the command line
+
+set CLASSPATH=%APP_HOME%\gradle\wrapper\gradle-wrapper.jar
+
+@rem Execute Gradle
+"%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -classpath "%CLASSPATH%" org.gradle.wrapper.GradleWrapperMain %CMD_LINE_ARGS%
+
+:end
+@rem End local scope for the variables with windows NT shell
+if "%ERRORLEVEL%"=="0" goto mainEnd
+
+:fail
+rem Set variable GRADLE_EXIT_CONSOLE if you need the _script_ return code instead of
+rem the _cmd.exe /c_ return code!
+if  not "" == "%GRADLE_EXIT_CONSOLE%" exit 1
+exit /b 1
+
+:mainEnd
+if "%OS%"=="Windows_NT" endlocal
+
+:omega

keecrack-cli/build.gradle

@@ -0,0 +1,13 @@
+dependencies {
+    compile project(':keecrack-lib')
+    compile project(':keecrack-gui')
+    implementation group: 'net.sourceforge.argparse4j', name: 'argparse4j', version: '0.8.1'
+}
+
+jar {
+    manifest {
+        attributes (
+                'Main-Class': 'com.wbrawner.keecrack.cli.Main'
+        )
+    }
+}

keecrack-cli/src/main/java/com/wbrawner/keecrack/cli/Main.java

@@ -0,0 +1,122 @@
+package com.wbrawner.keecrack.cli;
+
+import com.wbrawner.keecrack.lib.Code;
+import com.wbrawner.keecrack.lib.KeeCrack;
+import com.wbrawner.keecrack.lib.view.CrackingView;
+import net.sourceforge.argparse4j.ArgumentParsers;
+import net.sourceforge.argparse4j.impl.Arguments;
+import net.sourceforge.argparse4j.inf.ArgumentParser;
+import net.sourceforge.argparse4j.inf.ArgumentParserException;
+import net.sourceforge.argparse4j.inf.Namespace;
+
+import java.io.File;
+import java.time.Duration;
+import java.util.Locale;
+
+public class Main {
+    private static boolean isVerbose = false;
+
+    public static void main(String[] args) {
+        ArgumentParser parser = ArgumentParsers.newFor("KeeCrack")
+                .build()
+                .description("Brute force KeePass database files");
+        parser.addArgument("--verbose", "-v")
+                .help("Increase logging output")
+                .action(Arguments.storeTrue())
+                .dest("verbose");
+        parser.addArgument("--gui")
+                .action(Arguments.storeTrue())
+                .help("launch the graphical interface (ignores other options)")
+                .dest("gui");
+        parser.addArgument("--word-list", "-w")
+                .help("a file containing newline-separated words to use as the passwords")
+                .dest("wordlist")
+                .metavar("WORD-LIST-FILE");
+        parser.addArgument("--key-file", "-k")
+                .help("the key file to use with the database")
+                .dest("keyfile")
+                .metavar("KEY-FILE");
+
+        parser.addArgument("DATABASE")
+                .dest("database")
+                .help("the database file to brute force");
+
+        try {
+            Namespace res = parser.parseArgs(args);
+            isVerbose = res.getBoolean("verbose");
+            KeeCrack keeCrack = KeeCrack.getInstance();
+            keeCrack.setCrackingView(new CLICrackingView());
+
+            String databasePath = res.getString("database");
+            if (databasePath != null) {
+                keeCrack.setDatabaseFile(new File(databasePath));
+            }
+
+            String keyfilePath = res.getString("keyfile");
+            if (keyfilePath != null) {
+                keeCrack.setKeyFile(new File(keyfilePath));
+            }
+
+            String wordlistPath = res.getString("wordlist");
+            if (wordlistPath != null) ;
+            keeCrack.setWordlistFile(new File(wordlistPath));
+
+            if (res.getBoolean("gui")) {
+                com.wbrawner.keecrack.gui.Main.main(new String[]{});
+            } else {
+                keeCrack.attack();
+            }
+        } catch (ArgumentParserException e) {
+            parser.handleError(e);
+        }
+    }
+
+    static class CLICrackingView implements CrackingView {
+        @Override
+        public void onPasswordGuess(String password) {
+            if (isVerbose)
+                System.out.println("Guessing password: " + password);
+        }
+
+        @Override
+        public void onResult(String password, int guessCount, Duration timeElapsed) {
+            String message;
+            if (password == null)
+                message = String.format(
+                        Locale.US,
+                        "Unable to guess password after %d attempts in %s",
+                        guessCount,
+                        timeElapsed.toString().toLowerCase().substring(2)
+                );
+            else
+                message = String.format(
+                        "Successfully guessed password after %d attempts in %s: %s",
+                        guessCount,
+                        timeElapsed.toString().toLowerCase().substring(2),
+                        password
+                );
+            System.out.println(message);
+        }
+
+        @Override
+        public void onError(Code code) {
+            String message = "";
+            switch (code) {
+                case ERROR_MISSING_DATABASE_FILE:
+                    message = "Please specify a database file that you have read access to";
+                    break;
+                case ERROR_MISSING_WORD_LIST_FILE:
+                    message = "Please specify a word list file that you have read access to";
+                    break;
+                case ERROR_CRACKING_INTERRUPTED:
+                    message = "Aborted.";
+                    break;
+                case ERROR_FILE_READ:
+                    message = "An error occurred while trying to read one of the files";
+                    break;
+            }
+            System.err.println(message);
+            System.exit(code.ordinal());
+        }
+    }
+}

keecrack-gui/build.gradle

@@ -0,0 +1,22 @@
+dependencies {
+    compile project(':keecrack-lib')
+}
+
+apply plugin: 'javafx-gradle-plugin'
+
+jar {
+    manifest {
+        attributes (
+                'Main-Class': 'com.wbrawner.keecrack.gui.Main'
+        )
+    }
+}
+
+jfx {
+    // minimal requirement for jfxJar-task
+    mainClass = 'com.wbrawner.keecrack.gui.Main'
+
+    // minimal requirement for jfxNative-task
+    vendor = 'William Brawner'
+}
+

keecrack-gui/src/main/java/com/wbrawner/keecrack/gui/CrackingController.java

@@ -0,0 +1,97 @@
+package com.wbrawner.keecrack.gui;
+
+import com.wbrawner.keecrack.lib.Code;
+import com.wbrawner.keecrack.lib.KeeCrack;
+import com.wbrawner.keecrack.lib.view.CrackingView;
+import javafx.application.Platform;
+import javafx.fxml.FXML;
+import javafx.fxml.Initializable;
+import javafx.scene.control.Label;
+import javafx.scene.control.ProgressIndicator;
+import javafx.stage.Stage;
+
+import java.net.URL;
+import java.time.Duration;
+import java.util.Locale;
+import java.util.ResourceBundle;
+
+public class CrackingController implements Initializable, CrackingView {
+    @FXML
+    private Label passwordLabel;
+    @FXML
+    private Label password;
+    @FXML
+    private ProgressIndicator progress;
+    @FXML
+    private Label timeElapsed;
+
+    private Stage stage;
+
+    @Override
+    public void initialize(URL location, ResourceBundle resources) {
+        final KeeCrack keeCrack = KeeCrack.getInstance();
+        keeCrack.setCrackingView(this);
+        new Thread(keeCrack::attack).start();
+    }
+
+    @Override
+    public void onPasswordGuess(String password) {
+        Platform.runLater(() -> setPassword(password));
+    }
+
+    @Override
+    public void onResult(final String password, final int guessCount, final Duration timeElapsed) {
+        Platform.runLater(() -> {
+            progress.setVisible(false);
+            setTimeElapsed(timeElapsed);
+            String message;
+            String title;
+            if (password != null) {
+                message = String.format(
+                        Locale.US,
+                        "Cracked in %d attempts.",
+                        guessCount
+                );
+                title = "Keecrack - Complete";
+            } else {
+                message = String.format(
+                        Locale.US,
+                        "Failed to crack password in %d attempts.",
+                        guessCount
+                );
+                title = "Keecrack - Failed";
+            }
+            setPasswordLabel(message);
+            setPassword(password);
+            stage.setTitle(title);
+        });
+    }
+
+    @Override
+    public void onError(Code code) {
+
+    }
+
+    private void onClose() {
+        KeeCrack keeCrack = KeeCrack.getInstance();
+        keeCrack.abort();
+        keeCrack.setCrackingView(null);
+    }
+
+    private void setPasswordLabel(String text) {
+        passwordLabel.setText(text);
+    }
+
+    private void setPassword(String text) {
+        password.setText(text);
+    }
+
+    private void setTimeElapsed(Duration timeElapsed) {
+        this.timeElapsed.setText(timeElapsed.toString().toLowerCase().substring(2));
+    }
+
+    public void setStage(Stage stage) {
+        this.stage = stage;
+        stage.setOnHidden(event -> onClose());
+    }
+}

keecrack-gui/src/main/java/com/wbrawner/keecrack/gui/Main.java

@@ -0,0 +1,24 @@
+package com.wbrawner.keecrack.gui;
+
+import javafx.application.Application;
+import javafx.fxml.FXMLLoader;
+import javafx.scene.Parent;
+import javafx.scene.Scene;
+import javafx.stage.Stage;
+
+import java.io.IOException;
+
+public class Main extends Application {
+
+    @Override
+    public void start(Stage primaryStage) throws IOException {
+        Parent root = FXMLLoader.load(getClass().getResource("/fxml/main.fxml"));
+        primaryStage.setTitle("Keecrack");
+        primaryStage.setScene(new Scene(root, 400, 400));
+        primaryStage.show();
+    }
+
+    public static void main(String[] args) {
+        launch(args);
+    }
+}

keecrack-gui/src/main/java/com/wbrawner/keecrack/gui/MainController.java

@@ -0,0 +1,129 @@
+package com.wbrawner.keecrack.gui;
+
+import com.wbrawner.keecrack.lib.Code;
+import com.wbrawner.keecrack.lib.KeeCrack;
+import com.wbrawner.keecrack.lib.view.FormView;
+import javafx.fxml.FXML;
+import javafx.fxml.FXMLLoader;
+import javafx.fxml.Initializable;
+import javafx.scene.Parent;
+import javafx.scene.Scene;
+import javafx.scene.control.Button;
+import javafx.scene.control.TextField;
+import javafx.stage.FileChooser;
+import javafx.stage.Modality;
+import javafx.stage.Stage;
+
+import java.io.File;
+import java.io.IOException;
+import java.net.URL;
+import java.util.ResourceBundle;
+
+public class MainController implements Initializable, FormView {
+
+    @FXML private TextField database;
+    @FXML private TextField key;
+    @FXML private TextField wordlist;
+    @FXML private Button crackButton;
+
+    @Override
+    public void initialize(URL location, ResourceBundle resources) {
+        KeeCrack keeCrack = KeeCrack.getInstance();
+        keeCrack.setFormView(this);
+        database.setOnMouseClicked(event -> {
+            if (KeeCrack.getInstance().isCracking()) {
+                return;
+            }
+
+            File databaseFile = getFile("KeePass Database Files", "kdbx");
+            keeCrack.setDatabaseFile(databaseFile);
+        });
+
+        key.setOnMouseClicked(event -> {
+            if (KeeCrack.getInstance().isCracking()) {
+                return;
+            }
+
+            File keyFile = getFile("KeePass Database Key Files", "*");
+            keeCrack.setKeyFile(keyFile);
+        });
+
+        wordlist.setOnMouseClicked(event -> {
+            if (KeeCrack.getInstance().isCracking()) {
+                return;
+            }
+
+            File wordlistFile = getFile("Text Files", "txt");
+            keeCrack.setWordlistFile(wordlistFile);
+        });
+
+        crackButton.setOnMouseClicked(event -> {
+            try {
+                if (KeeCrack.getInstance().isCracking()) {
+                    return;
+                }
+                Stage stage = new Stage();
+                stage.setTitle("Keecrack - Cracking...");
+                stage.initModality(Modality.WINDOW_MODAL);
+                FXMLLoader loader = new FXMLLoader(getClass().getResource("/fxml/cracking.fxml"));
+                loader.setControllerFactory(param -> {
+                    try {
+                        Object controller = param.newInstance();
+                        if (controller instanceof CrackingController) {
+                            ((CrackingController) controller).setStage(stage);
+                        }
+                        return controller;
+                    } catch (InstantiationException | IllegalAccessException e) {
+                        e.printStackTrace();
+                    }
+                    return null;
+                });
+                Parent root = loader.load();
+                stage.setScene(new Scene(root, 200, 200));
+                stage.showAndWait();
+            } catch (IOException e) {
+                e.printStackTrace();
+            }
+        });
+
+        if (keeCrack.getDatabaseFile() != null)
+            onDatabaseFileSet(keeCrack.getDatabaseFile().getName());
+        if (keeCrack.getKeyFile() != null)
+            onKeyFileSet(keeCrack.getKeyFile().getName());
+        if (keeCrack.getWordlistFile() != null)
+            onWordListFileSet(keeCrack.getWordlistFile().getName());
+    }
+
+
+    private File getFile(String description, String... extensions) {
+        FileChooser fileChooser = new FileChooser();
+        fileChooser.setInitialDirectory(new File(System.getProperty("user.home")));
+        fileChooser.setTitle("Keecrack - Select a file");
+        fileChooser.setSelectedExtensionFilter(
+                new FileChooser.ExtensionFilter(description, extensions)
+        );
+        Stage stage = new Stage();
+        stage.setOnHidden(e -> KeeCrack.getInstance().setFormView(null));
+        return fileChooser.showOpenDialog(stage);
+    }
+
+    @Override
+    public void onDatabaseFileSet(String name) {
+        database.setText(name);
+    }
+
+    @Override
+    public void onKeyFileSet(String name) {
+        key.setText(name);
+    }
+
+    @Override
+    public void onWordListFileSet(String name) {
+        wordlist.setText(name);
+    }
+
+    @Override
+    public void onError(Code code) {
+
+    }
+}

keecrack-gui/src/main/resources/fxml/cracking.fxml

@@ -0,0 +1,18 @@
+<?xml version="1.0" encoding="UTF-8"?>
+
+<?import javafx.scene.control.Label?>
+<?import javafx.scene.control.ProgressIndicator?>
+<?import javafx.scene.layout.*?>
+<AnchorPane xmlns:fx="http://javafx.com/fxml/1" prefHeight="400.0" prefWidth="600.0" xmlns="http://javafx.com/javafx/8"
+            fx:controller="com.wbrawner.keecrack.gui.CrackingController">
+   <children>
+      <VBox alignment="CENTER" prefHeight="200.0" prefWidth="100.0" AnchorPane.bottomAnchor="0.0" AnchorPane.leftAnchor="0.0" AnchorPane.rightAnchor="0.0" AnchorPane.topAnchor="0.0">
+         <children>
+            <ProgressIndicator fx:id="progress" />
+            <Label fx:id="timeElapsed" />
+            <Label fx:id="passwordLabel" text="Trying password: " />
+            <Label fx:id="password" />
+         </children>
+      </VBox>
+   </children>
+</AnchorPane>

keecrack-gui/src/main/resources/fxml/main.fxml

@@ -0,0 +1,61 @@
+<?xml version="1.0" encoding="UTF-8"?>
+
+<?import javafx.geometry.Insets?>
+<?import javafx.scene.control.*?>
+<?import javafx.scene.Cursor?>
+<?import javafx.scene.layout.*?>
+<AnchorPane xmlns:fx="http://javafx.com/fxml/1" prefHeight="400.0" prefWidth="600.0" xmlns="http://javafx.com/javafx/8"
+            fx:controller="com.wbrawner.keecrack.gui.MainController">
+   <children>
+      <VBox alignment="CENTER" prefHeight="200.0" prefWidth="100.0" AnchorPane.bottomAnchor="0.0" AnchorPane.leftAnchor="0.0" AnchorPane.rightAnchor="0.0" AnchorPane.topAnchor="0.0">
+         <children>
+            <GridPane alignment="CENTER" prefHeight="90.0" prefWidth="539.0">
+              <columnConstraints>
+                <ColumnConstraints halignment="RIGHT" minWidth="100.0" prefWidth="20.0" />
+                <ColumnConstraints hgrow="SOMETIMES" minWidth="10.0" prefWidth="100.0" />
+              </columnConstraints>
+              <rowConstraints>
+                <RowConstraints minHeight="10.0" prefHeight="30.0" vgrow="SOMETIMES" />
+                <RowConstraints minHeight="10.0" prefHeight="30.0" vgrow="SOMETIMES" />
+                <RowConstraints minHeight="10.0" prefHeight="30.0" vgrow="SOMETIMES" />
+              </rowConstraints>
+               <children>
+                  <Label alignment="CENTER_RIGHT" text="Database" />
+                   <TextField fx:id="database" editable="false" promptText="Select" GridPane.columnIndex="1">
+                     <cursor>
+                        <Cursor fx:constant="HAND" />
+                     </cursor>
+                     <GridPane.margin>
+                        <Insets left="20.0" />
+                     </GridPane.margin>
+                  </TextField>
+                  <Label text="Key File" GridPane.rowIndex="1" />
+                   <TextField fx:id="key" editable="false" promptText="Select (Optional)" GridPane.columnIndex="1"
+                              GridPane.rowIndex="1">
+                     <cursor>
+                        <Cursor fx:constant="HAND" />
+                     </cursor>
+                     <GridPane.margin>
+                        <Insets left="20.0" />
+                     </GridPane.margin>
+                  </TextField>
+                   <TextField fx:id="wordlist" editable="false" promptText="Select" GridPane.columnIndex="1"
+                              GridPane.rowIndex="2">
+                     <cursor>
+                        <Cursor fx:constant="HAND" />
+                     </cursor>
+                     <GridPane.margin>
+                        <Insets left="20.0" />
+                     </GridPane.margin>
+                  </TextField>
+                  <Label text="Word List" GridPane.rowIndex="2" />
+               </children>
+               <padding>
+                  <Insets left="20.0" right="20.0" />
+               </padding>
+            </GridPane>
+            <Button fx:id="crackButton" mnemonicParsing="false" prefHeight="25.0" prefWidth="111.0" text="Crack" />
+         </children>
+      </VBox>
+   </children>
+</AnchorPane>

keecrack-lib/build.gradle

@@ -0,0 +1,21 @@
+apply plugin: 'jacoco'
+
+dependencies {
+    implementation group: 'org.linguafranca.pwdb', name: 'KeePassJava2-kdbx', version: '2.1.4'
+}
+
+jacocoTestReport {
+    reports {
+        xml.enabled true
+        html.enabled true
+        html.destination file("${buildDir}/jacoco")
+    }
+}
+
+jar {
+    manifest {
+        attributes (
+            'Main-Class': 'com.wbrawner.keecrack.lib.KeeCrack'
+        )
+    }
+}

keecrack-lib/src/main/java/com/wbrawner/keecrack/lib/Code.java

@@ -0,0 +1,12 @@
+package com.wbrawner.keecrack.lib;
+
+/**
+ * Rather than simple string error messages, an enum is used so that the error messages can be translated by any UI
+ */
+public enum Code {
+    ERROR_FILE_READ,
+    ERROR_CRACKING_INTERRUPTED,
+    ERROR_CRACKING_IN_PROGRESS,
+    ERROR_MISSING_DATABASE_FILE,
+    ERROR_MISSING_WORD_LIST_FILE
+}

keecrack-lib/src/main/java/com/wbrawner/keecrack/lib/KeeCrack.java

@@ -0,0 +1,206 @@
+package com.wbrawner.keecrack.lib;
+
+import com.wbrawner.keecrack.lib.view.CrackingView;
+import com.wbrawner.keecrack.lib.view.FormView;
+import org.linguafranca.pwdb.kdbx.KdbxCreds;
+import org.linguafranca.pwdb.kdbx.stream_3_1.KdbxHeader;
+import org.linguafranca.pwdb.kdbx.stream_3_1.KdbxSerializer;
+
+import java.io.*;
+import java.lang.ref.WeakReference;
+import java.time.Duration;
+import java.time.Instant;
+import java.util.Objects;
+import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicReference;
+
+public class KeeCrack {
+    private static final AtomicReference<KeeCrack> singleton = new AtomicReference<>(null);
+    private final Object keyFileLock = new Object();
+    private WeakReference<FormView> formView = new WeakReference<>(null);
+    private WeakReference<CrackingView> crackingView = new WeakReference<>(null);
+    private final AtomicBoolean isCracking = new AtomicBoolean(false);
+    private File databaseFile;
+    private File keyFile;
+    private File wordlistFile;
+
+    /**
+     * This is used to abort cracking
+     */
+    private final AtomicBoolean abort = new AtomicBoolean(false);
+
+    private int guessCount = 0;
+
+    private KeeCrack() {
+    }
+
+    public static KeeCrack getInstance() {
+        if (singleton.get() == null) {
+            singleton.set(new KeeCrack());
+        }
+
+        return singleton.get();
+    }
+
+    /**
+     * Call this to reset the state of the KeeCrack instance. Note that you will need to set the views again after
+     * calling this
+     */
+    public void reset() {
+        setDatabaseFile(null);
+        setKeyFile(null);
+        setWordlistFile(null);
+        setCrackingView(null);
+        setFormView(null);
+    }
+
+    public void abort() {
+        abort.set(true);
+    }
+
+    /**
+     * Call this to begin brute-forcing the database file.
+     */
+    public void attack() {
+        if (databaseFile == null || !databaseFile.exists() || !databaseFile.canRead()) {
+            sendErrorCode(Code.ERROR_MISSING_DATABASE_FILE);
+            return;
+        }
+
+        if (wordlistFile == null || !wordlistFile.exists() || !wordlistFile.canRead()) {
+            sendErrorCode(Code.ERROR_MISSING_WORD_LIST_FILE);
+            return;
+        }
+
+        if (isCracking.get()) {
+            sendErrorCode(Code.ERROR_CRACKING_IN_PROGRESS);
+            return;
+        }
+        isCracking.set(true);
+
+        guessCount = 0;
+        Instant startTime = Instant.now();
+
+        try (BufferedReader wordReader = new BufferedReader(new FileReader(wordlistFile))) {
+            String line = null;
+            boolean haveCorrectPassword = false;
+            while (!haveCorrectPassword && (line = wordReader.readLine()) != null) {
+                if (abort.get()) {
+                    sendErrorCode(Code.ERROR_CRACKING_INTERRUPTED);
+                    abort.set(false);
+                    return;
+                }
+                CrackingView view = crackingView.get();
+                if (view != null)
+                    view.onPasswordGuess(line);
+                haveCorrectPassword = guessPassword(line);
+            }
+
+            CrackingView view = crackingView.get();
+            if (view != null) {
+                Duration duration = Duration.between(startTime, Instant.now());
+                String password = null;
+                if (haveCorrectPassword) {
+                    password = line;
+                }
+                view.onResult(password, guessCount, duration);
+            }
+        } catch (IOException e) {
+            e.printStackTrace();
+            sendErrorCode(Code.ERROR_FILE_READ);
+        } finally {
+            isCracking.set(false);
+        }
+    }
+
+    private boolean guessPassword(String password) {
+        guessCount++;
+        try (InputStream inputStream = new FileInputStream(databaseFile)) {
+            KdbxHeader databaseHeader = new KdbxHeader();
+            KdbxCreds credentials;
+            synchronized (keyFileLock) {
+                if (keyFile == null) {
+                    credentials = new KdbxCreds(password.getBytes());
+                } else {
+                    credentials = new KdbxCreds(password.getBytes(), new FileInputStream(keyFile));
+                }
+            }
+            KdbxSerializer.createUnencryptedInputStream(credentials, databaseHeader, inputStream);
+            return true;
+        } catch (IllegalStateException ignored) {
+            // This happens when an incorrect guess occurs. Expected behavior, so we ignore it
+        } catch (Exception e) {
+            e.printStackTrace();
+        }
+        return false;
+    }
+
+
+    void sendErrorCode(Code code) {
+        try {
+            Objects.requireNonNull(crackingView.get()).onError(code);
+        } catch (NullPointerException ignored) {
+        }
+        try {
+            Objects.requireNonNull(formView.get()).onError(code);
+        } catch (NullPointerException ignored) {
+        }
+    }
+
+
+    public File getDatabaseFile() {
+        return databaseFile;
+    }
+
+    public void setDatabaseFile(File databaseFile) {
+        this.databaseFile = databaseFile;
+        try {
+            String response = (databaseFile == null) ? null : databaseFile.getName();
+            //noinspection ConstantConditions
+            formView.get().onDatabaseFileSet(response);
+        } catch (NullPointerException ignored) {
+        }
+    }
+
+    public File getKeyFile() {
+        return keyFile;
+    }
+
+    public void setKeyFile(File keyFile) {
+        synchronized (keyFileLock) {
+            this.keyFile = keyFile;
+        }
+        try {
+            String response = (keyFile == null) ? null : keyFile.getName();
+            //noinspection ConstantConditions
+            formView.get().onKeyFileSet(response);
+        } catch (NullPointerException ignored) {
+        }
+    }
+
+    public File getWordlistFile() {
+        return wordlistFile;
+    }
+
+    public void setWordlistFile(File wordlistFile) {
+        this.wordlistFile = wordlistFile;
+        try {
+            String response = (wordlistFile == null) ? null : wordlistFile.getName();
+            //noinspection ConstantConditions
+            formView.get().onWordListFileSet(response);
+        } catch (NullPointerException ignored) {
+        }
+    }
+
+    public void setFormView(FormView formView) {
+        this.formView = new WeakReference<>(formView);
+    }
+
+    public void setCrackingView(CrackingView crackingView) {
+        this.crackingView = new WeakReference<>(crackingView);
+    }
+
+    public boolean isCracking() {
+        return isCracking.get();
+    }
+}

keecrack-lib/src/main/java/com/wbrawner/keecrack/lib/view/BaseView.java

@@ -0,0 +1,7 @@
+package com.wbrawner.keecrack.lib.view;
+
+import com.wbrawner.keecrack.lib.Code;
+
+public interface BaseView {
+    void onError(Code code);
+}

keecrack-lib/src/main/java/com/wbrawner/keecrack/lib/view/CrackingView.java

@@ -0,0 +1,19 @@
+package com.wbrawner.keecrack.lib.view;
+
+import java.time.Duration;
+
+public interface CrackingView extends BaseView {
+    /**
+     * Called prior to each guess
+     * @param password The password that will be guessed
+     */
+    void onPasswordGuess(final String password);
+
+    /**
+     * Called when the password has been successfully guessed, or there are no more passwords to guess
+     * @param password The password, if successfully guessed, or null if no passwords were successful
+     * @param guessCount The number of passwords guessed
+     * @param timeElapsed The time taken to guess the password
+     */
+    void onResult(final String password, final int guessCount, final Duration timeElapsed);
+}

keecrack-lib/src/main/java/com/wbrawner/keecrack/lib/view/FormView.java

@@ -0,0 +1,7 @@
+package com.wbrawner.keecrack.lib.view;
+
+public interface FormView extends BaseView {
+    void onDatabaseFileSet(String name);
+    void onKeyFileSet(String name);
+    void onWordListFileSet(String name);
+}

keecrack-lib/src/test/java/com/wbrawner/keecrack/lib/KeeCrackTest.java

@@ -0,0 +1,174 @@
+package com.wbrawner.keecrack.lib;
+
+import com.wbrawner.keecrack.lib.view.CrackingView;
+import com.wbrawner.keecrack.lib.view.FormView;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.io.File;
+import java.io.IOException;
+import java.time.Duration;
+
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertNull;
+import static org.mockito.ArgumentMatchers.anyString;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.*;
+
+public class KeeCrackTest {
+    private CrackingView mockCrackingView;
+    private FormView mockFormView;
+    private KeeCrack keeCrack;
+
+    @Before
+    public void setUp() {
+        mockCrackingView = mock(CrackingView.class);
+        mockFormView = mock(FormView.class);
+        keeCrack = KeeCrack.getInstance();
+    }
+
+    @After
+    public void tearDown() {
+        keeCrack.reset();
+        Utils.rmdir(Utils.getTmpDir());
+    }
+
+    @Test
+    public void resetTest() {
+        keeCrack.setDatabaseFile(new File("Database"));
+        keeCrack.setKeyFile(new File("Keyfile"));
+        keeCrack.setWordlistFile(new File("WordList"));
+        assertNotNull(keeCrack.getDatabaseFile());
+        assertNotNull(keeCrack.getKeyFile());
+        assertNotNull(keeCrack.getWordlistFile());
+        keeCrack.reset();
+        assertNull(keeCrack.getDatabaseFile());
+        assertNull(keeCrack.getKeyFile());
+        assertNull(keeCrack.getWordlistFile());
+    }
+
+    @Test
+    public void abortTest() throws IOException {
+        keeCrack.setDatabaseFile(Utils.getDatabase("123456.kdbx"));
+        keeCrack.setWordlistFile(Utils.getWordList("valid-words.txt"));
+        keeCrack.abort();
+        keeCrack.setCrackingView(mockCrackingView);
+        keeCrack.attack();
+        verify(mockCrackingView, times(1)).onError(Code.ERROR_CRACKING_INTERRUPTED);
+    }
+
+    @Test
+    public void sendErrorTest() {
+        keeCrack.setCrackingView(mockCrackingView);
+        keeCrack.setFormView(mockFormView);
+        keeCrack.sendErrorCode(Code.ERROR_CRACKING_IN_PROGRESS);
+        verify(mockCrackingView, times(1)).onError(Code.ERROR_CRACKING_IN_PROGRESS);
+        verify(mockFormView, times(1)).onError(Code.ERROR_CRACKING_IN_PROGRESS);
+    }
+
+    @Test
+    public void sendErrorWithoutCrackingViewTest() {
+        keeCrack.setFormView(mockFormView);
+        keeCrack.sendErrorCode(Code.ERROR_CRACKING_IN_PROGRESS);
+        verify(mockFormView, times(1)).onError(Code.ERROR_CRACKING_IN_PROGRESS);
+    }
+
+    @Test
+    public void sendErrorWithoutFormViewTest() {
+        keeCrack.setCrackingView(mockCrackingView);
+        keeCrack.sendErrorCode(Code.ERROR_CRACKING_IN_PROGRESS);
+        verify(mockCrackingView, times(1)).onError(Code.ERROR_CRACKING_IN_PROGRESS);
+    }
+
+    @Test
+    public void errorWithoutDatabaseFileTest() throws IOException {
+        keeCrack.setWordlistFile(Utils.getWordList("valid-words.txt"));
+        keeCrack.setCrackingView(mockCrackingView);
+        keeCrack.attack();
+        verify(mockCrackingView, times(1)).onError(Code.ERROR_MISSING_DATABASE_FILE);
+    }
+
+    @Test
+    public void errorWithoutWordlistFileTest() throws IOException {
+        keeCrack.setDatabaseFile(Utils.getDatabase("123456.kdbx"));
+        keeCrack.setCrackingView(mockCrackingView);
+        keeCrack.attack();
+        verify(mockCrackingView, times(1)).onError(Code.ERROR_MISSING_WORD_LIST_FILE);
+    }
+
+
+    @Test
+    public void guessCorrectPasswordTest() throws IOException {
+        keeCrack.setDatabaseFile(Utils.getDatabase("123456.kdbx"));
+        keeCrack.setWordlistFile(Utils.getWordList("valid-words.txt"));
+        keeCrack.setCrackingView(mockCrackingView);
+        keeCrack.attack();
+        verify(mockCrackingView, times(1)).onPasswordGuess("123456");
+        verify(mockCrackingView, times(1)).onResult(eq("123456"), eq(1), any(Duration.class));
+    }
+
+    @Test
+    public void guessCorrectPasswordAndKeyTest() throws IOException {
+        keeCrack.setDatabaseFile(Utils.getDatabase("123456-key.kdbx"));
+        keeCrack.setKeyFile(Utils.getKeyFile());
+        keeCrack.setWordlistFile(Utils.getWordList("valid-words.txt"));
+        keeCrack.setCrackingView(mockCrackingView);
+        keeCrack.attack();
+        verify(mockCrackingView, times(1)).onPasswordGuess("123456");
+        verify(mockCrackingView, times(1)).onResult(eq("123456"), eq(1), any(Duration.class));
+    }
+
+    @Test
+    public void keepGuessingUntilCorrectPasswordTest() throws IOException {
+        keeCrack.setDatabaseFile(Utils.getDatabase("redwings.kdbx"));
+        keeCrack.setWordlistFile(Utils.getWordList("valid-words.txt"));
+        keeCrack.setCrackingView(mockCrackingView);
+        keeCrack.attack();
+        verify(mockCrackingView, times(2)).onPasswordGuess(anyString());
+        verify(mockCrackingView, times(1)).onResult(eq("redwings"), eq(2), any(Duration.class));
+    }
+
+    @Test
+    public void keepGuessingUntilCorrectPasswordAndKeyTest() throws IOException {
+        keeCrack.setDatabaseFile(Utils.getDatabase("redwings-key.kdbx"));
+        keeCrack.setKeyFile(Utils.getKeyFile());
+        keeCrack.setWordlistFile(Utils.getWordList("valid-words.txt"));
+        keeCrack.setCrackingView(mockCrackingView);
+        keeCrack.attack();
+        verify(mockCrackingView, times(2)).onPasswordGuess(anyString());
+        verify(mockCrackingView, times(1)).onResult(eq("redwings"), eq(2), any(Duration.class));
+    }
+
+    @Test
+    public void failToCrackPasswordTest() throws IOException {
+        keeCrack.setDatabaseFile(Utils.getDatabase("123456.kdbx"));
+        keeCrack.setWordlistFile(Utils.getWordList("invalid-words.txt"));
+        keeCrack.setCrackingView(mockCrackingView);
+        keeCrack.attack();
+        verify(mockCrackingView, times(3)).onPasswordGuess(anyString());
+        verify(mockCrackingView, times(1)).onResult(eq(null), eq(3), any(Duration.class));
+    }
+
+    @Test
+    public void failToCrackInvalidPasswordAndValidKeyTest() throws IOException {
+        keeCrack.setDatabaseFile(Utils.getDatabase("123456-key.kdbx"));
+        keeCrack.setKeyFile(Utils.getKeyFile());
+        keeCrack.setWordlistFile(Utils.getWordList("invalid-words.txt"));
+        keeCrack.setCrackingView(mockCrackingView);
+        keeCrack.attack();
+        verify(mockCrackingView, times(3)).onPasswordGuess(anyString());
+        verify(mockCrackingView, times(1)).onResult(eq(null), eq(3), any(Duration.class));
+    }
+
+    @Test
+    public void failToCrackValidPasswordAndInvalidKeyTest() throws IOException {
+        keeCrack.setDatabaseFile(Utils.getDatabase("123456-key.kdbx"));
+        keeCrack.setKeyFile(Utils.getInvalidKeyFile());
+        keeCrack.setWordlistFile(Utils.getWordList("valid-words.txt"));
+        keeCrack.setCrackingView(mockCrackingView);
+        keeCrack.attack();
+        verify(mockCrackingView, times(3)).onPasswordGuess(anyString());
+        verify(mockCrackingView, times(1)).onResult(eq(null), eq(3), any(Duration.class));
+    }
+}

keecrack-lib/src/test/java/com/wbrawner/keecrack/lib/Utils.java

@@ -0,0 +1,63 @@
+package com.wbrawner.keecrack.lib;
+
+import java.io.*;
+
+@SuppressWarnings("ResultOfMethodCallIgnored")
+class Utils {
+    private static final File tmpDir = new File(
+            System.getProperty("java.io.tmpdir"),
+            "KeeCrackTests-" + System.currentTimeMillis()
+    );
+
+    static File getDatabase(String name) throws IOException {
+        return copyResourceToFile("/databases/" + name);
+    }
+
+    static File getWordList(String name) throws IOException {
+        return copyResourceToFile("/wordlists/" + name);
+    }
+
+    static File getKeyFile() throws IOException {
+        return copyResourceToFile("/keyfiles/key.xml");
+    }
+
+    static File getInvalidKeyFile() throws IOException {
+        return copyResourceToFile("/keyfiles/invalid-key.xml");
+    }
+
+    private static File copyResourceToFile(String resourceName) throws IOException {
+        if (!resourceName.startsWith("/")) {
+            resourceName = "/" + resourceName;
+        }
+
+        InputStream input = Utils.class.getResource(resourceName).openStream();
+        byte[] bytes = new byte[input.available()];
+        input.read(bytes);
+        String fileName = resourceName.substring(resourceName.lastIndexOf('/'));
+        File outputFile = new File(getTmpDir(), fileName);
+        OutputStream output = new FileOutputStream(outputFile);
+        output.write(bytes);
+        return outputFile;
+    }
+
+    static File getTmpDir() {
+        if (!tmpDir.exists()) {
+            tmpDir.mkdirs();
+        }
+
+        return tmpDir;
+    }
+
+    static void rmdir(File dir) {
+        if (dir == null) return;
+
+        if (dir.isDirectory()) {
+            //noinspection ConstantConditions
+            for (File file : dir.listFiles()) {
+                rmdir(file);
+            }
+        }
+
+        dir.delete();
+    }
+}

keecrack-lib/src/test/resources/keyfiles/invalid-key.xml

@@ -0,0 +1,9 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<KeyFile>
+    <Meta>
+        <Version>1.00</Version>
+    </Meta>
+    <Key>
+        <Data>e5xTFwUylSKzNl90CODEfMwBjsbS2xvrzOWgX3dT+U8=</Data>
+    </Key>
+</KeyFile>

keecrack-lib/src/test/resources/keyfiles/key.xml

@@ -0,0 +1,9 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<KeyFile>
+    <Meta>
+        <Version>1.00</Version>
+    </Meta>
+    <Key>
+        <Data>e4xTFwUylSKzNl90CODEfMwBjsbS2xvrzOWgX3dT+U8=</Data>
+    </Key>
+</KeyFile>

keecrack-lib/src/test/resources/wordlists/invalid-words.txt

@@ -0,0 +1,3 @@
+654321
+wingsred
+wordpass

keecrack-lib/src/test/resources/wordlists/valid-words.txt

@@ -0,0 +1,3 @@
+123456
+redwings
+password

settings.gradle

@@ -0,0 +1 @@
+include 'keecrack-lib', 'keecrack-gui', 'keecrack-cli'