Java Tutorial/Security/SHA Secure Hash Algorithm

Материал из Java эксперт
Перейти к: навигация, поиск

An alternate SHA Interleave algorithm as implemented in the SRP distribution

   <source lang="java">

/*

* JBoss, Home of Professional Open Source.
* Copyright 2008, Red Hat Middleware LLC, and individual contributors
* as indicated by the @author tags. See the copyright.txt file in the
* distribution for a full listing of individual contributors.
*
* This is free software; you can redistribute it and/or modify it
* under the terms of the GNU Lesser General Public License as
* published by the Free Software Foundation; either version 2.1 of
* the License, or (at your option) any later version.
*
* This software 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
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this software; if not, write to the Free
* Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
* 02110-1301 USA, or see the FSF site: http://www.fsf.org.
*/

import java.io.ByteArrayOutputStream; import java.security.MessageDigest; import java.security.MessageDigestSpi; import java.security.NoSuchAlgorithmException; import java.security.ProviderException; /** An alternate SHA Interleave algorithm as implemented in the SRP

distribution. This version reverses the even and odd byte streams before
performing the SHA digest.
This product includes software developed by Tom Wu and Eugene
Jhong for the SRP Distribution (http://srp.stanford.edu/srp/).

@author Scott.Stark@jboss.org @version $Revision: 85945 $

  • /

public class SHAReverseInterleave extends MessageDigestSpi {

  private static final int SHA_HASH_LEN = 20;
  private ByteArrayOutputStream evenBytes;
  private ByteArrayOutputStream oddBytes;
  private int count;
  private boolean skipLeadingZeros;
  private MessageDigest sha;
  /** Creates a new instance of SHAReverseInterleave
   @exception ProviderException thrown if MessageDigest.getInstance("SHA")
   throws a NoSuchAlgorithmException.
   */
  public SHAReverseInterleave()
  {
     try
     {
        sha = MessageDigest.getInstance("SHA");
     }
     catch(NoSuchAlgorithmException e)
     {
        throw new ProviderException("Failed to obtain SHA MessageDigest");
     }
     evenBytes = new ByteArrayOutputStream();
     oddBytes = new ByteArrayOutputStream();
     engineReset();
  }
  protected int engineGetDigestLength()
  {
     return 2 * SHA_HASH_LEN;
  }
  /**
   * Completes the digest computation by performing final
   * operations such as padding. Once engineDigest has
   * been called, the engine should be reset (see
   * {@link #engineReset() engineReset}).
   * Resetting is the responsibility of the
   * engine implementor.
   *
   * @return the array of bytes for the resulting digest value.
   */
  protected byte[] engineDigest()
  {
     byte[] E = evenBytes.toByteArray();
     // If the count is odd, drop the first byte
     int length = E.length;
     if( count % 2 == 1 )
        length --;
     // Reverse the order of the even bytes
     byte[] tmp = new byte[length];
     for(int i = 0; i < length; i ++)
     {
        tmp[i] = E[E.length - i - 1];
        System.out.println("E["+i+"] = "+tmp[i]);
     }
     E = tmp;
     byte[] G = sha.digest(E);
     byte[] F = oddBytes.toByteArray();
     // Reverse the order of the even bytes
     tmp = new byte[F.length];
     for(int i = 0; i < F.length; i ++)
     {
        tmp[i] = F[F.length - i - 1];
        System.out.println("F["+i+"] = "+tmp[i]);
     }
     F = tmp;
     sha.reset();
     byte[] H = sha.digest(F);
     length = G.length + H.length;
     byte[] digest = new byte[length];
     for(int i = 0; i < G.length; ++i)
        digest[2 * i] = G[i];
     for(int i = 0; i < H.length; ++i)
        digest[2 * i + 1] = H[i];
     engineReset();
     return digest;
  }
  /**
   * Resets the digest for further use.
   */
  protected void engineReset()
  {
     skipLeadingZeros = true;
     count = 0;
     evenBytes.reset();
     oddBytes.reset();
     sha.reset();
  }
  /**
   * Updates the digest using the specified byte.
   *
   * @param input the byte to use for the update.
   */
  protected void engineUpdate(byte input)
  {
     if( skipLeadingZeros == true && input == 0 )
        return;
     skipLeadingZeros = false;
     if( count % 2 == 0 )
        evenBytes.write(input);
     else
        oddBytes.write(input);
     count ++;
  }
  /**
   * Updates the digest using the specified array of bytes,
   * starting at the specified offset.
   *
   * @param input the array of bytes to use for the update.
   * @param offset the offset to start from in the array of bytes.
   * @param len the input of bytes to use, starting at
   * offset.
   */
  protected void engineUpdate(byte[] input, int offset, int len)
  {
     for(int i = offset; i < offset+len; i ++)
        engineUpdate(input[i]);
  }

}</source>





Receive Self Mac

   <source lang="java">

import java.io.FileInputStream; import java.io.ObjectInputStream; import java.security.MessageDigest; public class MainClass {

 public static void main(String args[]) throws Exception {
   FileInputStream fis = new FileInputStream("test");
   ObjectInputStream ois = new ObjectInputStream(fis);
   Object o = ois.readObject();
   if (!(o instanceof String)) {
     System.out.println("Unexpected data in file");
     System.exit(-1);
   }
   String data = (String) o;
   System.out.println("Got message " + data);
   o = ois.readObject();
   if (!(o instanceof byte[])) {
     System.out.println("Unexpected data in file");
     System.exit(-1);
   }
   byte origDigest[] = (byte[]) o;
   byte pass[] = "aaa".getBytes();
   byte buf[] = data.getBytes();
   MessageDigest md = MessageDigest.getInstance("SHA");
   md.update(pass);
   md.update(buf);
   byte digest1[] = md.digest();
   md.update(pass);
   md.update(digest1);
   System.out.println(MessageDigest.isEqual(md.digest(), origDigest));
 }

}</source>





Save SHA Message

   <source lang="java">

import java.io.FileOutputStream; import java.io.ObjectOutputStream; import java.security.MessageDigest; public class MainClass {

 public static void main(String args[]) throws Exception {
   FileOutputStream fos = new FileOutputStream("test");
   MessageDigest md = MessageDigest.getInstance("SHA");
   ObjectOutputStream oos = new ObjectOutputStream(fos);
   String data = "thee";
   byte buf[] = data.getBytes();
   md.update(buf);
   oos.writeObject(data);
   oos.writeObject(md.digest());
 }

}</source>





SHA MessageDigest

   <source lang="java">

import java.io.FileInputStream; import java.io.ObjectInputStream; import java.security.MessageDigest; public class MainClass {

 public static void main(String args[]) throws Exception {
   FileInputStream fis = new FileInputStream("test");
   ObjectInputStream ois = new ObjectInputStream(fis);
   Object o = ois.readObject();
   if (!(o instanceof String)) {
     System.out.println("Unexpected data in file");
     System.exit(-1);
   }
   String data = (String) o;
   System.out.println("Got message " + data);
   o = ois.readObject();
   if (!(o instanceof byte[])) {
     System.out.println("Unexpected data in file");
     System.exit(-1);
   }
   byte origDigest[] = (byte[]) o;
   MessageDigest md = MessageDigest.getInstance("SHA");
   md.update(data.getBytes());
   if (MessageDigest.isEqual(md.digest(), origDigest))
     System.out.println("Message is valid");
   else
     System.out.println("Message was corrupted");
 }

}</source>