jdk/src/share/classes/com/sun/crypto/provider/CipherTextStealing.java
changeset 2 90ce3da70b43
child 5506 202f599c92aa
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/jdk/src/share/classes/com/sun/crypto/provider/CipherTextStealing.java	Sat Dec 01 00:00:00 2007 +0000
@@ -0,0 +1,215 @@
+/*
+ * Copyright 2004-2007 Sun Microsystems, Inc.  All Rights Reserved.
+ * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
+ *
+ * This code is free software; you can redistribute it and/or modify it
+ * under the terms of the GNU General Public License version 2 only, as
+ * published by the Free Software Foundation.  Sun designates this
+ * particular file as subject to the "Classpath" exception as provided
+ * by Sun in the LICENSE file that accompanied this code.
+ *
+ * This code 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 General Public License
+ * version 2 for more details (a copy is included in the LICENSE file that
+ * accompanied this code).
+ *
+ * You should have received a copy of the GNU General Public License version
+ * 2 along with this work; if not, write to the Free Software Foundation,
+ * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
+ *
+ * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
+ * CA 95054 USA or visit www.sun.com if you need additional information or
+ * have any questions.
+ */
+
+package com.sun.crypto.provider;
+
+import javax.crypto.IllegalBlockSizeException;
+import javax.crypto.ShortBufferException;
+
+/**
+ * This class represents ciphers in cipher text stealing (CTS) mode.
+ * <br>CTS provides a way to allow block ciphers to operate on partial
+ * blocks without padding, and all bits of the message go through
+ * the encryption algorithm, rather than simply being XOR'd.
+ * <br>More details can be found in RFC 2040 section 8 "Description
+ * of RC5-CTS".
+ *
+ * <p>This mode is implemented independently of a particular cipher.
+ * Ciphers to which this mode should apply (e.g., DES) must be
+ * <i>plugged-in</i> using the constructor.
+ *
+ * <p>NOTE#1: CTS requires the input data to be at least one block
+ * long. Thus, callers of this class has to buffer the input data
+ * to make sure the input data passed to encryptFinal()/decryptFinal()
+ * is not shorter than a block.
+ * <p>NOTE#2: This class does not deal with buffering or padding
+ * just like all other cipher mode implementations.
+ *
+ * @author Valerie Peng
+ */
+
+final class CipherTextStealing extends CipherBlockChaining {
+
+    CipherTextStealing(SymmetricCipher embeddedCipher) {
+        super(embeddedCipher);
+    }
+
+    /**
+     * Gets the name of this feedback mode.
+     *
+     * @return the string <code>CBC</code>
+     */
+    String getFeedback() {
+        return "CTS";
+    }
+
+    /**
+     * Performs the last encryption operation.
+     *
+     * <p>The input plain text <code>plain</code>, starting at
+     * <code>plainOffset</code> and ending at
+     * <code>(plainOffset + len - 1)</code>, is encrypted.
+     * The result is stored in <code>cipher</code>, starting at
+     * <code>cipherOffset</code>.
+     *
+     * <p>It is the application's responsibility to make sure that
+     * <code>plainLen</code> is a multiple of the embedded cipher's block size,
+     * as any excess bytes are ignored.
+     *
+     * @param plain the buffer with the input data to be encrypted
+     * @param plainOffset the offset in <code>plain</code>
+     * @param plainLen the length of the input data
+     * @param cipher the buffer for the result
+     * @param cipherOffset the offset in <code>cipher</code>
+     */
+    void encryptFinal(byte[] plain, int plainOffset, int plainLen,
+                      byte[] cipher, int cipherOffset)
+        throws IllegalBlockSizeException {
+
+        if (plainLen < blockSize) {
+            throw new IllegalBlockSizeException("input is too short!");
+        } else if (plainLen == blockSize) {
+            encrypt(plain, plainOffset, plainLen, cipher, cipherOffset);
+        } else {
+            // number of bytes in the last block
+            int nLeft = plainLen % blockSize;
+            if (nLeft == 0) {
+                encrypt(plain, plainOffset, plainLen, cipher, cipherOffset);
+                // swap the last two blocks after encryption
+                int lastBlkIndex = cipherOffset + plainLen - blockSize;
+                int nextToLastBlkIndex = lastBlkIndex - blockSize;
+                byte[] tmp = new byte[blockSize];
+                System.arraycopy(cipher, lastBlkIndex, tmp, 0, blockSize);
+                System.arraycopy(cipher, nextToLastBlkIndex,
+                                 cipher, lastBlkIndex, blockSize);
+                System.arraycopy(tmp, 0, cipher, nextToLastBlkIndex,
+                                 blockSize);
+            } else {
+                int newPlainLen = plainLen - (blockSize + nLeft);
+                if (newPlainLen > 0) {
+                    encrypt(plain, plainOffset, newPlainLen, cipher,
+                            cipherOffset);
+                    plainOffset += newPlainLen;
+                    cipherOffset += newPlainLen;
+                }
+
+                // Do final CTS step for last two blocks (the second of which
+                // may or may not be incomplete).
+                byte[] tmp = new byte[blockSize];
+                // now encrypt the next-to-last block
+                for (int i = 0; i < blockSize; i++) {
+                    tmp[i] = (byte) (plain[plainOffset+i] ^ r[i]);
+                }
+                byte[] tmp2 = new byte[blockSize];
+                embeddedCipher.encryptBlock(tmp, 0, tmp2, 0);
+                System.arraycopy(tmp2, 0, cipher,
+                                 cipherOffset+blockSize, nLeft);
+                // encrypt the last block
+                for (int i=0; i<nLeft; i++) {
+                    tmp2[i] = (byte)
+                        (plain[plainOffset+blockSize+i] ^ tmp2[i]);
+                }
+                embeddedCipher.encryptBlock(tmp2, 0, cipher, cipherOffset);
+            }
+        }
+    }
+
+    /**
+     * Performs decryption operation.
+     *
+     * <p>The input cipher text <code>cipher</code>, starting at
+     * <code>cipherOffset</code> and ending at
+     * <code>(cipherOffset + len - 1)</code>, is decrypted.
+     * The result is stored in <code>plain</code>, starting at
+     * <code>plainOffset</code>.
+     *
+     * <p>It is the application's responsibility to make sure that
+     * <code>cipherLen</code> is a multiple of the embedded cipher's block
+     * size, as any excess bytes are ignored.
+     *
+     * <p>It is also the application's responsibility to make sure that
+     * <code>init</code> has been called before this method is called.
+     * (This check is omitted here, to avoid double checking.)
+     *
+     * @param cipher the buffer with the input data to be decrypted
+     * @param cipherOffset the offset in <code>cipherOffset</code>
+     * @param cipherLen the length of the input data
+     * @param plain the buffer for the result
+     * @param plainOffset the offset in <code>plain</code>
+     */
+    void decryptFinal(byte[] cipher, int cipherOffset, int cipherLen,
+                      byte[] plain, int plainOffset)
+        throws IllegalBlockSizeException {
+        if (cipherLen < blockSize) {
+            throw new IllegalBlockSizeException("input is too short!");
+        } else if (cipherLen == blockSize) {
+            decrypt(cipher, cipherOffset, cipherLen, plain, plainOffset);
+        } else {
+            // number of bytes in the last block
+            int nLeft = cipherLen % blockSize;
+            if (nLeft == 0) {
+                // swap the last two blocks before decryption
+                int lastBlkIndex = cipherOffset + cipherLen - blockSize;
+                int nextToLastBlkIndex =
+                    cipherOffset + cipherLen - 2*blockSize;
+                byte[] tmp = new byte[2*blockSize];
+                System.arraycopy(cipher, lastBlkIndex, tmp, 0, blockSize);
+                System.arraycopy(cipher, nextToLastBlkIndex,
+                                 tmp, blockSize, blockSize);
+                int cipherLen2 = cipherLen-2*blockSize;
+                decrypt(cipher, cipherOffset, cipherLen2, plain, plainOffset);
+                decrypt(tmp, 0, 2*blockSize, plain, plainOffset+cipherLen2);
+            } else {
+                int newCipherLen = cipherLen-(blockSize+nLeft);
+                if (newCipherLen > 0) {
+                    decrypt(cipher, cipherOffset, newCipherLen, plain,
+                            plainOffset);
+                    cipherOffset += newCipherLen;
+                    plainOffset += newCipherLen;
+                }
+                // Do final CTS step for last two blocks (the second of which
+                // may or may not be incomplete).
+
+                // now decrypt the next-to-last block
+                byte[] tmp = new byte[blockSize];
+                embeddedCipher.decryptBlock(cipher, cipherOffset, tmp, 0);
+                for (int i = 0; i < nLeft; i++) {
+                    plain[plainOffset+blockSize+i] =
+                        (byte) (cipher[cipherOffset+blockSize+i] ^ tmp[i]);
+                }
+
+                // decrypt the last block
+                System.arraycopy(cipher, cipherOffset+blockSize, tmp, 0,
+                                 nLeft);
+                embeddedCipher.decryptBlock(tmp, 0, plain, plainOffset);
+                //System.arraycopy(r, 0, tmp, 0, r.length);
+                for (int i=0; i<blockSize; i++) {
+                    plain[plainOffset+i] = (byte)
+                        (plain[plainOffset+i]^r[i]);
+                }
+            }
+        }
+    }
+}