1 files changed, 104 insertions, 0 deletions
diff --git a/base/java-tools/src/com/netscape/cmstools/key/KeyGenerateCLI.java b/base/java-tools/src/com/netscape/cmstools/key/KeyGenerateCLI.java
new file mode 100644
index 000000000..5edf9489c
--- /dev/null
+++ b/base/java-tools/src/com/netscape/cmstools/key/KeyGenerateCLI.java
@@ -0,0 +1,104 @@
+package com.netscape.cmstools.key;
+
+import java.util.ArrayList;
+import java.util.Arrays;
+import java.util.List;
+
+import org.apache.commons.cli.CommandLine;
+import org.apache.commons.cli.Option;
+import org.apache.commons.cli.ParseException;
+
+import com.netscape.certsrv.key.KeyRequestResource;
+import com.netscape.certsrv.key.KeyRequestResponse;
+import com.netscape.cmstools.cli.CLI;
+import com.netscape.cmstools.cli.MainCLI;
+
+public class KeyGenerateCLI extends CLI {
+    public KeyCLI keyCLI;
+
+    public KeyGenerateCLI(KeyCLI keyCLI) {
+        super("generate", "Generate key", keyCLI);
+        this.keyCLI = keyCLI;
+    }
+
+    public void printHelp() {
+        formatter.printHelp(getFullName() + " <Client Key ID> [OPTIONS]", options);
+    }
+
+    public void execute(String[] args) {
+        Option option = new Option(null, "key-algorithm", true,
+                "Algorithm to be used to create a key.\n Supported types [AES,DES,DES3,RC2,RC4,DESede]");
+        option.setArgName("Key algorithm");
+        option.setRequired(true);
+        options.addOption(option);
+
+        option = new Option(null, "key-size", true,
+                "Size of the key to be generated. Required for all algorithms AES and RC2.\n"
+                        + "Valid size values:\n   AES - 128, 192. 256.\n   RC2 - >=8, <=128");
+        option.setArgName("Key size");
+        options.addOption(option);
+
+        option = new Option(null, "usages", true, "Comma seperated list of usages."
+                + "\n Usage1,Usage2,Usage3.. .\n "
+                + "Valid usages: [wrap, unwrap, sign, verify, encrypt, decrypt].");
+        option.setArgName("Usages");
+        options.addOption(option);
+
+        CommandLine cmd = null;
+        try {
+            cmd = parser.parse(options, args);
+
+        } catch (ParseException e) {
+            System.err.println("Error: " + e.getMessage());
+            printHelp();
+            System.exit(1);
+        }
+
+        String[] cmdArgs = cmd.getArgs();
+        if (cmdArgs.length < 1) {
+            System.err.println("Error: Missing the Client Key Id");
+            printHelp();
+            System.exit(1);
+        }
+
+        String clientKeyId = cmdArgs[0];
+        String keyAlgorithm = cmd.getOptionValue("key-algorithm");
+        String keySize = cmd.getOptionValue("key-size");
+
+        if (keySize == null) {
+            switch (keyAlgorithm) {
+            case KeyRequestResource.DES3_ALGORITHM:
+            case KeyRequestResource.DESEDE_ALGORITHM:
+                keySize = "168";
+                break;
+            case KeyRequestResource.DES_ALGORITHM:
+                keySize = "56";
+                break;
+            case KeyRequestResource.RC4_ALGORITHM:
+                keySize = "0";
+                break;
+            case KeyRequestResource.AES_ALGORITHM:
+            case KeyRequestResource.RC2_ALGORITHM:
+                System.err.println("Error: Key size must be specified for the algorithm used.");
+                printHelp();
+                System.exit(1);
+            default:
+                System.err.println("Error: Algorithm not supported.");
+                printHelp();
+                System.exit(1);
+            }
+        }
+        List<String> usagesList = null;
+        if (cmd.getOptionValue("usages") != null) {
+            String[] usages = cmd.getOptionValue("usages").split(",");
+            usagesList = new ArrayList<String>(Arrays.asList(usages));
+        }
+
+        KeyRequestResponse response = keyCLI.keyClient.generateSymmetricKey(clientKeyId, keyAlgorithm,
+                Integer.parseInt(keySize),
+                usagesList, null);
+
+        MainCLI.printMessage("Key generation request info");
+        KeyCLI.printKeyRequestInfo(response.getRequestInfo());
+    }
+}