#include<stdio.h>
#include<stdlib.h>
#include<bcrandom.h>
#include<obliv.h>
#include<obliv_common.h>

void unpackBitString(bool dest[],char src[],size_t boolCount)
{
  unsigned i,j;
  for(i=0;8*i<boolCount;i++) 
  { char ch = src[i];
    for(j=0;j<8 && i*8+j<boolCount;++j,ch>>=1) dest[i*8+j] = (ch&1);
  }
}

int main(int argc, char* argv[])
{
  int n;
  const int len = 10;
  if(argc<5 || sscanf(argv[4],"%d",&n)!=1) 
  { fprintf(stderr,"Usage %s <port> <--|server> <H|M|P|Q> <n>\n",argv[0]);
    return 1;
  }
	ProtocolDesc pd;
  dhRandomInit();
  const char* remote_host = (strcmp(argv[2],"--")==0?NULL:argv[2]);
  int me = remote_host?2:1;
  int err;
	//protocolUseStdio(&pd);
  if(me==1) err = protocolAcceptTcp2P(&pd,argv[1]);
  else err = protocolConnectTcp2P(&pd,remote_host,argv[1]);
  if(err!=0) { fprintf(stderr,"TCP connection error\n"); return 2; }
	setCurrentParty(&pd,me);
  pd.error=0;
  BCipherRandomGen *gen = newBCipherRandomGen();
	if(me==1)
	{
		char *opt0 = malloc(n*len), *opt1 = malloc(n*len);
    randomizeBuffer(gen,opt0,n*len);
    randomizeBuffer(gen,opt1,n*len);
    OTsender s;
    if(argv[3][0]=='H') 
      s = honestOTExtSenderAbstract(honestOTExtSenderNew(&pd,2));
    else if(argv[3][0]=='P')
      s = maliciousOTExtSenderAbstract(otExtSenderNew_byPair(&pd,2));
    else if(argv[3][0]=='Q')
      s = maliciousOTExtSenderAbstract(otExtSenderNew_byPhair(&pd,2));
    else
      s = maliciousOTExtSenderAbstract(otExtSenderNew(&pd,2));
    s.send(s.sender,opt0,opt1,n,len);
    otSenderRelease(&s);
    free(opt0); free(opt1);
	}
	else 
	{
		int i;
    char *selpack = malloc((n+7)/8);
    bool *sel = malloc(sizeof(bool)*n);
    randomizeBuffer(gen,selpack,(n+7)/8);
    unpackBitString(sel,selpack,n);
    char *output = malloc(n*len);
    OTrecver r;
    if(argv[3][0]=='H')
      r = honestOTExtRecverAbstract(honestOTExtRecverNew(&pd,1));
    else if(argv[3][0]=='P')
      r = maliciousOTExtRecverAbstract(otExtRecverNew_byPair(&pd,1));
    else if(argv[3][0]=='Q')
      r = maliciousOTExtRecverAbstract(otExtRecverNew_byPhair(&pd,1));
    else
      r = maliciousOTExtRecverAbstract(otExtRecverNew(&pd,1));
    r.recv(r.recver,output,sel,n,len);
    otRecverRelease(&r);
    free(output);
    free(selpack);
    free(sel);
	}
  releaseBCipherRandomGen(gen);
	cleanupProtocol(&pd);
	return 0;
}