tools/testing/java/com/google/crypto/tink/testing/DeterministicAeadCli.java

// Copyright 2017 Google Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
////////////////////////////////////////////////////////////////////////////////

package com.google.crypto.tink.testing;

import com.google.crypto.tink.DeterministicAead;
import com.google.crypto.tink.KeysetHandle;

/**
 * A command-line utility for testing DeterministicAead-primitives.
 * It requires 5 arguments:
 *   keyset-file:  name of the file with the keyset to be used for encryption
 *   operation: the actual DeterministicAead-operation, i.e.
 *   "encryptdeterministically" or "decryptdeterministically"
 *   input-file:  name of the file with input (plaintext for encryption, or
 *                or ciphertext for decryption)
 *   associated-data-file:  name of the file containing associated data
 *   output-file:  name of the file for the resulting output
 */
public class DeterministicAeadCli {
  public static void main(String[] args) throws Exception {
    if (args.length != 5) {
      System.out.println("Usage: DeterministicAeadCli keyset-file operation "
          + "input-file associated-data-file output-file");
      System.exit(1);
    }
    String keysetFilename = args[0];
    String operation = args[1];
    String inputFilename = args[2];
    String associatedDataFile = args[3];
    String outputFilename = args[4];
    if (!(operation.equals("encryptdeterministically")
        || operation.equals("decryptdeterministically"))) {
      System.out.println("Unknown operation '" + operation + "'.\nExpected "
          + "'encryptdeterministically' or 'decryptdeterministically'.");
      System.exit(1);
    }
    System.out.println("Using keyset from file " + keysetFilename + " to " + operation
        + " file " + inputFilename + " with associated data from file " + associatedDataFile + ".");
    System.out.println("The resulting output will be written to file " + outputFilename);

    // Init Tink.
    CliUtil.initTink();

    // Read the keyset.
    System.out.println("Reading the keyset...");
    KeysetHandle keysetHandle = CliUtil.readKeyset(keysetFilename);

    // Get the primitive.
    System.out.println("Getting the primitive...");
    DeterministicAead daead = keysetHandle.getPrimitive(DeterministicAead.class);

    // Read the input.
    byte[] input = CliUtil.read(inputFilename);
    byte[] aad = CliUtil.read(associatedDataFile);

    // Compute the output.
    System.out.println("performing operation " + operation + "...");
    byte[] output;
    if (operation.equals("encryptdeterministically")) {
      output = daead.encryptDeterministically(input, aad);
    } else { // operation.equals("decryptdeterministically")
      output = daead.decryptDeterministically(input, aad);
    }

    // Write the output to the output file.
    CliUtil.write(output, outputFilename);

    System.out.println("All done.");
  }
}