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

// If party 1:      --       port opt0 opt1 n len
// If party 2: remote_server port sel n len
int main(int argc, char* argv[])
{
  ProtocolDesc pd;
  dhRandomInit();
  //protocolUseStdio(&pd);
  const char* remote_host = (strcmp(argv[1],"--")==0?NULL:argv[1]);
  int me = (remote_host?2:1), err;
  setCurrentParty(&pd,me);
  if(me==1) err = protocolAcceptTcp2P(&pd,argv[2]);
  else err = protocolConnectTcp2P(&pd,remote_host,argv[2]);
  if(err!=0) { fprintf(stderr,"TCP connection error\n"); return 2; }
  if(pd.thisParty==1)
  {
    char *opt0 = argv[3];
    char *opt1 = argv[4];
    int n = atoi(argv[5]);
    int len = atoi(argv[6]);
    struct HonestOTExtSender *sender = honestOTExtSenderNew(&pd, 2);
    honestOTExtSend1Of2(sender, opt0, opt1, n, len);
    honestOTExtSenderRelease(sender);
  }
  else
  {
    int i;
    int n = atoi(argv[4]);
    bool *sel = malloc(n*sizeof(bool));
    for(i=0;i<n;i++) sel[i]=argv[3][i]-'0';
    int len = atoi(argv[5]);
    char *output = malloc(n*len);
    struct HonestOTExtRecver *recver = honestOTExtRecverNew(&pd, 1);
    honestOTExtRecv1Of2(recver, output, sel, n, len);
    honestOTExtRecverRelease(recver);
    fprintf(stderr, "%s\n", output);
    free(sel);
    free(output);
  }
  cleanupProtocol(&pd);
  return 0;
}