X-Git-Url: https://git.m6w6.name/?a=blobdiff_plain;f=libtest%2Fport.cc;h=d8e8c12d566beca74fc96836a9f0635b67342b5a;hb=61b948a92ab79734b009a438b5cf3bfad03cb59d;hp=fefb6a6d394ada0c72b0c480b61b7b38c140a406;hpb=015b6142ad6c3b496437d397416aacb776a42aed;p=awesomized%2Flibmemcached diff --git a/libtest/port.cc b/libtest/port.cc index fefb6a6d..d8e8c12d 100644 --- a/libtest/port.cc +++ b/libtest/port.cc @@ -1,25 +1,40 @@ /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab: - * - * libtest * - * Copyright (C) 2011 Data Differential, http://datadifferential.com/ + * Data Differential YATL (i.e. libtest) library * - * This library is free software; you can redistribute it and/or - * modify it under the terms of the GNU Lesser General Public - * License as published by the Free Software Foundation; either - * version 3 of the License, or (at your option) any later version. + * Copyright (C) 2012 Data Differential, http://datadifferential.com/ * - * This library 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 - * Lesser General Public License for more details. + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are + * met: + * + * * Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * + * * Redistributions in binary form must reproduce the above + * copyright notice, this list of conditions and the following disclaimer + * in the documentation and/or other materials provided with the + * distribution. + * + * * The names of its contributors may not be used to endorse or + * promote products derived from this software without specific prior + * written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS + * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT + * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR + * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT + * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, + * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT + * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, + * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY + * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT + * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE + * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. * - * You should have received a copy of the GNU Lesser General Public - * License along with this library; if not, write to the Free Software - * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA */ -#include +#include "libtest/yatlcon.h" #include #include @@ -35,11 +50,25 @@ #include #include +#include +#include + #include -#include #include +#ifndef SOCK_CLOEXEC +# define SOCK_CLOEXEC 0 +#endif + +#ifndef SOCK_NONBLOCK +# define SOCK_NONBLOCK 0 +#endif + +#ifndef FD_CLOEXEC +# define FD_CLOEXEC 0 +#endif + #ifndef __INTEL_COMPILER #pragma GCC diagnostic ignored "-Wold-style-cast" #endif @@ -47,13 +76,36 @@ using namespace libtest; struct socket_st { - std::vector fd; + typedef std::vector< std::pair< int, in_port_t> > socket_port_t; + socket_port_t _pair; + in_port_t last_port; + + socket_st(): + last_port(0) + { } + + void release(in_port_t _arg) + { + for (socket_port_t::iterator iter= _pair.begin(); + iter != _pair.end(); + ++iter) + { + if ((*iter).second == _arg) + { + shutdown((*iter).first, SHUT_RDWR); + close((*iter).first); + } + } + } ~socket_st() { - for(std::vector::iterator iter= fd.begin(); iter != fd.end(); iter++) + for (socket_port_t::iterator iter= _pair.begin(); + iter != _pair.end(); + ++iter) { - close(*iter); + shutdown((*iter).first, SHUT_RDWR); + close((*iter).first); } } }; @@ -74,52 +126,105 @@ in_port_t default_port() return global_port; } +void release_port(in_port_t arg) +{ + all_socket_fd.release(arg); +} + in_port_t get_free_port() { - in_port_t ret_port= in_port_t(0); + const in_port_t default_port= in_port_t(-1); int retries= 1024; - while (retries--) + in_port_t ret_port; + while (--retries) { + ret_port= default_port; int sd; - if ((sd= socket(AF_INET, SOCK_STREAM, 0)) != -1) + if ((sd= socket(AF_INET, SOCK_STREAM, 0)) != SOCKET_ERROR) { int optval= 1; - if (setsockopt(sd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval)) != -1) - { + if (setsockopt(sd, SOL_SOCKET, SO_REUSEADDR, &optval, sizeof(optval)) != SOCKET_ERROR) + { struct sockaddr_in sin; sin.sin_port= 0; sin.sin_addr.s_addr= 0; sin.sin_addr.s_addr= INADDR_ANY; sin.sin_family= AF_INET; - if (bind(sd, (struct sockaddr *)&sin,sizeof(struct sockaddr_in) ) != -1) + int bind_ret; + do { - socklen_t addrlen= sizeof(sin); + if ((bind_ret= bind(sd, (struct sockaddr *)&sin, sizeof(struct sockaddr_in) )) != SOCKET_ERROR) + { + socklen_t addrlen= sizeof(sin); - if (getsockname(sd, (struct sockaddr *)&sin, &addrlen) != -1) + if (getsockname(sd, (struct sockaddr *)&sin, &addrlen) != -1) + { + ret_port= sin.sin_port; + } + } + else { - ret_port= sin.sin_port; + if (errno != EADDRINUSE) + { + Error << strerror(errno); + } } - } - } - all_socket_fd.fd.push_back(sd); + if (errno == EADDRINUSE) + { + libtest::dream(2, 0); + } + } while (bind_ret == -1 and errno == EADDRINUSE); + + all_socket_fd._pair.push_back(std::make_pair(sd, ret_port)); + } + else + { + Error << strerror(errno); + } + } + else + { + Error << strerror(errno); } - if (ret_port > 1024) + if (ret_port == default_port) + { + Error << "no ret_port set:" << strerror(errno); + } + else if (ret_port > 1024 and ret_port != all_socket_fd.last_port) { break; } } // We handle the case where if we max out retries, we still abort. + if (retries == 0) + { + FATAL("No port could be found, exhausted retry"); + } + + if (ret_port == 0) + { + FATAL("No port could be found"); + } + + if (ret_port == default_port) + { + FATAL("No port could be found"); + } + if (ret_port <= 1024) { - fatal_message("No port could be found"); + FATAL("No port could be found, though some where available below or at 1024"); } + all_socket_fd.last_port= ret_port; + release_port(ret_port); + return ret_port; }