speck_cpa_cw/cw_firmware/speck3264.c

#include <stdio.h>
#include <stdint.h>
#include "speck.h"


void Words16ToBytes(u16 words[],u8 bytes[],int numwords)
{
    int i,j=0;
    for(i=0;i<numwords;i++){
        bytes[j]=(u8)words[i];
        bytes[j+1]=(u8)(words[i]>>8);
        j+=2;
    }
}

void BytesToWords16(u8 bytes[],u16 words[],int numbytes)
{
    int i,j=0;
    for(i=0;i<numbytes/2;i++){
        words[i]=(u16)bytes[j] | ((u16)bytes[j+1]<<8);
        j+=2;
    }
}

void Speck3264KeySchedule(u16 K[],u16 rk[])
{
    u16 i,D=K[3],C=K[2],B=K[1],A=K[0];

    for(i=0;i<22;){
        rk[i]=A; ER16(B,A,i++);
        rk[i]=A; ER16(C,A,i++);
        rk[i]=A; ER16(D,A,i++);
    }
}


void Speck3264Encrypt(u16 Pt[],u16 Ct[],u16 rk[])
{
    u16 i;
    Ct[0]=Pt[0]; Ct[1]=Pt[1];

    for(i=0;i<22;) ER16(Ct[1],Ct[0],rk[i++]);
}


void Speck3264Decrypt(u16 Pt[],u16 Ct[],u16 rk[])
{
    int i;
    Pt[0]=Ct[0]; Pt[1]=Ct[1];

    for(i=21;i>=0;) DR16(Pt[1],Pt[0],rk[i--]);
}



void EncryptBlock(u8 pt[], u8 k[], u8 ct[]) {

    u16 Pt[2] = {0};
    u16 K[4] = {0};
    u16 rk[34] = {0};
    u16 Ct[2] = {0};

    BytesToWords16(pt,Pt,8);
    BytesToWords16(k,K,16);

    Speck3264KeySchedule(K,rk);

    for (int i=0; i < 16; i++)
    {
        printf("Key: 0x%x\n", rk[i]);
    }
    Speck3264Encrypt(Pt,Ct,rk);
    Words16ToBytes(Ct,ct,2);
}


int main() {

    u8 key[8] = {0x1, 0x2,0x3,0x4,0x5,0x6,0x7,0x8};
    u8 pt[4] = {0xde, 0xad, 0xbe, 0xef};
    u8 ct[4] = {0x0};

    EncryptBlock(pt, key, ct);
    printf("[[ Speck 32/64  ]]\n");

    printf("The output: \n");
    for (int i = 0; i < 4; i++) {
        printf("- %08x\n", ct[i]);
    }
    return 0;
}