diff options
Diffstat (limited to '')
| -rw-r--r--[l---------] | java-c-jni/sha3sum.java | 486 |
1 files changed, 485 insertions, 1 deletions
diff --git a/java-c-jni/sha3sum.java b/java-c-jni/sha3sum.java index 27a2d3e..078dd61 120000..100644 --- a/java-c-jni/sha3sum.java +++ b/java-c-jni/sha3sum.java @@ -1 +1,485 @@ -../pure-java/sha3sum.java
\ No newline at end of file +/** + * sha3sum – SHA-3 (Keccak) checksum calculator + * + * Copyright © 2013 Mattias Andrée (maandree@member.fsf.org) + * + * This program is free software: you can redistribute it and/or modify + * it under the terms of the GNU Affero General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Affero General Public License for more details. + * + * You should have received a copy of the GNU Affero General Public License + * along with this program. If not, see <http://www.gnu.org/licenses/>. + */ + +import java.io.*; +import java.util.*; + + +/** + * SHA-3/Keccak checksum calculator + * + * @author Mattias Andrée <a href="mailto:maandree@member.fsf.org">maandree@member.fsf.org</a> + */ +public class sha3sum +{ + /** + * This is the main entry point of the program + * + * @param args Command line arguments + * @throws IOException On I/O error (such as broken pipes) + */ + public static void main(String[] args) throws IOException + { + run("sha3sum", args); + } + + + /** + * Run the program + * + * @param cmd The command + * @param argv Command line arguments + * @throws IOException On I/O error (such as broken pipes) + */ + public static void run(String cmd, String[] argv) throws IOException + { + if (cmd.indexOf('/') >= 0) + cmd = cmd.substring(cmd.lastIndexOf('/') + 1); + if (cmd.endsWith(".jar")) + cmd = cmd.substring(0, cmd.length() - 4); + cmd = cmd.intern(); + + Integer O = null; int _o = 512; /* --outputsize */ + if (cmd == "sha3-224sum") _o = 224; + else if (cmd == "sha3-256sum") _o = 256; + else if (cmd == "sha3-384sum") _o = 384; + else if (cmd == "sha3-512sum") _o = 512; + Integer S = null; int _s = 1600; /* --statesize */ + Integer C = null; int _c = _s - (_o << 1); /* --capacity */ + Integer R = null; int _r = _s - _c; /* --bitrate */ + Integer W = null; int _w = _s / 25; /* --wordsize */ + Integer I = null; int _i = 1; /* --iterations */ + Integer J = null; int _j = 1; /* --squeezes */ + int o = 0, s = 0, r = 0, c = 0, w = 0, i = 0, j = 0; + + boolean binary = false, hex = false; + int multi = 0; + + String[] files = new String[argv.length + 1]; + int fptr = 0; + boolean dashed = false; + String[] linger = null; + + String[] args = new String[argv.length + 1]; + System.arraycopy(argv, 0, args, 0, argv.length); + for (int a = 0, an = args.length; a < an; a++) + { String arg = args[a]; + arg = arg == null ? null : arg.intern(); + if (linger != null) + { + linger[0] = linger[0].intern(); + if ((linger[0] == "-h") || (linger[0] == "--help")) + { + System.out.println(""); + System.out.println("SHA-3/Keccak checksum calculator"); + System.out.println(""); + System.out.println("USAGE: sha3sum [option...] < file"); + System.out.println(" sha3sum [option...] file..."); + System.out.println(""); + System.out.println(""); + System.out.println("OPTIONS:"); + System.out.println(" -r BITRATE"); + System.out.println(" --bitrate The bitrate to use for SHA-3. (default: " + _r + ")"); + System.out.println(" "); + System.out.println(" -c CAPACITY"); + System.out.println(" --capacity The capacity to use for SHA-3. (default: " + _c + ")"); + System.out.println(" "); + System.out.println(" -w WORDSIZE"); + System.out.println(" --wordsize The word size to use for SHA-3. (default: " + _w + ")"); + System.out.println(" "); + System.out.println(" -o OUTPUTSIZE"); + System.out.println(" --outputsize The output size to use for SHA-3. (default: " + _o + ")"); + System.out.println(" "); + System.out.println(" -s STATESIZE"); + System.out.println(" --statesize The state size to use for SHA-3. (default: " + _s + ")"); + System.out.println(" "); + System.out.println(" -i ITERATIONS"); + System.out.println(" --iterations The number of hash iterations to run. (default: " + _i + ")"); + System.out.println(" "); + System.out.println(" -j SQUEEZES"); + System.out.println(" --squeezes The number of hash squeezes to run. (default: " + _j + ")"); + System.out.println(" "); + System.out.println(" -h"); + System.out.println(" --hex Read the input in hexadecimal, rather than binary."); + System.out.println(" "); + System.out.println(" -b"); + System.out.println(" --binary Print the checksum in binary, rather than hexadecimal."); + System.out.println(" "); + System.out.println(" -m"); + System.out.println(" --multi Print the chechsum at all iterations."); + System.out.println(""); + System.out.println(""); + System.out.println("COPYRIGHT:"); + System.out.println(""); + System.out.println("Copyright © 2013 Mattias Andrée (maandree@member.fsf.org)"); + System.out.println(""); + System.out.println("This program is free software: you can redistribute it and/or modify"); + System.out.println("it under the terms of the GNU Affero General Public License as published by"); + System.out.println("the Free Software Foundation, either version 3 of the License, or"); + System.out.println("(at your option) any later version."); + System.out.println(""); + System.out.println("This program is distributed in the hope that it will be useful,"); + System.out.println("but WITHOUT ANY WARRANTY; without even the implied warranty of"); + System.out.println("MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the"); + System.out.println("GNU Affero General Public License for more details."); + System.out.println(""); + System.out.println("You should have received a copy of the GNU Affero General Public License"); + System.out.println("along with this program. If not, see <http://www.gnu.org/licenses/>."); + System.out.println(""); + System.exit(0); + } + else + { + if (linger[1] == null) + { + linger[1] = arg; + arg = null; + } + if ((linger[0] == "-r") || (linger[0] == "--bitrate")) + R = Integer.valueOf(linger[1]); + else if ((linger[0] == "-c") || (linger[0] == "--capacity")) + C = Integer.valueOf(linger[1]); + else if ((linger[0] == "-w") || (linger[0] == "--wordsize")) + W = Integer.valueOf(linger[1]); + else if ((linger[0] == "-o") || (linger[0] == "--outputsize")) + O = Integer.valueOf(linger[1]); + else if ((linger[0] == "-s") || (linger[0] == "--statesize")) + S = Integer.valueOf(linger[1]); + else if ((linger[0] == "-i") || (linger[0] == "--iterations")) + I = Integer.valueOf(linger[1]); + else if ((linger[0] == "-j") || (linger[0] == "--squeezes")) + J = Integer.valueOf(linger[1]); + else + { + System.err.println(cmd + ": unrecognised option: " + linger[0]); + System.exit(1); + } + } + linger = null; + if (arg == null) + continue; + } + if (arg == null) + continue; + if (dashed) + files[fptr++] = arg == "-" ? null : arg; + else if (arg == "--") + dashed = true; + else if (arg == "-") + files[fptr++] = null; + else if (arg.startsWith("--")) + if (arg.indexOf('=') >= 0) + linger = new String[] { arg.substring(0, arg.indexOf('=')), arg.substring(arg.indexOf('=') + 1) }; + else + if (arg == "--binary") + binary = true; + else if (arg == "--multi") + multi++; + else if (arg == "--hex") + hex = true; + else + linger = new String[] { arg, null }; + else if (arg.startsWith("-")) + { + arg = arg.substring(1); + if (arg.charAt(0) == 'b') + { + binary = true; + arg = arg.substring(1); + } + else if (arg.charAt(0) == 'm') + { + multi++; + arg = arg.substring(1); + } + else if (arg.charAt(0) == 'h') + { + hex = true; + arg = arg.substring(1); + } + else if (arg.length() == 1) + linger = new String[] { "-" + arg, null }; + else + linger = new String[] { "-" + arg.charAt(0), arg.substring(1) }; + } + else + files[fptr++] = arg; + } + + + i = I == null ? _i : I.intValue(); + j = J == null ? _j : J.intValue(); + + if (S != null) + { s = S.intValue(); + if ((s <= 0) || (s > 1600) || (s % 25 != 0)) + { System.err.println(cmd + ": the state size must be a positive multiple of 25 and is limited to 1600."); + System.exit(6); + } } + + if (W != null) + { w = W.intValue(); + if ((w <= 0) || (w > 64)) + { System.err.println(cmd + ": the word size must be positive and is limited to 64."); + System.exit(6); + } + if ((S != null) && (s != w * 25)) + { System.err.println(cmd + ": the state size must be 25 times of the word size."); + System.exit(6); + } + else if (S == null) + S = new Integer(w * 25); + } + + if (C != null) + { c = C.intValue(); + if ((c <= 0) || ((c & 7) != 0)) + { System.err.println(cmd + ": the capacity must be a positive multiple of 8."); + System.exit(6); + } } + + if (R != null) + { r = R.intValue(); + if ((r <= 0) || ((r & 7) != 0)) + { System.err.println(cmd + ": the bitrate must be a positive multiple of 8."); + System.exit(6); + } } + + if (O != null) + { o = O.intValue(); + if (o <= 0) + { System.err.println(cmd + ": the output size must be positive."); + System.exit(6); + } } + + + if ((R == null) && (C == null) && (O == null)) // s? + { c = -((r = (o = ((((s = S == null ? _s : s) << 5) / 100 + 7) >> 3) << 3) << 1) - s); + o = o < 8 ? 8 : o; + } + else if ((R == null) && (C == null)) // !o s? + { r = _r; + c = _c; + s = S == null ? (r + c) : s; + } + else if (R == null) // !c o? s? + { r = (s = S == null ? _s : s) - c; + o = O == null ? (c == 8 ? 8 : (c << 1)) : o; + } + else if (C == null) // !r o? s? + { c = (s = S == null ? _s : s) - r; + o = O == null ? (c == 8 ? 8 : (c << 1)) : o; + } + else // !r !c o? s? + { s = S == null ? (r + c) : s; + o = O == null ? (c == 8 ? 8 : (c << 1)) : o; + } + + + System.err.println("Bitrate: " + r); + System.err.println("Capacity: " + c); + System.err.println("Word size: " + w); + System.err.println("State size: " + s); + System.err.println("Output size: " + o); + System.err.println("Iterations: " + i); + System.err.println("Squeezes: " + j); + + + if (r > s) + { System.err.println(cmd + ": the bitrate must not be higher than the state size."); + System.exit(6); + } + if (c > s) + { System.err.println(cmd + ": the capacity must not be higher than the state size."); + System.exit(6); + } + if (r + c != s) + { System.err.println(cmd + ": the sum of the bitrate and the capacity must equal the state size."); + System.exit(6); + } + + + if (fptr == 0) + files[fptr++] = null; + if (i < 1) + { + System.err.println(cmd + ": sorry, I will only do at least one hash iteration!"); + System.exit(3); + } + if (j < 1) + { + System.err.println(cmd + ": sorry, I will only do at least one squeeze iteration!"); + System.exit(3); + } + + byte[] stdin = null; + boolean fail = false; + String filename; + + for (int f = 0; f < fptr; f++) + { String rc = ""; + String fn = (filename = files[f]) == null ? "/dev/stdin" : filename; + InputStream file = null; + try + { + byte[] bs; + if ((filename != null) || (stdin == null)) + { + file = new FileInputStream(fn); + SHA3.initialise(r, c, o); + int blksize = 4096; /** XXX os.stat(os.path.realpath(fn)).st_size; **/ + byte[] chunk = new byte[blksize]; + for (;;) + { + int read = file.read(chunk, 0, blksize); + if (read <= 0) + break; + if (hex == false) + SHA3.update(chunk, read); + else + { + int n = read >> 1; + for (int _ = 0; _ < n; _++) + { byte a = chunk[_ << 1], b = chunk[(_ << 1) | 1]; + chunk[_] = (byte)((((a & 15) + (a <= '9' ? 0 : 9)) << 4) | ((b & 15) + (b <= '9' ? 0 : 9))); + } + SHA3.update(chunk, n); + } + } + bs = SHA3.digest(j == 1); + if (j > 2) + SHA3.fastSqueeze(j - 2); + if (j > 1) + bs = SHA3.squeeze(); + if (filename == null) + stdin = bs; + } + else + bs = stdin; + if (multi == 0) + { + for (int _ = 1; _ < i; _++) + { + SHA3.initialise(r, c, o); + bs = SHA3.digest(bs, j == 1); + if (j > 2) + SHA3.fastSqueeze(j - 2); + if (j > 1) + bs = SHA3.squeeze(); + } + if (binary) + System.out.write(bs); + else + { for (int b = 0, bn = bs.length; b < bn; b++) + { rc += "0123456789ABCDEF".charAt((bs[b] >> 4) & 15); + rc += "0123456789ABCDEF".charAt(bs[b] & 15); + } + rc += " " + (filename == null ? "-" : filename) + "\n"; + System.out.print(rc); + } + } + else if (multi == 1) + { + byte[] out = null; + if (binary) + System.out.write(bs); + else + { + out = new byte[(bs.length << 1) + 1]; + for (int b = 0, bn = bs.length; b < bn; b++) + { out[ b << 1 ] = (byte)("0123456789ABCDEF".charAt((bs[b] >> 4) & 15)); + out[(b << 1) | 1] = (byte)("0123456789ABCDEF".charAt(bs[b] & 15)); + } + out[out.length - 1] = '\n'; + System.out.write(out); + } + for (int _ = 1; _ < i; _++) + { + SHA3.initialise(r, c, o); + bs = SHA3.digest(bs, j == 1); + if (j > 2) + SHA3.fastSqueeze(j - 2); + if (j > 1) + bs = SHA3.squeeze(); + if (binary) + System.out.write(bs); + else + { + for (int b = 0, bn = bs.length; b < bn; b++) + { out[ b << 1 ] = (byte)("0123456789ABCDEF".charAt((bs[b] >> 4) & 15)); + out[(b << 1) | 1] = (byte)("0123456789ABCDEF".charAt(bs[b] & 15)); + } + System.out.write(out); + } + } + } + else + { + HashSet<String> got = new HashSet<String>(); + String loop = null; + byte[] out = new byte[(bs.length << 1)]; + for (int _ = 0; _ < i; _++) + { + if (_ > 0) + { SHA3.initialise(r, c, o); + bs = SHA3.digest(bs, j == 1); + if (j > 2) + SHA3.fastSqueeze(j - 2); + if (j > 1) + bs = SHA3.squeeze(); + } + for (int b = 0, bn = bs.length; b < bn; b++) + { out[ b << 1 ] = (byte)("0123456789ABCDEF".charAt((bs[b] >> 4) & 15)); + out[(b << 1) | 1] = (byte)("0123456789ABCDEF".charAt(bs[b] & 15)); + } + String now = new String(out, "UTF-8"); + if (loop == null) + if (got.contains(now)) + loop = now; + else + got.add(now); + if ((loop != null) && (loop.equals(now))) + now = "\033[31m" + now + "\033[00m"; + System.out.println(now); + } + if (loop != null) + System.err.println("\033[01;31mLoop found\033[00m"); + } + System.out.flush(); + } + catch (final IOException err) + { System.err.println(cmd + ": cannot read file: " + filename + ": " + err); + fail = true; + } + finally + { if (file != null) + try + { file.close(); + } + catch (final Throwable ignore) + { //ignore + } } } + + System.out.flush(); + if (fail) + System.exit(5); + } + +} + |