/*
 * 所有的字符串作為輸入時.
 * ***********************************************************
 * 在加密和解密的初始階段，輸入數(shù)組 in 按照下述規(guī)則復(fù)制到狀態(tài)矩陣中： s[r,c]=in[r+4c]
 * for 0 ≤ r < 4 且 0 ≤ c < Nb
 * 在加密和解密的結(jié)束階段，狀態(tài)矩陣將按照下述規(guī)則被復(fù)制到輸出數(shù)組 out 中： out[r+4c]=
 * s[r,c] for 0 ≤ r < 4 且 0 ≤ c < Nb
 */
public class Rijndael {
  /* 狀態(tài)包含的列(32-bit 字)的個數(shù)。對于AES，分組輸入為128bit時,Nb＝4 */
  private final int Nb = 4;
  private String input = "";
  private String state = "";

  public Rijndael() {
  }
  public Rijndael(String state) {
    this.state = state;
    this.input = state;
  }
  /**
   * 字節(jié)變換,將一串輸入通過SBox變換輸出.
   * @param input
   *          type : 16Byte String .
   * @return state .type : 16Byte String .
   */
  public String subBytes(String input) {
    Operator op = new Operator();
    this.state = subBytes(op.ToByte(input));
    return this.state;
  }
  public String invSubBytes(String input) {
    Operator op = new Operator();
    this.state = invSubBytes(op.ToByte(input));
    return this.state;
  }
  public String invSubBytes(byte[] input) {
    SBox sBox = new SBox();
    int length = input.length;
    String Tstate = "";
    for (int i = 0; i < length; i++) {
      Tstate += (char) sBox.invSubByte((byte) input[i]);
    }
    return Tstate;
  }
  /**
   * 字節(jié)變換,將一串輸入通過SBox變換輸出.
   * @param input
   *          type : Byte[16]
   * @return state .type : 16Byte String .
   */
  public String subBytes(byte[] input) {
    SBox sBox = new SBox();
    int length = input.length;
    String Tstate = "";
    for (int i = 0; i < length; i++) {
      Tstate += (char) sBox.subByte((byte) input[i]);
    }
    return Tstate;
  }
  /**
   * 行移位.將16個字符看成4個字節(jié)為一列順序,組成的4*4的矩陣. *
   * 將輸入的矩陣的每行按行號(from 0)大小進(jìn)行循環(huán)左移.
   * @param input
   * @return state .type : 16Byte String .
   */
  public String shiftRows(String input) {
    this.state = "";
    byte[][] inputMatrix = new byte[4][Nb];
    byte[][] stateMatrix = new byte[4][Nb];
    stringToMatrix(input, inputMatrix);// 將字符串的輸入,放入一個矩陣.
    /*
     * 先按行操作.對輸入矩陣進(jìn)行操作,并復(fù)制給狀態(tài)矩陣.****************
     * 將行中的4個字節(jié),按行號(from 0)大小進(jìn)行循環(huán)左移
     */
    for (int r = 0; r < 4; r++) {
      // 行序
      for (int c = 0; c < Nb; c++) {
        // 列序
        stateMatrix[r][c] = inputMatrix[r][(c + r) % Nb];
      }
    }
    matrixToString(stateMatrix);// 將狀態(tài)矩陣輸入到字符串.
    /*
     * 置空,讓JRE自動處理內(nèi)存. inputMatrix = null; stateMatrix =
     * null;
     */
    return this.state;
  }
  /**
   * 逆算行移位.將16個字符看成4個字節(jié)為一列順序,組成的4*4的矩陣. *
   * 將輸入的矩陣的每行按行號(from 3 to 0)大小進(jìn)行循環(huán)右移.
   * @param input
   * @return state .type : 16Byte String .
   */
  public String invShiftRows(String input) {
    this.state = "";
    byte[][] inputMatrix = new byte[4][Nb];
    byte[][] stateMatrix = new byte[4][Nb];
    stringToMatrix(input, inputMatrix);// 將字符串的輸入,放入一個矩陣.
    /*
     * 先按行操作.對輸入矩陣進(jìn)行操作,并復(fù)制給狀態(tài)矩陣.****************
     * 將行中的4個字節(jié),按行號(from 3 to 0)大小進(jìn)行循環(huán)右移
     */
    for (int r = 0; r < 4; r++) {
      // 行序
      for (int c = 0; c < Nb; c++) {
        // 列序
        stateMatrix[r][c] = inputMatrix[r][(c - (r + 1) + Nb) % Nb];
      }
    }
    matrixToString(stateMatrix);// 將狀態(tài)矩陣輸入到字符串.
    /*
     * 置空,讓JRE自動處理內(nèi)存. inputMatrix = null; stateMatrix =
     * null;
     */
    return this.state;
  }
  /**
   * 將狀態(tài)矩陣輸入到字符串.以每列為單位.
   * @param matrix
   */
  private void matrixToString(byte[][] matrix) {
    /* 先按列操作. 將狀態(tài)矩陣的元素賦給字符串. */
    for (int c = 0; c < Nb; c++) {
      // 列序
      for (int r = 0; r < 4; r++) {
        // 行序
        this.state += (char) matrix[r][c];
      }
    }
  }
  /**
   * 將字符串的輸入,放入一個矩陣. 每4字節(jié)為單位先按每列進(jìn)行放.
   * @param input
   * @param matrix
   */
  private void stringToMatrix(String input, byte[][] matrix) {
    /* 按每列操作. 將字符串放放輸入矩陣 */
    for (int c = 0; c < Nb; c++) {
      // 列序
      for (int r = 0; r < 4; r++) {
        // 行序
        matrix[r][c] = (byte) input.charAt(r + c * 4);
      }
    }
  }
  /**
   * 列混合.Hex {03,01,01,02}在行上按行號+1,即(1~4)循環(huán)右移.***
   * 相乘,按GF(2^8次)Mod.得到狀態(tài)矩陣.***********************
   * GF(8)=x^8+ x^4+x^3+x^1+1,Binary : 1 0001 1011.*
   * GF(8)= 11B(16進(jìn)制.).
   * @param input
   *          type : String
   * @return state .type : 16Byte String .
   */
  public String mixColumns(String input) {
    this.state = "";
    byte[][] aMatrix = new byte[4][4];// AES中定義的a(x).
    byte[] temp = { (byte) 0x03, (byte) 0x01, (byte) 0x01, (byte) 0x02 };
    /*
     * 循環(huán)右移. 移位個數(shù):1~4 , 賦給a(x)
     */
    for (int r = 0; r < 4; r++) {
      // 行序
      for (int c = 0; c < Nb; c++) {
        // 列序
        aMatrix[r][c] = temp[(c - (r + 1) + Nb) % Nb];
        // System.out.print((byte)aMatrix[r][c] + " ");
      }
      // System.out.println();
    }
    byte[][] inputMatrix = new byte[4][Nb];
    stringToMatrix(input, inputMatrix);// 賦給狀態(tài)矩陣
    Operator op = new Operator();
    byte[][] stateMatrix = new byte[4][Nb];
    for (int r = 0; r < Nb; r++) {
      for (int m = 0; m < 4; m++) {
        for (int q = 0; q < 4; q++) {
          stateMatrix[m][r] ^=
            op.mulByTwoPolynomials(aMatrix[m][q], inputMatrix[q][r]);
        }
      }
    }
    this.matrixToString(stateMatrix);// 將矩陣賦給字符串.
    return this.state;
  }
  /**
   * 逆算列混合.Hex
   * {0x0b,0x0d,0x09,0x0e}在行上按行號+1,即(1~4)循環(huán)右移.
   * 相乘,按GF(2^8次)Mod.得到狀態(tài)矩陣.***********************
   * GF(8)=x^8+ x^4+x^3+x^1+1,Binary : 1 0001 1011.*
   * GF(8)= 11B(16進(jìn)制.).
   * @param input
   *          type : String
   * @return state .type : 16Byte String .
   */
  public String invMixColumns(String input) {
    this.state = "";
    byte[][] aMatrix = new byte[4][4];// AES中定義的a(x).
    byte[] temp = { (byte) 0x0b, (byte) 0x0d, (byte) 0x09, (byte) 0x0e };
    /*
     * 循環(huán)右移. 移位個數(shù):1~4 , 賦給a(x)
     */
    for (int r = 0; r < 4; r++) {
      // 行序
      for (int c = 0; c < Nb; c++) {
        // 列序
        aMatrix[r][c] = temp[(c - (r + 1) + Nb) % Nb];
        // System.out.print((byte)aMatrix[r][c] + " ");
      }
      // System.out.println();
    }
    byte[][] inputMatrix = new byte[4][Nb];
    stringToMatrix(input, inputMatrix);// 賦給狀態(tài)矩陣
    Operator g = new Operator();
    byte[][] stateMatrix = new byte[4][Nb];
    for (int r = 0; r < Nb; r++) {
      for (int m = 0; m < 4; m++) {
        for (int q = 0; q < 4; q++) {
          stateMatrix[m][r] ^=
            g.mulByTwoPolynomials(aMatrix[m][q], inputMatrix[q][r]);
        }
      }
    }
    this.matrixToString(stateMatrix);// 將矩陣賦給字符串.
    return this.state;
  }
  /**
   * 加入輪密鑰.
   * @param input
   *          type : String
   * @return state .type : 16Byte String .
   */
  public String addRoundKey(String input, String key) {
    String mw = "";
    Operator g = new Operator();
    int keyc = key.length() / 4;
    byte[][] w = new byte[keyc][4];
    byte[][] k = new byte[keyc][4];
    g.ToByte(w, input);
    g.ToByte(k, key);
    for (int i = 0; i < keyc; i++) {
      w[i] = g.XOR(w[i], k[i]);
      mw += g.ToString(w[i]);
    }
    // System.out.println(mw);
    return mw;
  }
  /**
   * rijndael算法.合并了四步.
   * @param input
   * @return state .type : 16Byte String .
   */
  public String rijndael(String input, String key) {
    this.state = addRoundKey(input, key);
    Key K = new Key();
    for (int i = 1; i <= 10; i++) {
      key = K.SKey(key, i);
      this.state = subBytes(this.state);
      this.state = shiftRows(this.state);
      if (i < 10) {
        this.state = mixColumns(this.state);
      }
      this.state = addRoundKey(this.state, key);
    }
    return this.state;
  }
  public static void main(String[] args) {
    Rijndael r = new Rijndael();
    String input = "";
    String key = "";
    byte[] input1 =
      { 0x00, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, (byte) 0x88,
          (byte) 0x99, (byte) 0xaa, (byte) 0xbb, (byte) 0xcc, (byte) 0xdd,
          (byte) 0xee, (byte) 0xff };
    byte[] key1 =
      { 0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0a, 0x0b,
          0x0c, 0x0d, 0x0e, 0x0f };
    for (int i = 0; i <= 15; i++) {
      input += (char) input1[i];
      key += (char) key1[i];
    }
    String z = r.rijndael(input, key);
    System.out.println(z);
    for (int i = 0; i < z.length(); i++) {
      byte MN = (byte) z.charAt(i);
      System.out.print(Integer.toHexString(MN) + "****");
    }
  }
  public String getState() {
    return state;
  }
  private void setState(String state) {
    this.state = state;
  }
  public String getInput() {
    return input;
  }
  public void setInput(String input) {
    this.input = input;
  }
}