/**
* sha3sum – SHA-3 (Keccak) checksum calculator
*
* Copyright © 2013, 2014 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(" -x");
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 checksum at all iterations.");
System.out.println("");
System.out.println("");
System.out.println("COPYRIGHT:");
System.out.println("");
System.out.println("Copyright © 2013, 2014 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) == 'x')
{
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 k = 0; k < n; k++)
{ byte a = chunk[k << 1], b = chunk[(k << 1) | 1];
chunk[k] = (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 k = 1; k < i; k++)
{
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 k = 1; k < i; k++)
{
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 k = 0; k < i; k++)
{
if (k > 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);
}
}
