PeerStreamer

/*

  Copyright 2003 John Knottenbelt

  This program is free software; you can redistribute it and/or modify

  it under the terms of the GNU General Public License as published by

  the Free Software Foundation; either version 2 of the License, or

  (at your option) any later version.

  This program is distributed in the hope that it will be useful,

  but WITHOUT ANY WARRANTY; without even the implied warranty of

  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License

  along with this program; if not, write to the Free Software

  Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.

*/

#include "config.h"

#include "unixserversocket.h"

#include <iostream>

#include <cstring>

#include <cstdlib>

#include <cerrno>

#include <unistd.h>

#include <sys/types.h>

#include <sys/socket.h>

#include <sys/un.h>

#include <fcntl.h>

UnixServerSocket::UnixServerSocket(const std::string &filename)

  : filename(filename)

{

  fd = socket(PF_UNIX, SOCK_STREAM, 0);

  if (!fd) {

    std::cerr << "Fatal error: failed to create socket for listening:\n"

              << strerror(errno) << std::endl;

    exit(1);

  }

  struct sockaddr_un addr;

  addr.sun_family = AF_UNIX;

  strcpy(addr.sun_path, filename.c_str());

  while (bind(fd, (struct sockaddr *) &addr, sizeof(addr)) < 0) {

    if (errno == EADDRINUSE) {

      // Possibly old socket

      if (connect(fd, (struct sockaddr *) &addr, sizeof(addr)) >= 0) {

        close(fd);

        std::cerr << "Fatal error: dvbd already running on socket " << filename << std::endl;

        exit(1);

      }

      else 

        unlink(filename.c_str());

    }

    else {

      // Some other error

      std::cerr << "Fatal error: failed to bind socket to " << filename << "\n"

              << strerror(errno) << std::endl;

      exit(1);

    }

  }

  if (listen(fd, 5) < 0) {

    std::cerr << "Fatal error: failed to listen on socket " << filename << "\n"

              << strerror(errno) << std::endl;

    exit(1);

  }

}

void UnixServerSocket::setNonBlocking()

{

  long flags = fcntl(fd, F_GETFL, 0); 

  flags |= O_NONBLOCK;

  fcntl(fd, F_SETFL, flags);

}

UnixServerSocket::~UnixServerSocket()

{

  close(fd);

  unlink(filename.c_str());

}

void UnixServerSocket::addSelectFDs(Select &s) const

{

  s.addReadFD(fd);

}

bool UnixServerSocket::isReady(const Select &s) const

{

  return s.readyToRead(fd);

}

int UnixServerSocket::accept()

{

  return ::accept(fd, NULL, 0);

}