Subversion Repositories aes-encryption

/////////////////////////////////////////////////////////////////////

////                                                             ////

////  AES Cipher Top Level                                       ////

////                                                             ////

////                                                             ////

////  Author: Rudolf Usselmann                                   ////

////          rudi@asics.ws                                      ////

////                                                             ////

////                                                             ////

////  Downloaded from: http://www.opencores.org/cores/aes_core/  ////

////                                                             ////

/////////////////////////////////////////////////////////////////////

////                                                             ////

//// Copyright (C) 2000-2002 Rudolf Usselmann                    ////

////                         www.asics.ws                        ////

////                         rudi@asics.ws                       ////

////                                                             ////

//// This source file may be used and distributed without        ////

//// restriction provided that this copyright statement is not   ////

//// removed from the file and that any derivative work contains ////

//// the original copyright notice and the associated disclaimer.////

////                                                             ////

////     THIS SOFTWARE IS PROVIDED ``AS IS'' AND WITHOUT ANY     ////

//// EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED   ////

//// TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS   ////

//// FOR A PARTICULAR PURPOSE. IN NO EVENT SHALL THE AUTHOR      ////

//// OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,         ////

//// INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES    ////

//// (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE   ////

//// GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR        ////

//// BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF  ////

//// LIABILITY, WHETHER IN  CONTRACT, STRICT LIABILITY, OR TORT  ////

//// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT  ////

//// OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE         ////

//// POSSIBILITY OF SUCH DAMAGE.                                 ////

////                                                             ////

/////////////////////////////////////////////////////////////////////

//// Modified to achieve 10 cycle - 10 stage functionality           ////

//// By Tariq Bashir Ahmad                                           ////       

////  tariq.bashir@gmail.com                                         ////

////  http://www.ecs.umass.edu/~tbashir                              ////       

`timescale 1 ns/1 ps

module aes_cipher_top(clk, rst, ld, done, key, text_in, text_out);

input           clk, rst;

input           ld;

output          done;

input   [127:0]  key;

input   [127:0]  text_in;

output  [127:0]  text_out;

input aes_en;

reg     [127:0]  text_in_r;

reg     [127:0]  text_out_stage1;

reg     [127:0]  text_out_stage2;

reg     [127:0]  text_out_stage3;

reg     [127:0]  text_out_stage4;

reg     [127:0]  text_out_stage5;

reg     [127:0]  text_out_stage6;

reg     [127:0]  text_out_stage7;

reg     [127:0]  text_out_stage8;

reg     [127:0]  text_out_stage9;

reg     [127:0]  text_out;     //10th stage output

////////////////////////////////////////////////////////////////////

//

// Local Wires

//

wire    [31:0]   w0, w1, w2, w3, w4, w5, w6, w7, w8, w9, w10, w11, w12, w13, w14, w15, w16, w17, w18, w19, w20, w21, w22,

               w23, w24, w25, w26, w27, w28, w29, w30, w31, w32, w33, w34, w35, w36, w37, w38, w39, w40, w41, w42, w43;

/*reg   [127:0] text_out_temp;

*/

//round 1 wires

reg     [7:0]    sa00, sa01, sa02, sa03;

reg     [7:0]    sa10, sa11, sa12, sa13;

reg     [7:0]    sa20, sa21, sa22, sa23;

reg     [7:0]    sa30, sa31, sa32, sa33;

wire    [7:0]    sa00_next, sa01_next, sa02_next, sa03_next;

wire    [7:0]    sa10_next, sa11_next, sa12_next, sa13_next;

wire    [7:0]    sa20_next, sa21_next, sa22_next, sa23_next;

wire    [7:0]    sa30_next, sa31_next, sa32_next, sa33_next;

wire    [7:0]    sa00_sub, sa01_sub, sa02_sub, sa03_sub;

wire    [7:0]    sa10_sub, sa11_sub, sa12_sub, sa13_sub;

wire  [7:0]      sa20_sub, sa21_sub, sa22_sub, sa23_sub;

wire    [7:0]    sa30_sub, sa31_sub, sa32_sub, sa33_sub;

wire    [7:0]    sa00_sr, sa01_sr, sa02_sr, sa03_sr;

wire    [7:0]    sa10_sr, sa11_sr, sa12_sr, sa13_sr;

wire    [7:0]    sa20_sr, sa21_sr, sa22_sr, sa23_sr;

wire    [7:0]    sa30_sr, sa31_sr, sa32_sr, sa33_sr;

wire    [7:0]    sa00_mc, sa01_mc, sa02_mc, sa03_mc;

wire    [7:0]    sa10_mc, sa11_mc, sa12_mc, sa13_mc;

wire    [7:0]    sa20_mc, sa21_mc, sa22_mc, sa23_mc;

wire    [7:0]    sa30_mc, sa31_mc, sa32_mc, sa33_mc;

//round2 wires

wire    [7:0]    sa00_next_round2, sa01_next_round2, sa02_next_round2, sa03_next_round2;

wire    [7:0]    sa10_next_round2, sa11_next_round2, sa12_next_round2, sa13_next_round2;

wire    [7:0]    sa20_next_round2, sa21_next_round2, sa22_next_round2, sa23_next_round2;

wire    [7:0]    sa30_next_round2, sa31_next_round2, sa32_next_round2, sa33_next_round2;

wire    [7:0]    sa00_sub_round2, sa01_sub_round2, sa02_sub_round2, sa03_sub_round2;

wire    [7:0]    sa10_sub_round2, sa11_sub_round2, sa12_sub_round2, sa13_sub_round2;

wire  [7:0]      sa20_sub_round2, sa21_sub_round2, sa22_sub_round2, sa23_sub_round2;

wire    [7:0]    sa30_sub_round2, sa31_sub_round2, sa32_sub_round2, sa33_sub_round2;

wire    [7:0]    sa00_sr_round2, sa01_sr_round2, sa02_sr_round2, sa03_sr_round2;

wire    [7:0]    sa10_sr_round2, sa11_sr_round2, sa12_sr_round2, sa13_sr_round2;

wire    [7:0]    sa20_sr_round2, sa21_sr_round2, sa22_sr_round2, sa23_sr_round2;

wire    [7:0]    sa30_sr_round2, sa31_sr_round2, sa32_sr_round2, sa33_sr_round2;

wire    [7:0]    sa00_mc_round2, sa01_mc_round2, sa02_mc_round2, sa03_mc_round2;

wire    [7:0]    sa10_mc_round2, sa11_mc_round2, sa12_mc_round2, sa13_mc_round2;

wire    [7:0]    sa20_mc_round2, sa21_mc_round2, sa22_mc_round2, sa23_mc_round2;

wire    [7:0]    sa30_mc_round2, sa31_mc_round2, sa32_mc_round2, sa33_mc_round2;

//round3 wires

wire    [7:0]    sa00_next_round3, sa01_next_round3, sa02_next_round3, sa03_next_round3;

wire    [7:0]    sa10_next_round3, sa11_next_round3, sa12_next_round3, sa13_next_round3;

wire    [7:0]    sa20_next_round3, sa21_next_round3, sa22_next_round3, sa23_next_round3;

wire    [7:0]    sa30_next_round3, sa31_next_round3, sa32_next_round3, sa33_next_round3;

wire    [7:0]    sa00_sub_round3, sa01_sub_round3, sa02_sub_round3, sa03_sub_round3;

wire    [7:0]    sa10_sub_round3, sa11_sub_round3, sa12_sub_round3, sa13_sub_round3;

wire  [7:0]      sa20_sub_round3, sa21_sub_round3, sa22_sub_round3, sa23_sub_round3;

wire    [7:0]    sa30_sub_round3, sa31_sub_round3, sa32_sub_round3, sa33_sub_round3;

wire    [7:0]    sa00_sr_round3, sa01_sr_round3, sa02_sr_round3, sa03_sr_round3;

wire    [7:0]    sa10_sr_round3, sa11_sr_round3, sa12_sr_round3, sa13_sr_round3;

wire    [7:0]    sa20_sr_round3, sa21_sr_round3, sa22_sr_round3, sa23_sr_round3;

wire    [7:0]    sa30_sr_round3, sa31_sr_round3, sa32_sr_round3, sa33_sr_round3;

wire    [7:0]    sa00_mc_round3, sa01_mc_round3, sa02_mc_round3, sa03_mc_round3;

wire    [7:0]    sa10_mc_round3, sa11_mc_round3, sa12_mc_round3, sa13_mc_round3;

wire    [7:0]    sa20_mc_round3, sa21_mc_round3, sa22_mc_round3, sa23_mc_round3;

wire    [7:0]    sa30_mc_round3, sa31_mc_round3, sa32_mc_round3, sa33_mc_round3;

//round4 wires

wire    [7:0]    sa00_next_round4, sa01_next_round4, sa02_next_round4, sa03_next_round4;

wire    [7:0]    sa10_next_round4, sa11_next_round4, sa12_next_round4, sa13_next_round4;

wire    [7:0]    sa20_next_round4, sa21_next_round4, sa22_next_round4, sa23_next_round4;

wire    [7:0]    sa30_next_round4, sa31_next_round4, sa32_next_round4, sa33_next_round4;

wire    [7:0]    sa00_sub_round4, sa01_sub_round4, sa02_sub_round4, sa03_sub_round4;

wire    [7:0]    sa10_sub_round4, sa11_sub_round4, sa12_sub_round4, sa13_sub_round4;

wire  [7:0]      sa20_sub_round4, sa21_sub_round4, sa22_sub_round4, sa23_sub_round4;

wire    [7:0]    sa30_sub_round4, sa31_sub_round4, sa32_sub_round4, sa33_sub_round4;

wire    [7:0]    sa00_sr_round4, sa01_sr_round4, sa02_sr_round4, sa03_sr_round4;

wire    [7:0]    sa10_sr_round4, sa11_sr_round4, sa12_sr_round4, sa13_sr_round4;

wire    [7:0]    sa20_sr_round4, sa21_sr_round4, sa22_sr_round4, sa23_sr_round4;

wire    [7:0]    sa30_sr_round4, sa31_sr_round4, sa32_sr_round4, sa33_sr_round4;

wire    [7:0]    sa00_mc_round4, sa01_mc_round4, sa02_mc_round4, sa03_mc_round4;

wire    [7:0]    sa10_mc_round4, sa11_mc_round4, sa12_mc_round4, sa13_mc_round4;

wire    [7:0]    sa20_mc_round4, sa21_mc_round4, sa22_mc_round4, sa23_mc_round4;

wire    [7:0]    sa30_mc_round4, sa31_mc_round4, sa32_mc_round4, sa33_mc_round4;

//round5 wires

wire    [7:0]    sa00_next_round5, sa01_next_round5, sa02_next_round5, sa03_next_round5;

wire    [7:0]    sa10_next_round5, sa11_next_round5, sa12_next_round5, sa13_next_round5;

wire    [7:0]    sa20_next_round5, sa21_next_round5, sa22_next_round5, sa23_next_round5;

wire    [7:0]    sa30_next_round5, sa31_next_round5, sa32_next_round5, sa33_next_round5;

wire    [7:0]    sa00_sub_round5, sa01_sub_round5, sa02_sub_round5, sa03_sub_round5;

wire    [7:0]    sa10_sub_round5, sa11_sub_round5, sa12_sub_round5, sa13_sub_round5;

wire  [7:0]      sa20_sub_round5, sa21_sub_round5, sa22_sub_round5, sa23_sub_round5;

wire    [7:0]    sa30_sub_round5, sa31_sub_round5, sa32_sub_round5, sa33_sub_round5;

wire    [7:0]    sa00_sr_round5, sa01_sr_round5, sa02_sr_round5, sa03_sr_round5;

wire    [7:0]    sa10_sr_round5, sa11_sr_round5, sa12_sr_round5, sa13_sr_round5;

wire    [7:0]    sa20_sr_round5, sa21_sr_round5, sa22_sr_round5, sa23_sr_round5;

wire    [7:0]    sa30_sr_round5, sa31_sr_round5, sa32_sr_round5, sa33_sr_round5;

wire    [7:0]    sa00_mc_round5, sa01_mc_round5, sa02_mc_round5, sa03_mc_round5;

wire    [7:0]    sa10_mc_round5, sa11_mc_round5, sa12_mc_round5, sa13_mc_round5;

wire    [7:0]    sa20_mc_round5, sa21_mc_round5, sa22_mc_round5, sa23_mc_round5;

wire    [7:0]    sa30_mc_round5, sa31_mc_round5, sa32_mc_round5, sa33_mc_round5;

//round6 wires

wire    [7:0]    sa00_next_round6, sa01_next_round6, sa02_next_round6, sa03_next_round6;

wire    [7:0]    sa10_next_round6, sa11_next_round6, sa12_next_round6, sa13_next_round6;

wire    [7:0]    sa20_next_round6, sa21_next_round6, sa22_next_round6, sa23_next_round6;

wire    [7:0]    sa30_next_round6, sa31_next_round6, sa32_next_round6, sa33_next_round6;

wire    [7:0]    sa00_sub_round6, sa01_sub_round6, sa02_sub_round6, sa03_sub_round6;

wire    [7:0]    sa10_sub_round6, sa11_sub_round6, sa12_sub_round6, sa13_sub_round6;

wire  [7:0]      sa20_sub_round6, sa21_sub_round6, sa22_sub_round6, sa23_sub_round6;

wire    [7:0]    sa30_sub_round6, sa31_sub_round6, sa32_sub_round6, sa33_sub_round6;

wire    [7:0]    sa00_sr_round6, sa01_sr_round6, sa02_sr_round6, sa03_sr_round6;

wire    [7:0]    sa10_sr_round6, sa11_sr_round6, sa12_sr_round6, sa13_sr_round6;

wire    [7:0]    sa20_sr_round6, sa21_sr_round6, sa22_sr_round6, sa23_sr_round6;

wire    [7:0]    sa30_sr_round6, sa31_sr_round6, sa32_sr_round6, sa33_sr_round6;

wire    [7:0]    sa00_mc_round6, sa01_mc_round6, sa02_mc_round6, sa03_mc_round6;

wire    [7:0]    sa10_mc_round6, sa11_mc_round6, sa12_mc_round6, sa13_mc_round6;

wire    [7:0]    sa20_mc_round6, sa21_mc_round6, sa22_mc_round6, sa23_mc_round6;

wire    [7:0]    sa30_mc_round6, sa31_mc_round6, sa32_mc_round6, sa33_mc_round6;

//round7 wires

wire    [7:0]    sa00_next_round7, sa01_next_round7, sa02_next_round7, sa03_next_round7;

wire    [7:0]    sa10_next_round7, sa11_next_round7, sa12_next_round7, sa13_next_round7;

wire    [7:0]    sa20_next_round7, sa21_next_round7, sa22_next_round7, sa23_next_round7;

wire    [7:0]    sa30_next_round7, sa31_next_round7, sa32_next_round7, sa33_next_round7;

wire    [7:0]    sa00_sub_round7, sa01_sub_round7, sa02_sub_round7, sa03_sub_round7;

wire    [7:0]    sa10_sub_round7, sa11_sub_round7, sa12_sub_round7, sa13_sub_round7;

wire  [7:0]      sa20_sub_round7, sa21_sub_round7, sa22_sub_round7, sa23_sub_round7;

wire    [7:0]    sa30_sub_round7, sa31_sub_round7, sa32_sub_round7, sa33_sub_round7;

wire    [7:0]    sa00_sr_round7, sa01_sr_round7, sa02_sr_round7, sa03_sr_round7;

wire    [7:0]    sa10_sr_round7, sa11_sr_round7, sa12_sr_round7, sa13_sr_round7;

wire    [7:0]    sa20_sr_round7, sa21_sr_round7, sa22_sr_round7, sa23_sr_round7;

wire    [7:0]    sa30_sr_round7, sa31_sr_round7, sa32_sr_round7, sa33_sr_round7;

wire    [7:0]    sa00_mc_round7, sa01_mc_round7, sa02_mc_round7, sa03_mc_round7;

wire    [7:0]    sa10_mc_round7, sa11_mc_round7, sa12_mc_round7, sa13_mc_round7;

wire    [7:0]    sa20_mc_round7, sa21_mc_round7, sa22_mc_round7, sa23_mc_round7;

wire    [7:0]    sa30_mc_round7, sa31_mc_round7, sa32_mc_round7, sa33_mc_round7;

//round8 wires

wire    [7:0]    sa00_next_round8, sa01_next_round8, sa02_next_round8, sa03_next_round8;

wire    [7:0]    sa10_next_round8, sa11_next_round8, sa12_next_round8, sa13_next_round8;

wire    [7:0]    sa20_next_round8, sa21_next_round8, sa22_next_round8, sa23_next_round8;

wire    [7:0]    sa30_next_round8, sa31_next_round8, sa32_next_round8, sa33_next_round8;

wire    [7:0]    sa00_sub_round8, sa01_sub_round8, sa02_sub_round8, sa03_sub_round8;

wire    [7:0]    sa10_sub_round8, sa11_sub_round8, sa12_sub_round8, sa13_sub_round8;

wire  [7:0]      sa20_sub_round8, sa21_sub_round8, sa22_sub_round8, sa23_sub_round8;

wire    [7:0]    sa30_sub_round8, sa31_sub_round8, sa32_sub_round8, sa33_sub_round8;

wire    [7:0]    sa00_sr_round8, sa01_sr_round8, sa02_sr_round8, sa03_sr_round8;

wire    [7:0]    sa10_sr_round8, sa11_sr_round8, sa12_sr_round8, sa13_sr_round8;

wire    [7:0]    sa20_sr_round8, sa21_sr_round8, sa22_sr_round8, sa23_sr_round8;

wire    [7:0]    sa30_sr_round8, sa31_sr_round8, sa32_sr_round8, sa33_sr_round8;

wire    [7:0]    sa00_mc_round8, sa01_mc_round8, sa02_mc_round8, sa03_mc_round8;

wire    [7:0]    sa10_mc_round8, sa11_mc_round8, sa12_mc_round8, sa13_mc_round8;

wire    [7:0]    sa20_mc_round8, sa21_mc_round8, sa22_mc_round8, sa23_mc_round8;

wire    [7:0]    sa30_mc_round8, sa31_mc_round8, sa32_mc_round8, sa33_mc_round8;

//round9 wires

wire    [7:0]    sa00_next_round9, sa01_next_round9, sa02_next_round9, sa03_next_round9;

wire    [7:0]    sa10_next_round9, sa11_next_round9, sa12_next_round9, sa13_next_round9;

wire    [7:0]    sa20_next_round9, sa21_next_round9, sa22_next_round9, sa23_next_round9;

wire    [7:0]    sa30_next_round9, sa31_next_round9, sa32_next_round9, sa33_next_round9;

wire    [7:0]    sa00_sub_round9, sa01_sub_round9, sa02_sub_round9, sa03_sub_round9;

wire    [7:0]    sa10_sub_round9, sa11_sub_round9, sa12_sub_round9, sa13_sub_round9;

wire  [7:0]      sa20_sub_round9, sa21_sub_round9, sa22_sub_round9, sa23_sub_round9;

wire    [7:0]    sa30_sub_round9, sa31_sub_round9, sa32_sub_round9, sa33_sub_round9;

wire    [7:0]    sa00_sr_round9, sa01_sr_round9, sa02_sr_round9, sa03_sr_round9;

wire    [7:0]    sa10_sr_round9, sa11_sr_round9, sa12_sr_round9, sa13_sr_round9;

wire    [7:0]    sa20_sr_round9, sa21_sr_round9, sa22_sr_round9, sa23_sr_round9;

wire    [7:0]    sa30_sr_round9, sa31_sr_round9, sa32_sr_round9, sa33_sr_round9;

wire    [7:0]    sa00_mc_round9, sa01_mc_round9, sa02_mc_round9, sa03_mc_round9;

wire    [7:0]    sa10_mc_round9, sa11_mc_round9, sa12_mc_round9, sa13_mc_round9;

wire    [7:0]    sa20_mc_round9, sa21_mc_round9, sa22_mc_round9, sa23_mc_round9;

wire    [7:0]    sa30_mc_round9, sa31_mc_round9, sa32_mc_round9, sa33_mc_round9;

//round10 wires

wire    [7:0]    sa00_next_round10, sa01_next_round10, sa02_next_round10, sa03_next_round10;

wire    [7:0]    sa10_next_round10, sa11_next_round10, sa12_next_round10, sa13_next_round10;

wire    [7:0]    sa20_next_round10, sa21_next_round10, sa22_next_round10, sa23_next_round10;

wire    [7:0]    sa30_next_round10, sa31_next_round10, sa32_next_round10, sa33_next_round10;

wire    [7:0]    sa00_sub_round10, sa01_sub_round10, sa02_sub_round10, sa03_sub_round10;

wire    [7:0]    sa10_sub_round10, sa11_sub_round10, sa12_sub_round10, sa13_sub_round10;

wire  [7:0]      sa20_sub_round10, sa21_sub_round10, sa22_sub_round10, sa23_sub_round10;

wire    [7:0]    sa30_sub_round10, sa31_sub_round10, sa32_sub_round10, sa33_sub_round10;

wire    [7:0]    sa00_sr_round10, sa01_sr_round10, sa02_sr_round10, sa03_sr_round10;

wire    [7:0]    sa10_sr_round10, sa11_sr_round10, sa12_sr_round10, sa13_sr_round10;

wire    [7:0]    sa20_sr_round10, sa21_sr_round10, sa22_sr_round10, sa23_sr_round10;

wire    [7:0]    sa30_sr_round10, sa31_sr_round10, sa32_sr_round10, sa33_sr_round10;

reg             done, ld_r;

reg     [3:0]    dcnt;

reg             done2;

////////////////////////////////////////////////////////////////////

//

// Misc Logic

//

always @(posedge clk)

begin

        if(rst) begin dcnt <=  4'h0;     end

        else

        if(ld)  begin   dcnt <=  4'hb;   end

        else

        if(|dcnt) begin dcnt <=  dcnt - 4'h1;  end

end

always @(posedge clk) done <=  !(|dcnt[3:1]) & dcnt[0] & !ld;

always @(posedge clk) if(ld) text_in_r <=  text_in;

always @(posedge clk) ld_r <=  ld;

////////////////////////////////////////////////////////////////////

// key expansion

aes_key_expand_128 u0(

        .clk(           clk     ),

        .key(           key     ),

        .w0(            w0      ),

        .w1(            w1      ),

        .w2(            w2      ),

        .w3(            w3      ),

        .w4(            w4      ),

        .w5(            w5      ),

        .w6(            w6      ),

        .w7(            w7      ),

        .w8(            w8      ),

        .w9(            w9      ),

        .w10(           w10     ),

        .w11(           w11     ),

        .w12(           w12     ),

        .w13(           w13     ),

        .w14(           w14     ),

        .w15(           w15     ),

        .w16(           w16     ),

        .w17(           w17     ),

        .w18(           w18     ),

        .w19(           w19     ),

        .w20(           w20     ),

        .w21(           w21     ),

        .w22(           w22     ),

        .w23(           w23     ),

        .w24(           w24     ),

        .w25(           w25     ),

        .w26(           w26     ),

        .w27(           w27     ),

        .w28(           w28     ),

        .w29(           w29     ),

        .w30(           w30     ),

        .w31(           w31     ),

        .w32(           w32     ),

        .w33(           w33     ),

        .w34(           w34     ),

        .w35(           w35     ),

        .w36(           w36     ),

        .w37(           w37     ),

        .w38(           w38     ),

        .w39(           w39     ),

        .w40(           w40     ),

        .w41(           w41     ),

        .w42(           w42     ),

        .w43(           w43     )

                                        );

always @(posedge clk)

begin

        sa33 <=   text_in_r[007:000] ^ w3[07:00]; //sa33_mc_round2 ^ w3[07:00];

        sa23 <=   text_in_r[015:008] ^ w3[15:08]; //sa23_mc_round2 ^ w3[15:08];

        sa13 <=   text_in_r[023:016] ^ w3[23:16]; //sa13_mc_round2 ^ w3[23:16];

        sa03 <=   text_in_r[031:024] ^ w3[31:24]; //sa03_mc_round2 ^ w3[31:24];

        sa32 <=   text_in_r[039:032] ^ w2[07:00]; //sa32_mc_round2 ^ w2[07:00];

        sa22 <=   text_in_r[047:040] ^ w2[15:08]; //sa22_mc_round2 ^ w2[15:08];

        sa12 <=   text_in_r[055:048] ^ w2[23:16]; //sa12_mc_round2 ^ w2[23:16];

        sa02 <=   text_in_r[063:056] ^ w2[31:24]; //sa02_mc_round2 ^ w2[31:24];

        sa31 <=   text_in_r[071:064] ^ w1[07:00]; //sa31_mc_round2 ^ w1[07:00];

        sa21 <=   text_in_r[079:072] ^ w1[15:08]; //sa21_mc_round2 ^ w1[15:08];

        sa11 <=   text_in_r[087:080] ^ w1[23:16]; //sa11_mc_round2 ^ w1[23:16];

        sa01 <=   text_in_r[095:088] ^ w1[31:24]; //sa01_mc_round2 ^ w1[31:24];

        sa30 <=   text_in_r[103:096] ^ w0[07:00]; //sa30_mc_round2 ^ w0[07:00];

        sa20 <=   text_in_r[111:104] ^ w0[15:08]; //sa20_mc_round2 ^ w0[15:08];

        sa10 <=   text_in_r[119:112] ^ w0[23:16]; //sa10_mc_round2 ^ w0[23:16];

        sa00 <=   text_in_r[127:120] ^ w0[31:24]; //sa00_mc_round2 ^ w0[31:24];

end

//sbox lookup

aes_sbox us00(  .a(     sa00    ), .d(  sa00_sub        ));

aes_sbox us01(  .a(     sa01    ), .d(  sa01_sub        ));

aes_sbox us02(  .a(     sa02    ), .d(  sa02_sub        ));

aes_sbox us03(  .a(     sa03    ), .d(  sa03_sub        ));

aes_sbox us10(  .a(     sa10    ), .d(  sa10_sub        ));

aes_sbox us11(  .a(     sa11    ), .d(  sa11_sub        ));

aes_sbox us12(  .a(     sa12    ), .d(  sa12_sub        ));

aes_sbox us13(  .a(     sa13    ), .d(  sa13_sub        ));

aes_sbox us20(  .a(     sa20    ), .d(  sa20_sub        ));

aes_sbox us21(  .a(     sa21    ), .d(  sa21_sub        ));

aes_sbox us22(  .a(     sa22    ), .d(  sa22_sub        ));

aes_sbox us23(  .a(     sa23    ), .d(  sa23_sub        ));

aes_sbox us30(  .a(     sa30    ), .d(  sa30_sub        ));

aes_sbox us31(  .a(     sa31    ), .d(  sa31_sub        ));

aes_sbox us32(  .a(     sa32    ), .d(  sa32_sub        ));

aes_sbox us33(  .a(     sa33    ), .d(  sa33_sub        ));

//shift rows

assign sa00_sr = sa00_sub;              //

assign sa01_sr = sa01_sub;              //no shift

assign sa02_sr = sa02_sub;              //

assign sa03_sr = sa03_sub;              //

assign sa10_sr = sa11_sub;              //

assign sa11_sr = sa12_sub;              // left shift by 1

assign sa12_sr = sa13_sub;              //

assign sa13_sr = sa10_sub;              //

assign sa20_sr = sa22_sub;              //

assign sa21_sr = sa23_sub;              //      left shift by 2

assign sa22_sr = sa20_sub;              //

assign sa23_sr = sa21_sub;              //

assign sa30_sr = sa33_sub;              //

assign sa31_sr = sa30_sub;              // left shift by 3

assign sa32_sr = sa31_sub;              //

assign sa33_sr = sa32_sub;              //

// mix column operation

assign {sa00_mc, sa10_mc, sa20_mc, sa30_mc}  = mix_col(sa00_sr,sa10_sr,sa20_sr,sa30_sr);

assign {sa01_mc, sa11_mc, sa21_mc, sa31_mc}  = mix_col(sa01_sr,sa11_sr,sa21_sr,sa31_sr);

assign {sa02_mc, sa12_mc, sa22_mc, sa32_mc}  = mix_col(sa02_sr,sa12_sr,sa22_sr,sa32_sr);

assign {sa03_mc, sa13_mc, sa23_mc, sa33_mc}  = mix_col(sa03_sr,sa13_sr,sa23_sr,sa33_sr);

//// add round key

assign sa00_next_round2 = sa00_mc ^ w4[31:24];

assign sa01_next_round2 = sa01_mc ^ w5[31:24];

assign sa02_next_round2 = sa02_mc ^ w6[31:24];

assign sa03_next_round2 = sa03_mc ^ w7[31:24];

assign sa10_next_round2 = sa10_mc ^ w4[23:16];

assign sa11_next_round2 = sa11_mc ^ w5[23:16];

assign sa12_next_round2 = sa12_mc ^ w6[23:16];

assign sa13_next_round2 = sa13_mc ^ w7[23:16];

assign sa20_next_round2 = sa20_mc ^ w4[15:08];

assign sa21_next_round2 = sa21_mc ^ w5[15:08];

assign sa22_next_round2 = sa22_mc ^ w6[15:08];

assign sa23_next_round2 = sa23_mc ^ w7[15:08];

assign sa30_next_round2 = sa30_mc ^ w4[07:00];

assign sa31_next_round2 = sa31_mc ^ w5[07:00];

assign sa32_next_round2 = sa32_mc ^ w6[07:00];

assign sa33_next_round2 = sa33_mc ^ w7[07:00];

always @(posedge clk)

        begin

                  text_out_stage1[127:120] <=  sa00_next_round2;

                  text_out_stage1[095:088] <=  sa01_next_round2;

                  text_out_stage1[063:056] <=  sa02_next_round2;

                  text_out_stage1[031:024] <=  sa03_next_round2;

                  text_out_stage1[119:112] <=  sa10_next_round2;

                  text_out_stage1[087:080] <=  sa11_next_round2;

                  text_out_stage1[055:048] <=  sa12_next_round2;

                  text_out_stage1[023:016] <=  sa13_next_round2;

                  text_out_stage1[111:104] <=  sa20_next_round2;

                  text_out_stage1[079:072] <=  sa21_next_round2;

                  text_out_stage1[047:040] <=  sa22_next_round2;

                  text_out_stage1[015:008] <=  sa23_next_round2;

                  text_out_stage1[103:096] <=  sa30_next_round2;

                  text_out_stage1[071:064] <=  sa31_next_round2;

                  text_out_stage1[039:032] <=  sa32_next_round2;

                  text_out_stage1[007:000] <=  sa33_next_round2;

        end

//////////////////////  round 2 //////////////////////////////////

//sbox lookup

aes_sbox us00_round2(   .a(text_out_stage1[127:120]             ), .d(  sa00_sub_round2 ));

aes_sbox us01_round2(   .a(text_out_stage1[095:088]             ), .d(  sa01_sub_round2 ));

aes_sbox us02_round2(   .a(text_out_stage1[063:056]             ), .d(  sa02_sub_round2 ));

aes_sbox us03_round2(   .a(text_out_stage1[031:024]             ), .d(  sa03_sub_round2 ));

aes_sbox us10_round2(   .a(text_out_stage1[119:112]             ), .d(  sa10_sub_round2 ));

aes_sbox us11_round2(   .a(text_out_stage1[087:080]             ), .d(  sa11_sub_round2 ));

aes_sbox us12_round2(   .a(text_out_stage1[055:048]             ), .d(  sa12_sub_round2 ));

aes_sbox us13_round2(   .a(text_out_stage1[023:016]             ), .d(  sa13_sub_round2 ));

aes_sbox us20_round2(   .a(text_out_stage1[111:104]             ), .d(  sa20_sub_round2 ));

aes_sbox us21_round2(   .a(text_out_stage1[079:072]             ), .d(  sa21_sub_round2 ));

aes_sbox us22_round2(   .a(text_out_stage1[047:040]             ), .d(  sa22_sub_round2 ));

aes_sbox us23_round2(   .a(text_out_stage1[015:008]             ), .d(  sa23_sub_round2 ));

aes_sbox us30_round2(   .a(text_out_stage1[103:096]             ), .d(  sa30_sub_round2 ));

aes_sbox us31_round2(   .a(text_out_stage1[071:064]             ), .d(  sa31_sub_round2 ));

aes_sbox us32_round2(   .a(text_out_stage1[039:032]             ), .d(  sa32_sub_round2 ));

aes_sbox us33_round2(   .a(text_out_stage1[007:000]             ), .d(  sa33_sub_round2 ));

//shift rows

assign sa00_sr_round2 = sa00_sub_round2;                //

assign sa01_sr_round2 = sa01_sub_round2;                //no shift

assign sa02_sr_round2 = sa02_sub_round2;                //

assign sa03_sr_round2 = sa03_sub_round2;                //

assign sa10_sr_round2 = sa11_sub_round2;                //