mcryptmodule.java

来自「RESIN 3.2 最新源码」· Java 代码 · 共 695 行 · 第 1/2 页

/*
 * Copyright (c) 1998-2008 Caucho Technology -- all rights reserved
 *
 * This file is part of Resin(R) Open Source
 *
 * Each copy or derived work must preserve the copyright notice and this
 * notice unmodified.
 *
 * Resin Open Source is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * Resin Open Source 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, or any warranty
 * of NON-INFRINGEMENT.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Resin Open Source; if not, write to the
 *
 *   Free Software Foundation, Inc.
 *   59 Temple Place, Suite 330
 *   Boston, MA 02111-1307  USA
 *
 * @author Scott Ferguson
 */

package com.caucho.quercus.lib.mcrypt;

import com.caucho.quercus.annotation.Optional;
import com.caucho.quercus.annotation.ReturnNullAsFalse;
import com.caucho.quercus.env.ArrayValue;
import com.caucho.quercus.env.ArrayValueImpl;
import com.caucho.quercus.env.StringValue;
import com.caucho.quercus.env.BooleanValue;
import com.caucho.quercus.env.Env;
import com.caucho.quercus.env.LongValue;
import com.caucho.quercus.env.Value;
import com.caucho.quercus.module.AbstractQuercusModule;
import com.caucho.util.L10N;
import com.caucho.util.RandomUtil;
import com.caucho.vfs.Path;

import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * PHP encryption
 */
public class McryptModule extends AbstractQuercusModule {
  private static final L10N L = new L10N(McryptModule.class);

  private static final Logger log =
    Logger.getLogger(McryptModule.class.getName());

  public static final int MCRYPT_DEV_RANDOM = 0;
  public static final int MCRYPT_DEV_URANDOM = 1;
  public static final int MCRYPT_RAND = 2;

  public static final int MCRYPT_ENCRYPT = 0;
  public static final int MCRYPT_DECRYPT = 1;

  public static final String MCRYPT_MODE_ECB = "ecb";
  public static final String MCRYPT_MODE_CBC = "cbc";
  public static final String MCRYPT_MODE_CFB = "cfb";
  public static final String MCRYPT_MODE_OFB = "ofb";
  public static final String MCRYPT_MODE_NOFB = "nofb";
  public static final String MCRYPT_MODE_STREAM = "stream";

  public static final String MCRYPT_ARCFOUR = "arcfour";
  public static final String MCRYPT_BLOWFISH = "blowfish";
  public static final String MCRYPT_DES = "des";
  public static final String MCRYPT_3DES = "tripledes";
  public static final String MCRYPT_TRIPLEDES = "tripledes";
  public static final String MCRYPT_RC4 = "RC4";
  public static final String MCRYPT_RIJNDAEL_128 = "rijndael-128";
  public static final String MCRYPT_RIJNDAEL_192 = "rijndael-192";
  public static final String MCRYPT_RIJNDAEL_256 = "rijndael-256";

  public String []getLoadedExtensions()
  {
    return new String[] {  "mcrypt" };
  }

  /**
   * Encrypt with cbc
   */
  public static StringValue mcrypt_cbc(Env env,
				       String cipher,
				       byte []key,
				       byte []data,
				       int mode,
				       @Optional byte []iv)
  {
    try {
      Mcrypt mcrypt = new Mcrypt(env, cipher, "cbc");

      mcrypt.init(key, iv);

      byte []result;
      
      if (mode == MCRYPT_ENCRYPT)
        result = mcrypt.encrypt(data);
      else
        result = mcrypt.decrypt(data);
      
      return env.createBinaryBuilder(result, 0, result.length);
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  /**
   * Encrypt with cfb
   */
  public static StringValue mcrypt_cfb(Env env,
				       String cipher,
				       byte []key,
				       byte []data,
				       int mode,
				       @Optional byte []iv)
  {
    try {
      Mcrypt mcrypt = new Mcrypt(env, cipher, "cfb");

      mcrypt.init(key, iv);

      byte []result;
      
      if (mode == MCRYPT_ENCRYPT)
        result = mcrypt.encrypt(data);
      else
        result = mcrypt.decrypt(data);
      
      return env.createBinaryBuilder(result, 0, result.length);
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  /**
   * Creates the IV vector.
   */
  public static StringValue mcrypt_create_iv(Env env,
					     int size,
					     @Optional int randomMode)
  {
    StringValue bb = env.createBinaryBuilder(size);

    for (int i = 0; i < size; i++)
      bb.appendByte((byte) RandomUtil.nextInt(256));

    return bb;
  }

  /**
   * Decrypt
   */
  public static StringValue mcrypt_decrypt(Env env,
                                           String cipher,
                                           byte []key,
                                           byte []data,
                                           String mode,
                                           @Optional byte []iv)
  {
    try {
      Mcrypt mcrypt = new Mcrypt(env, cipher, mode);

      mcrypt.init(key, iv);

      byte []result = mcrypt.decrypt(data);

      return env.createBinaryBuilder(result, 0, result.length);
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  /**
   * Encrypt with cfb
   */
  public static StringValue mcrypt_ecb(Env env,
                                       String cipher,
                                       byte []key,
                                       byte []data,
                                       int mode,
                                       @Optional byte []iv)
  {
    try {
      Mcrypt mcrypt = new Mcrypt(env, cipher, "ecb");

      mcrypt.init(key, iv);

      byte []result;
      
      if (mode == MCRYPT_ENCRYPT)
        result = mcrypt.encrypt(data);
      else
        result = mcrypt.decrypt(data);
      
      return env.createBinaryBuilder(result, 0, result.length);
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  /**
   * Returns the algorithms name
   */
  public static String mcrypt_enc_get_algorithms_name(Mcrypt mcrypt)
  {
    if (mcrypt != null)
      return mcrypt.get_algorithms_name();
    else
      return "";
  }

  /**
   * Returns the block size
   */
  public static int mcrypt_enc_get_block_size(Mcrypt mcrypt)
  {
    if (mcrypt != null)
      return mcrypt.get_iv_size();
    else
      return 0;
  }

  /**
   * Returns the IV size
   */
  public static int mcrypt_enc_get_iv_size(Mcrypt mcrypt)
  {
    if (mcrypt != null)
      return mcrypt.get_iv_size();
    else
      return 0;
  }

  /**
   * Returns the key size
   */
  public static int mcrypt_enc_get_key_size(Mcrypt mcrypt)
  {
    if (mcrypt != null)
      return mcrypt.get_key_size();
    else
      return 0;
  }

  /**
   * Returns the mode name
   */
  public static String mcrypt_enc_get_modes_name(Mcrypt mcrypt)
  {
    if (mcrypt != null)
      return mcrypt.get_modes_name();
    else
      return null;
  }

  /**
   * Returns the supported key sizes
   */
  public static Value mcrypt_enc_get_supported_key_sizes(Mcrypt mcrypt)
  {
    if (mcrypt != null)
      return mcrypt.get_supported_key_sizes();
    else
      return BooleanValue.FALSE;
  }

  /**
   * Returns true for block encoding modes
   */
  public static boolean mcrypt_enc_is_block_algorithm(Mcrypt mcrypt)
  {
    if (mcrypt != null)
      return mcrypt.is_block_algorithm();
    else
      return false;
  }

  /**
   * Returns true for block encoding modes
   */
  public static boolean mcrypt_enc_is_block_algorithm_mode(Mcrypt mcrypt)
  {
    if (mcrypt != null)
      return mcrypt.is_block_algorithm_mode();
    else
      return false;
  }

  /**
   * Returns true for block output modes
   */
  public static boolean mcrypt_enc_is_block_mode(Mcrypt mcrypt)
  {
    if (mcrypt != null)
      return mcrypt.is_block_mode();
    else
      return false;
  }

  /**
   * Returns true for block output modes
   */
  public static boolean mcrypt_enc_self_test(Mcrypt mcrypt)
  {
    if (mcrypt != null)
      return true;
    else
      return false;
  }

  /**
   * Encrypt
   */
  public static StringValue mcrypt_encrypt(Env env,
                                           String cipher,
                                           byte []key,
                                           byte []data,
                                           String mode,
                                           @Optional byte []iv)
  {
    try {
      Mcrypt mcrypt = new Mcrypt(env, cipher, mode);

      mcrypt.init(key, iv);

      return env.createBinaryBuilder(mcrypt.encrypt(data));
    } catch (Exception e) {
      throw new RuntimeException(e);
    }
  }

  /**
   * Initialize encrption
   */
  public static StringValue mcrypt_generic(Env env, Mcrypt mcrypt, byte []data)
  {
    if (mcrypt == null)
      return null;
    else
      return env.createBinaryBuilder(mcrypt.encrypt(data));