socket.cc revision 2
1/* 2 * Copyright (c) 2003 The Regents of The University of Michigan 3 * All rights reserved. 4 * 5 * Redistribution and use in source and binary forms, with or without 6 * modification, are permitted provided that the following conditions are 7 * met: redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer; 9 * redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution; 12 * neither the name of the copyright holders nor the names of its 13 * contributors may be used to endorse or promote products derived from 14 * this software without specific prior written permission. 15 * 16 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 17 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 18 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 19 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 20 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 21 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 22 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 26 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 */ 28 29#include <sys/types.h> 30#include <sys/socket.h> 31 32#include <netinet/in.h> 33#include <netinet/tcp.h> 34 35#include <errno.h> 36#include <unistd.h> 37 38#include "host.hh" 39#include "misc.hh" 40#include "socket.hh" 41 42using namespace std; 43 44//////////////////////////////////////////////////////////////////////// 45// 46// 47 48ListenSocket::ListenSocket() 49 : listening(false), fd(-1) 50{} 51 52ListenSocket::~ListenSocket() 53{ 54 if (fd != -1) 55 close(fd); 56} 57 58// Create a socket and configure it for listening 59bool 60ListenSocket::listen(int port, bool reuse) 61{ 62 if (listening) 63 panic("Socket already listening!"); 64 65 fd = ::socket(PF_INET, SOCK_STREAM, 0); 66 if (fd < 0) 67 panic("Can't create socket!"); 68 69 if (reuse) { 70 int i = 1; 71 if (::setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, (char *)&i, 72 sizeof(i)) < 0) 73 panic("ListenSocket(listen): setsockopt() SO_REUSEADDR failed!"); 74 } 75 76 struct sockaddr_in sockaddr; 77 sockaddr.sin_family = PF_INET; 78 sockaddr.sin_addr.s_addr = INADDR_ANY; 79 80 sockaddr.sin_port = htons(port); 81 int ret = ::bind(fd, (struct sockaddr *)&sockaddr, sizeof (sockaddr)); 82 if (ret != 0) { 83 if (ret == -1 && errno != EADDRINUSE) 84 panic("ListenSocket(listen): bind() failed!"); 85 return false; 86 } 87 88 if (::listen(fd, 1) == -1) 89 panic("ListenSocket(listen): listen() failed!"); 90 91 listening = true; 92 93 return true; 94} 95 96#if !defined(__OpenBSD__) && !defined(linux) 97typedef int socklen_t; 98#endif 99 100// Open a connection. Accept will block, so if you don't want it to, 101// make sure a connection is ready before you call accept. 102int 103ListenSocket::accept(bool nodelay) 104{ 105 struct sockaddr_in sockaddr; 106 socklen_t slen = sizeof (sockaddr); 107 int sfd = ::accept(fd, (struct sockaddr *)&sockaddr, &slen); 108 if (sfd != -1 && nodelay) { 109 int i = 1; 110 ::setsockopt(sfd, IPPROTO_TCP, TCP_NODELAY, (char *)&i, sizeof(i)); 111 } 112 113 return sfd; 114} 115