JAVA中RSA的类

JAVA中RSA的类

别处摘的，先保存下来

package com.google.algorithm;
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.OutputStreamWriter;
import java.math.BigInteger;
import java.security.KeyPair;
import java.security.KeyPairGenerator;
import java.security.NoSuchAlgorithmException;
import java.security.PrivateKey;
import java.security.PublicKey;
import java.security.SecureRandom;
import java.security.interfaces.RSAPrivateKey;
import java.security.interfaces.RSAPublicKey;

public class RSA
{
public static void main(String[] args) throws NoSuchAlgorithmException, IOException, ClassNotFoundException
{
KeyPair pair = generateKeyPair();
PrivateKey privateKey = pair.getPrivate();
PublicKey publicKey = pair.getPublic();

// Save Pirvate Key
FileOutputStream f1 = new FileOutputStream("private.ppk");
ObjectOutputStream b1 = new ObjectOutputStream(f1);
b1.writeObject(privateKey);

// Save Public Key
FileOutputStream f2 = new FileOutputStream("public.ppk");
ObjectOutputStream b2 = new ObjectOutputStream(f2);
b2.writeObject(publicKey);

encrypt();

decrypt();

}

public static void encrypt() throws IOException, ClassNotFoundException
{
String msg = "Hello World!";

FileInputStream f = new FileInputStream("public.ppk");
ObjectInputStream b = new ObjectInputStream(f);
RSAPublicKey pbk = (RSAPublicKey) b.readObject();

// RSA算法是使用整数进行加密的，在RSA公钥中包含有两个整数信息：e和n。对于明文数字m,计算密文的公式是m的e次方再与n求模。
BigInteger e = pbk.getPublicExponent();
BigInteger n = pbk.getModulus();

// 获取明文的大整数
byte ptext[] = msg.getBytes("utf-8");
BigInteger m = new BigInteger(ptext);

// 加密明文
BigInteger c = m.modPow(e, n);

// 打印密文c
System.out.println("c= " + c);

// 将密文以字符串形式保存在文件中
String cs = c.toString();
BufferedWriter out = new BufferedWriter(new OutputStreamWriter(new FileOutputStream("encrypt.dat"), "utf-8"));
out.write(cs, 0, cs.length());
out.close();

}

public static void decrypt() throws IOException, ClassNotFoundException
{
BufferedReader in = new BufferedReader(new InputStreamReader(new FileInputStream("encrypt.dat"), "utf-8"));
String ctext = in.readLine();
BigInteger c = new BigInteger(ctext);

// 获取私钥
FileInputStream f = new FileInputStream("private.ppk");
ObjectInputStream b = new ObjectInputStream(f);
RSAPrivateKey prk = (RSAPrivateKey) b.readObject();

// 获取私钥的参数d,n
BigInteger d = prk.getPrivateExponent();
BigInteger n = prk.getModulus();

// 解密明文
BigInteger m = c.modPow(d, n);


// 计算明文对应的字符串并输出。
byte[] mt = m.toByteArray();
System.out.println("PlainText is " + new String(mt, "utf-8"));

}

public static KeyPair generateKeyPair() throws NoSuchAlgorithmException
{
KeyPairGenerator keyPairGen = KeyPairGenerator.getInstance("RSA");
final int KEY_SIZE = 1024;
keyPairGen.initialize(KEY_SIZE, new SecureRandom());
KeyPair keyPair = keyPairGen.genKeyPair();
return keyPair;
}

}