diff --git a/.travis.yml b/.travis.yml index aac6a7a..2bfdf7a 100644 --- a/.travis.yml +++ b/.travis.yml @@ -3,4 +3,8 @@ language: java sudo: false notifications: - email: false \ No newline at end of file + email: false + +env: + - TEST_DIR=MD5 +script: cd $TEST_DIR && ./gradlew clean check diff --git a/MD5/.idea/misc.xml b/MD5/.idea/misc.xml new file mode 100644 index 0000000..bc8d0a3 --- /dev/null +++ b/MD5/.idea/misc.xml @@ -0,0 +1,7 @@ + + + + + + + \ No newline at end of file diff --git a/MD5/build.gradle b/MD5/build.gradle new file mode 100644 index 0000000..fddd200 --- /dev/null +++ b/MD5/build.gradle @@ -0,0 +1,16 @@ +group 'ru.spbau.dkaznacheev' +version '1.0-SNAPSHOT' + +apply plugin: 'java' + +sourceCompatibility = 1.8 + +repositories { + mavenCentral() +} + +dependencies { + testCompile group: 'junit', name: 'junit', version: '4.12' + compile group: 'org.jetbrains', name: 'annotations', version: '15.0' + +} diff --git a/MD5/gradle/wrapper/gradle-wrapper.jar b/MD5/gradle/wrapper/gradle-wrapper.jar new file mode 100644 index 0000000..01b8bf6 Binary files /dev/null and b/MD5/gradle/wrapper/gradle-wrapper.jar differ diff --git a/MD5/gradle/wrapper/gradle-wrapper.properties b/MD5/gradle/wrapper/gradle-wrapper.properties new file mode 100644 index 0000000..933b647 --- /dev/null +++ b/MD5/gradle/wrapper/gradle-wrapper.properties @@ -0,0 +1,5 @@ +distributionBase=GRADLE_USER_HOME +distributionPath=wrapper/dists +zipStoreBase=GRADLE_USER_HOME +zipStorePath=wrapper/dists +distributionUrl=https\://services.gradle.org/distributions/gradle-4.4-bin.zip diff --git a/MD5/gradlew b/MD5/gradlew new file mode 100755 index 0000000..cccdd3d --- /dev/null +++ b/MD5/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" "$@" diff --git a/MD5/gradlew.bat b/MD5/gradlew.bat new file mode 100644 index 0000000..e95643d --- /dev/null +++ b/MD5/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 diff --git a/MD5/settings.gradle b/MD5/settings.gradle new file mode 100644 index 0000000..9e7a94b --- /dev/null +++ b/MD5/settings.gradle @@ -0,0 +1,2 @@ +rootProject.name = 'MD5' + diff --git a/MD5/src/main/java/ForkJoinError.java b/MD5/src/main/java/ForkJoinError.java new file mode 100644 index 0000000..672f4f1 --- /dev/null +++ b/MD5/src/main/java/ForkJoinError.java @@ -0,0 +1,5 @@ +/** + * An exception that occurs during ForkJoinPool execution. + */ +public class ForkJoinError extends Error { +} diff --git a/MD5/src/main/java/MD5.java b/MD5/src/main/java/MD5.java new file mode 100644 index 0000000..c44576a --- /dev/null +++ b/MD5/src/main/java/MD5.java @@ -0,0 +1,143 @@ +import org.jetbrains.annotations.NotNull; + +import java.io.File; +import java.io.FileInputStream; +import java.io.IOException; +import java.nio.file.NotDirectoryException; +import java.security.DigestInputStream; +import java.security.MessageDigest; +import java.security.NoSuchAlgorithmException; +import java.util.LinkedList; +import java.util.List; +import java.util.concurrent.ForkJoinPool; +import java.util.concurrent.RecursiveTask; + + +/** + * Class containing MD5 hashing methods, single or multithreaded. + * Hashing rules: + * hash(dir) = MD5(name + hash(file1) + ...) + * hash(file) = MD5(file) + */ +public class MD5 { + + /** + * Hashes a single non-directory file. + * @param file file to hash + * @return hash of a file + */ + private static @NotNull byte[] fileHash(@NotNull File file) throws IOException, NoSuchAlgorithmException { + MessageDigest digest = MessageDigest.getInstance("MD5"); + byte[] buf = new byte[2048]; + try (FileInputStream inputStream = new FileInputStream(file); + DigestInputStream digestInputStream = new DigestInputStream(inputStream, digest) + ) { + while (digestInputStream.read(buf) > -1); + return digestInputStream.getMessageDigest().digest(); + } + + } + + /** + * Hashes a directory or file, runs on a single thread. + * @param root directory or file to hash + * @return hash of directory or file. + */ + public static @NotNull byte[] hashSingleThread(@NotNull File root) throws IOException, NoSuchAlgorithmException { + if (root.isDirectory()) { + MessageDigest digest = MessageDigest.getInstance("MD5"); + digest.update(root.getName().getBytes()); + File[] contents = root.listFiles(); + if (contents == null) { + throw new NotDirectoryException(root.getName()); + } + for (File file : contents) { + digest.update(hashSingleThread(file)); + } + return digest.digest(); + } else { + return fileHash(root); + } + } + + /** + * Hashes a directory or file, runs on multiple threads. + * @param root directory or file to hash + * @return hash of directory or file. + */ + public static @NotNull byte[] hashMultiThread(@NotNull File root) { + ForkJoinPool pool = new ForkJoinPool(); + HashRecursiveTask task = new HashRecursiveTask(root); + return pool.invoke(task); + } + + /** + * Recursive task for ForkJoin hashing. + */ + private static class HashRecursiveTask extends RecursiveTask { + + /** + * File to hash. + */ + private File file; + + public HashRecursiveTask(@NotNull File file) { + this.file = file; + } + + /** + * Returns hash of task's directory or file. + * @return hash of directory or file. + */ + @Override + protected @NotNull byte[] compute() { + try { + MessageDigest digest = MessageDigest.getInstance("MD5"); + if (file.isDirectory()) { + digest.update(file.getName().getBytes()); + File[] contents = file.listFiles(); + if (contents == null) { + throw new ForkJoinError(); + } + List tasks = new LinkedList<>(); + for (File content : contents) { + HashRecursiveTask task = new HashRecursiveTask(content); + task.fork(); + tasks.add(task); + } + for (HashRecursiveTask task : tasks) { + digest.update(task.join()); + } + return digest.digest(); + } else { + return fileHash(file); + } + } catch (NoSuchAlgorithmException | IOException e) { + throw new ForkJoinError(); + } + } + } + + public static void main(String[] args) { + File root = new File("./"); //note that it runs in project directory by default + long startSingleTime = System.nanoTime(); + try { + hashSingleThread(root); + } catch (IOException | NoSuchAlgorithmException e) { + e.printStackTrace(); + System.err.println("Error running single thread hashing!"); + } + long endSingleTime = System.nanoTime(); + long startMultiTime = System.nanoTime(); + try { + hashSingleThread(root); + } catch (IOException | NoSuchAlgorithmException e) { + System.err.println("Error running multithread hashing!"); + } + long endMultiTime = System.nanoTime(); + + System.out.println("Single Thread Hashing: " + (endSingleTime - startSingleTime) / 1000 + " ms"); + System.out.println("MultiThread Hashing: " + (endMultiTime - startMultiTime) / 1000 + " ms"); + } + +} diff --git a/MD5/src/test/java/MD5Test.java b/MD5/src/test/java/MD5Test.java new file mode 100644 index 0000000..29d0b07 --- /dev/null +++ b/MD5/src/test/java/MD5Test.java @@ -0,0 +1,91 @@ +import org.junit.Test; + +import java.io.*; +import java.util.Random; + +import static org.junit.Assert.*; + +public class MD5Test { + @Test + public void singleMD5equalsMultiMD5test() throws Exception { + File root = new File("./"); + byte[] hash1 = MD5.hashSingleThread(root); + byte[] hash2 = MD5.hashMultiThread(root); + root.delete(); + assertArrayEquals(hash1, hash2); + } + + @Test + public void singleMD5DoesNotChange() throws Exception { + File root = new File("./"); + byte[] hash1 = MD5.hashSingleThread(root); + byte[] hash2 = MD5.hashSingleThread(root); + root.delete(); + assertArrayEquals(hash1, hash2); + } + + @Test + public void multiMD5DoesNotChange() throws Exception { + File root = new File("./"); + byte[] hash1 = MD5.hashMultiThread(root); + byte[] hash2 = MD5.hashMultiThread(root); + root.delete(); + assertArrayEquals(hash1, hash2); + } + + @Test + public void singleFileMD5hash() throws Exception { + createFile("test.tmp", 20); + File root = new File("test.tmp"); + byte[] hash1 = MD5.hashSingleThread(root); + byte[] hash2 = MD5.hashMultiThread(root); + root.delete(); + assertArrayEquals(hash1, hash2); + } + + @Test + public void emptyFileMD5hash() throws Exception { + createFile("empty.tmp", 0); + File root = new File("empty.tmp"); + byte[] hash1 = MD5.hashSingleThread(root); + byte[] hash2 = MD5.hashMultiThread(root); + root.delete(); + assertArrayEquals(hash1, hash2); + } + + @Test + public void emptyDirectoryMD5hash() throws Exception { + File root = new File("emptydir"); + root.mkdirs(); + byte[] hash1 = MD5.hashSingleThread(root); + byte[] hash2 = MD5.hashMultiThread(root); + root.delete(); + assertArrayEquals(hash1, hash2); + } + + @Test (expected = IOException.class) + public void noSuchFileSingleThrows() throws Exception { + File root = new File("nosuchdir"); + MD5.hashSingleThread(root); + } + + @Test (expected = IOException.class) + public void noSuchFileMultiThrows() throws Exception { + File root = new File("nosuchdir"); + MD5.hashSingleThread(root); + } + + private void createFile(String name, int length) throws IOException { + File file = new File(name); + file.createNewFile(); + Random random = new Random(); + try ( + OutputStream outputStream = new FileOutputStream(file, false); + DataOutputStream dataOutputStream = new DataOutputStream(outputStream)) { + for (int i = 0; i < length; i++) { + dataOutputStream.writeByte((byte) random.nextInt()); + } + } + } + +} \ No newline at end of file