#include "Acceptor.h"

#include "InetAddress.h"
#include "TcpStream.h"

#include <stdio.h>
#include <sys/socket.h>

Acceptor::Acceptor(const InetAddress& listenAddr)
  : listenSock_(Socket::createTCP(listenAddr.family()))
{
  listenSock_.setReuseAddr(true);
  listenSock_.bindOrDie(listenAddr);
  listenSock_.listenOrDie();
}

TcpStreamPtr Acceptor::accept()
{
  // FIXME: use accept4
  int sockfd = ::accept(listenSock_.fd(), NULL, NULL);
  if (sockfd >= 0)
  {
    return TcpStreamPtr(new TcpStream(Socket(sockfd)));
  }
  else
  {
    perror("Acceptor::accept");
    return TcpStreamPtr();
  }
}

Socket Acceptor::acceptSocketOrDie()
{
  // FIXME: use accept4
  int sockfd = ::accept(listenSock_.fd(), NULL, NULL);
  if (sockfd >= 0)
  {
    return Socket(sockfd);
  }
  else
  {
    perror("Acceptor::acceptSocketOrDie");
    abort();
  }
}