#include "port.h"
#include <unistd.h>

#include "zcav_io.h"
#include "thread.h"

#include <cstdlib>
#include <cstring>

#define TOO_MANY_LOOPS 100

void usage()
{
  fprintf(stderr
       , "Usage: zcav [-b block-size[:chunk-size]] [-c count]\n"
         "            [-r [start offset:]end offset] [-w]\n"
         "            [-u uid-to-use:gid-to-use] [-g gid-to-use]\n"
#ifdef _LARGEFILE64_SOURCE
         "            [-s skip rate]\n"
#endif
         "            [-l log-file] [-f] file-name\n"
         "            [-l log-file [-f] file-name]...\n"
         "\n"
         "File name of \"-\" means standard input\n"
         "Count is the number of times to read the data (default 1).\n"
         "Max size is the amount of data to read from each device.\n"
         "\n"
         "Version: " BON_VERSION "\n");
  exit(1);
}

class MultiZcav : public Thread
{
public:
  MultiZcav();
  MultiZcav(int threadNum, const MultiZcav *parent);
  virtual ~MultiZcav();

  virtual int action(PVOID param);

  int runit();

  void setFileLogNames(const char *file, const char *log)
  {
    m_fileNames.push_back(file);
    m_logNames.push_back(log);
    m_readers->push_back(new ZcavRead);
  }

  void setSizes(int block_size, int chunk_size)
  {
    m_block_size = block_size;
    m_chunk_size = chunk_size;
    if(m_block_size < 1 || m_chunk_size < 1 || m_chunk_size > m_block_size)
      usage();
  }

  void setWrite(int do_write)
  {
    m_do_write = do_write;
  }

  void setLoops(int max_loops)
  {
    m_max_loops = max_loops;
    if(max_loops < 1 || max_loops > TOO_MANY_LOOPS)
      usage();
  }

  void setMaxSize(int max_size)
  {
    m_max_size = max_size;
    if(max_size < 1)
      usage();
  }

  void setStartOffset(int start_offset)
  {
    m_start_offset = start_offset;
    if(start_offset < 1)
      usage();
  }

  void setSkipRate(int skip_rate)
  {
    m_skip_rate = skip_rate;
    if(skip_rate < 2 || skip_rate > 20)
      usage();
  }

private:
  virtual Thread *newThread(int threadNum)
                  { return new MultiZcav(threadNum, this); }

  vector<const char *> m_fileNames, m_logNames;
  vector<ZcavRead *> *m_readers;

  int m_block_size, m_max_loops, m_max_size, m_start_offset, m_skip_rate;
  int m_chunk_size, m_do_write;

  MultiZcav(const MultiZcav &m);
  MultiZcav & operator =(const MultiZcav &m);
};

MultiZcav::MultiZcav()
{
  m_block_size = 256;
  m_max_loops = 1;
  m_max_size = 0;
  m_start_offset = 0;
  m_skip_rate = 1;
  m_chunk_size = DEFAULT_CHUNK_SIZE;
  m_do_write = 0;
  m_readers = new vector<ZcavRead *>;
}

MultiZcav::MultiZcav(int threadNum, const MultiZcav *parent)
 : Thread(threadNum, parent)
 , m_readers(parent->m_readers)
 , m_block_size(parent->m_block_size)
 , m_max_loops(parent->m_max_loops)
 , m_max_size(parent->m_max_size)
 , m_start_offset(parent->m_start_offset)
 , m_skip_rate(parent->m_skip_rate)
{
}

int MultiZcav::action(PVOID)
{
  ZcavRead *zc = (*m_readers)[getThreadNum() - 1];
  int rc = zc->Read(m_max_loops, m_max_size / m_block_size, m_write, m_skip_rate, m_start_offset / m_block_size);
  zc->Close();
  return rc;
}

MultiZcav::~MultiZcav()
{
  if(getThreadNum() < 1)
  {
    while(m_readers->size())
    {
      delete m_readers->back();
      m_readers->pop_back();
    }
    delete m_readers;
  }
}

int MultiZcav::runit()
{
  unsigned int i;
  unsigned int num_threads = m_fileNames.size();
  if(num_threads < 1)
    usage();
  for(i = 0; i < num_threads; i++)
  {
    if((*m_readers)[i]->Open(NULL, m_block_size, m_fileNames[i], m_logNames[i], m_chunk_size, m_do_write))
    {
      return 1;
    }
  }
  go(NULL, num_threads);
  int res = 0;
  while(num_threads)
  {
    char c = 0;
    if(Read(&c, 1, 0) != 1)
      printf("can't read!\n");
    num_threads--;
    if(c > res)
      res = c;
  }
  return res;
}

int main(int argc, char *argv[])
{
  MultiZcav mz;

  if(argc < 2)
    usage();

  char *userName = NULL, *groupName = NULL;
  int c;
  int do_write = 0;
  const char *log = "-";
  const char *file = "";
  while(-1 != (c = getopt(argc, argv, "-c:b:f:l:r:w"
#ifdef _LARGEFILE64_SOURCE
				     "s:"
#endif
                                     "u:g:")) )
  {
    switch(char(c))
    {
      case 'b':
      {
        int block_size, chunk_size;
        int rc = sscanf(optarg, "%d:%d", &block_size, &chunk_size);
        if(rc == 1)
          chunk_size = DEFAULT_CHUNK_SIZE;
        else if(rc != 2)
          usage();
        mz.setSizes(block_size, chunk_size);
      }
      break;
      case 'c':
        mz.setLoops(atoi(optarg));
      break;
      case 'l':
        log = optarg;
      break;
      case 'r':
      {
        int a, b, rc;
        rc = sscanf(optarg, "%d:%d", &a, &b);
        if(rc == 0)
          usage();
        if(rc == 1)
          mz.setMaxSize(a);
        else
        {
          mz.setStartOffset(a);
          mz.setMaxSize(b);
        }
      }
      break;
#ifdef _LARGEFILE64_SOURCE
      case 's':
        mz.setSkipRate(atoi(optarg));
      break;
#endif
      case 'g':
        if(groupName)
          usage();
        groupName = optarg;
      break;
      case 'u':
      {
        if(userName)
          usage();
        userName = strdup(optarg);
        int i;
        for(i = 0; userName[i] && userName[i] != ':'; i++) {}

        if(userName[i] == ':')
        {
          if(groupName)
            usage();
          userName[i] = '\0';
          groupName = &userName[i + 1];
        }
      }
      break;
      case 'w':
        mz.setWrite(1);
        do_write = 1;
      break;
      case 'f':
      case char(1):
        mz.setFileLogNames(optarg, log);
        file = optarg;
        log = "-";
      break;
      default:
        usage();
    }
  }

  if(userName || groupName)
  {
    if(bon_setugid(userName, groupName, false))
      return 1;
    if(userName)
      free(userName);
  }

  if(do_write)
  {
    fprintf(stderr, "Warning, writing to %s in 5 seconds.\n", file);
    sleep(5);
  }
  int rc = mz.runit();
  sleep(2); // time for all threads to complete
  return rc;
}