#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include "mceliece.h" /* -lmceliece */
#include "limits.inc"

static unsigned char pk[mceliece348864_PUBLICKEYBYTES];
static unsigned char c[mceliece348864_CIPHERTEXTBYTES];
static unsigned char k[mceliece348864_BYTES];

static void die_temp(const char *why,const char *why2)
{
  if (why2)
    fprintf(stderr,"mceliece348864-enc: fatal: %s: %s\n",why,why2);
  else
    fprintf(stderr,"mceliece348864-enc: fatal: %s\n",why);
  exit(111);
}

int main()
{
  FILE *pkfile;
  FILE *kfile;

  limits();

  pkfile = fdopen(4,"r");
  if (!pkfile) {
    fprintf(stderr,"mceliece348864-enc: usage: mceliece348864-enc >ciphertext 7>sessionkey 4<publickey\n");
    die_temp("fdopen 4 failed",strerror(errno));
  }

  kfile = fdopen(7,"w");
  if (!kfile) {
    fprintf(stderr,"mceliece348864-enc: usage: mceliece348864-enc >ciphertext 7>sessionkey 4<publickey\n");
    die_temp("fdopen 7 failed",strerror(errno));
  }

  if (fread(pk,1,sizeof pk,pkfile) < sizeof pk) {
    if (ferror(pkfile))
      die_temp("read publickey failed",strerror(errno));
    die_temp("read publickey failed","end of file");
  }
  fclose(pkfile);

  if (mceliece348864_enc(c,k,pk)) die_temp("encapsulation failed",0);

  if (fwrite(c,1,sizeof c,stdout) < sizeof c)
    die_temp("write ciphertext failed",strerror(errno));
  if (fflush(stdout))
    die_temp("write ciphertext failed",strerror(errno));
  fclose(stdout);

  if (fwrite(k,1,sizeof k,kfile) < sizeof k)
    die_temp("write sessionkey failed",strerror(errno));
  if (fflush(kfile))
    die_temp("write sessionkey failed",strerror(errno));
  fclose(kfile);

  return 0;
}
