1 files changed, 241 insertions, 0 deletions

diff --git a/src/cz/crcs/ectester/reader/Command.java b/src/cz/crcs/ectester/reader/Command.java
new file mode 100644
index 0000000..c5fd465
--- /dev/null
+++ b/src/cz/crcs/ectester/reader/Command.java
@@ -0,0 +1,241 @@
+package cz.crcs.ectester.reader;
+
+import cz.crcs.ectester.applet.ECTesterApplet;
+
+import javax.smartcardio.CardException;
+import javax.smartcardio.CommandAPDU;
+import javax.smartcardio.ResponseAPDU;
+import java.util.ArrayList;
+import java.util.List;
+
+/**
+ * @author Jan Jancar johny@neuromancer.sk
+ */
+public abstract class Command {
+    protected CommandAPDU cmd;
+    protected CardMngr cardManager;
+
+    protected Command(CardMngr cardManager) {
+        this.cardManager = cardManager;
+    }
+
+    public CommandAPDU getAPDU() {
+        return cmd;
+    }
+
+    /**
+     *
+     * @return
+     * @throws CardException
+     */
+    public abstract Response send() throws CardException;
+
+    public static List<Response> sendAll(List<Command> commands) throws CardException {
+        List<Response> result = new ArrayList<>();
+        for (Command cmd : commands) {
+            result.add(cmd.send());
+        }
+        return result;
+    }
+
+    /**
+     *
+     */
+    public static class Allocate extends Command {
+        private byte keyPair;
+        private short keyLength;
+        private byte keyClass;
+
+        /**
+         * Creates the INS_ALLOCATE instruction.
+         *
+         * @param cardManager
+         * @param keyPair     which keyPair to use, local/remote (KEYPAIR_* | ...)
+         * @param keyLength   key length to set
+         * @param keyClass    key class to allocate
+         */
+        public Allocate(CardMngr cardManager, byte keyPair, short keyLength, byte keyClass) {
+            super(cardManager);
+            this.keyPair = keyPair;
+            this.keyLength = keyLength;
+            this.keyClass = keyClass;
+
+            byte[] data = new byte[]{0, 0, keyClass};
+            Util.setShort(data, 0, keyLength);
+            this.cmd = new CommandAPDU(ECTesterApplet.CLA_ECTESTERAPPLET, ECTesterApplet.INS_ALLOCATE, keyPair, 0x00, data);
+        }
+
+        @Override
+        public Response.Allocate send() throws CardException {
+            long elapsed = -System.nanoTime();
+            ResponseAPDU response = cardManager.send(cmd);
+            elapsed += System.nanoTime();
+            return new Response.Allocate(response, elapsed, keyPair, keyLength, keyClass);
+        }
+    }
+
+    /**
+     *
+     */
+    public static class Set extends Command {
+        private byte keyPair;
+        private byte export;
+        private byte curve;
+        private short params;
+        private short corrupted;
+        private byte corruption;
+        private byte[] external;
+
+        /**
+         * Creates the INS_SET instruction.
+         *
+         * @param cardManager
+         * @param keyPair     which keyPair to set params on, local/remote (KEYPAIR_* || ...)
+         * @param export      whether to export set params from keyPair
+         * @param curve       curve to set (EC_Consts.CURVE_*)
+         * @param params      parameters to set (EC_Consts.PARAMETER_* | ...)
+         * @param corrupted   parameters to corrupt (EC_Consts.PARAMETER_* | ...)
+         * @param corruption  corruption type (EC_Consts.CORRUPTION_*)
+         * @param external    external curve data, can be null
+         */
+        public Set(CardMngr cardManager, byte keyPair, byte export, byte curve, short params, short corrupted, byte corruption, byte[] external) {
+            super(cardManager);
+            this.keyPair = keyPair;
+            this.export = export;
+            this.curve = curve;
+            this.params = params;
+            this.corrupted = corrupted;
+            this.corruption = corruption;
+            this.external = external;
+
+            int len = external != null ? 6 + 2 + external.length : 6;
+            byte[] data = new byte[len];
+            data[0] = curve;
+            Util.setShort(data, 1, params);
+            Util.setShort(data, 3, corrupted);
+            data[5] = corruption;
+            if (external != null) {
+                System.arraycopy(external, 0, data, 6, external.length);
+            }
+
+            this.cmd = new CommandAPDU(ECTesterApplet.CLA_ECTESTERAPPLET, ECTesterApplet.INS_SET, keyPair, export, data);
+        }
+
+        @Override
+        public Response.Set send() throws CardException {
+            long elapsed = -System.nanoTime();
+            ResponseAPDU response = cardManager.send(cmd);
+            elapsed += System.nanoTime();
+            return new Response.Set(response, elapsed, keyPair, export, curve, params, corrupted);
+        }
+    }
+
+    /**
+     *
+     */
+    public static class Generate extends Command {
+        private byte keyPair;
+        private byte export;
+
+        /**
+         * Creates the INS_GENERATE instruction.
+         *
+         * @param cardManager
+         * @param keyPair     which keyPair to generate, local/remote (KEYPAIR_* || ...)
+         * @param export      whether to export generated keys from keyPair
+         */
+        public Generate(CardMngr cardManager, byte keyPair, byte export) {
+            super(cardManager);
+            this.keyPair = keyPair;
+            this.export = export;
+
+            this.cmd = new CommandAPDU(ECTesterApplet.CLA_ECTESTERAPPLET, ECTesterApplet.INS_GENERATE, keyPair, export);
+        }
+
+        @Override
+        public Response.Generate send() throws CardException {
+            long elapsed = -System.nanoTime();
+            ResponseAPDU response = cardManager.send(cmd);
+            elapsed += System.nanoTime();
+            return new Response.Generate(response, elapsed, keyPair, export);
+        }
+    }
+
+    /**
+     *
+     */
+    public static class ECDH extends Command {
+        private byte pubkey;
+        private byte privkey;
+        private byte export;
+        private byte invalid;
+
+        /**
+         * Creates the INS_ECDH instruction.
+         *
+         * @param cardManager
+         * @param pubkey      keyPair to use for public key, (KEYPAIR_LOCAL || KEYPAIR_REMOTE)
+         * @param privkey     keyPair to use for private key, (KEYPAIR_LOCAL || KEYPAIR_REMOTE)
+         * @param export      whether to export ECDH secret
+         * @param invalid     whether to invalidate the pubkey before ECDH
+         */
+        public ECDH(CardMngr cardManager, byte pubkey, byte privkey, byte export, byte invalid) {
+            super(cardManager);
+            this.pubkey = pubkey;
+            this.privkey = privkey;
+            this.export = export;
+            this.invalid = invalid;
+
+            byte[] data = new byte[]{export, invalid};
+
+            this.cmd = new CommandAPDU(ECTesterApplet.CLA_ECTESTERAPPLET, ECTesterApplet.INS_ECDH, pubkey, privkey, data);
+        }
+
+        @Override
+        public Response.ECDH send() throws CardException {
+            long elapsed = -System.nanoTime();
+            ResponseAPDU response = cardManager.send(cmd);
+            elapsed += System.nanoTime();
+            return new Response.ECDH(response, elapsed, pubkey, privkey, export, invalid);
+        }
+    }
+
+    public static class ECDSA extends Command {
+        private byte keyPair;
+        private byte export;
+        private byte[] raw;
+
+        /**
+         * Creates the INS_ECDSA instruction.
+         *
+         * @param cardManager
+         * @param keyPair     keyPair to use for signing and verification (KEYPAIR_LOCAL || KEYPAIR_REMOTE)
+         * @param export      whether to export ECDSA signature
+         * @param raw         data to sign, can be null, in which case random data is signed.
+         */
+        public ECDSA(CardMngr cardManager, byte keyPair, byte export, byte[] raw) {
+            super(cardManager);
+            this.keyPair = keyPair;
+            this.export = export;
+            this.raw = raw;
+
+            int len = raw != null ? raw.length : 0;
+            byte[] data = new byte[2 + len];
+            Util.setShort(data, 0, (short) len);
+            if (raw != null) {
+                System.arraycopy(raw, 0, data, 2, len);
+            }
+
+            this.cmd = new CommandAPDU(ECTesterApplet.CLA_ECTESTERAPPLET, ECTesterApplet.INS_ECDSA, keyPair, export, data);
+        }
+
+        @Override
+        public Response.ECDSA send() throws CardException {
+            long elapsed = -System.nanoTime();
+            ResponseAPDU response = cardManager.send(cmd);
+            elapsed += System.nanoTime();
+            return new Response.ECDSA(response, elapsed, keyPair, export, raw);
+        }
+    }
+}
+