server.c

#include <sys/types.h>
#include <sys/time.h>
#include "armpmu_lib.h"
#include <sys/socket.h>
#include <netinet/in.h>
#include <openssl/aes.h>
#include <stdlib.h>
#include <stdio.h>
#include <time.h>
#include <ctype.h>
#include <unistd.h>
#include <stdarg.h>
#include <stdint.h>




unsigned char key[16];
AES_KEY expanded;
unsigned char zero[16];
unsigned char scrambledzero[16];
struct sockaddr_in server;
struct sockaddr_in client; socklen_t clientlen;
int s;
char in[1537];
int r;
char out[40];
FILE* fd; 
struct timeval tv_debut;
struct timeval tv_fin;
long long int duree;



void handle(char out[40],char in[],int len, FILE* f)
{
    int workareaLen = len * 3;
    unsigned char workarea[workareaLen];
    int i;
    for (i = 0;i < 40;++i) out[i] = 0;
    if (len < 16) return;
    for (i = 0;i < 16;++i) out[i] = in[i];
    for (i = 16;i < len;++i) workarea[i] = in[i];

    unsigned int ch1, cm1, o1, dr, dw;
    /*printf("(");*/
    /*for(i = 0; i < len; ++i) printf("%x", in[i]);*/
    /*printf(") ==> [AES_encrypt] ==> ");*/
    cyclecounter_flushall(fd);
    /**(unsigned int *) (out + 32) = get_cycle_count();*/
    enable_counters();
    gettimeofday(&tv_debut, NULL);
    *(unsigned int*) (out + 32) = 1000000 * tv_debut.tv_sec + tv_debut.tv_usec;

    AES_encrypt(in,workarea,&expanded);

    gettimeofday(&tv_fin, NULL);
    *(unsigned int *) (out + 36) = 1000000 * tv_fin.tv_sec + tv_fin.tv_usec;
    disable_counters();

    /**(unsigned int *) (out + 36) = get_cycle_count();*/
    ch1 = read_pmn(0), cm1 = read_pmn(1), o1 = read_flags();
    dr = read_pmn(2), dw = read_pmn(3);
    fprintf(f, "%u %u %u %u %u\n", ch1, cm1, o1, dr, dw);
    /*fflush(f);*/
    /*printf("[");*/
    /*for(i = 0; i < 16; ++i) printf("%x", workarea[i]);*/
    /*printf("] [");*/
    /*for(i = 16; i < len; ++i) printf("%x", workarea[i]);*/
    /*printf("] [");*/
    /*for(i = len; i < len * 2; ++i) printf("%x", workarea[i]);*/
    /*printf("]\n");*/

    /* a real server would now check AES-based authenticator, */
    /* process legitimate packets, and generate useful output */
    for (i = 0;i < 16;++i) out[16 + i] = scrambledzero[i];
    /*printf("RESPONSE ==> [");*/
    /*for(i = 0; i < 16; ++i) printf("%x", out[i]);*/
    /*printf("] [");*/
    /*for(i = 16; i < 32; ++i) printf("%x", out[i]);*/
    /*printf("] [");*/
    /*for(i = 32; i < 36; ++i) printf("%x", out[i]);*/
    /*printf("] [");*/
    /*for(i = 36; i < 40; ++i) printf("%x", out[i]);*/
    /*printf("]\n");*/

}

int main(int argc, char **argv)
{
    int i;
    if (read(0,key,sizeof key) < sizeof key) return 111;
    else {
        printf("\nKEY = ");
        for (i = 0; i < 16; i++) {
            printf("%x ", key[i]);
        }
        printf("\n");
    }
    AES_set_encrypt_key(key,128,&expanded);
    AES_encrypt(zero,scrambledzero,&expanded);
    printf("\nSCRAMBLED ZERO = ");
    for (i = 0; i < 16; i++) {
        printf("%x ", scrambledzero[i]);
    }
    printf("\n");

    if (!argv[1]) return 100;
    if (!inet_aton(argv[1],&server.sin_addr)) return 100;

    server.sin_family = AF_INET;
    server.sin_port = htons(10000);
    s = socket(AF_INET,SOCK_DGRAM,0);

    if (s == -1) return 111;
    if (bind(s,(struct sockaddr *) &server,sizeof server) == -1)
        return 111;

    fd = cyclecounter_open();
    FILE *f = fopen("/data/local/tmp/ch-cm-ovlf.txt", "w+");
    for (;;) {
        clientlen = sizeof client;
        r = recvfrom(s,in,sizeof in,0
        ,(struct sockaddr *) &client,&clientlen);
        if (r < 16) continue;
        if (r >= sizeof in) continue;
        handle(out,in,r,f);
        sendto(s,out,40,0,(struct sockaddr *) &client,clientlen);
    }
    cyclecounter_close(fd);
}