X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=src%2Fpipe.cpp;h=309f51b0496aa406572ffdda5c66c3668dcf417c;hb=56a046379458e5aa06881cc146b951d5b0711b85;hp=f0306f410929bcf8bb14fd8a681a23324140a82c;hpb=7ccf90cb13451e3d024eff6239815f92aff6e51b;p=metaproxy-moved-to-github.git diff --git a/src/pipe.cpp b/src/pipe.cpp index f0306f4..309f51b 100644 --- a/src/pipe.cpp +++ b/src/pipe.cpp @@ -1,15 +1,29 @@ +/* This file is part of Metaproxy. + Copyright (C) Index Data -/* $Id: pipe.cpp,v 1.1 2005-11-07 12:32:01 adam Exp $ - Copyright (c) 2005, Index Data. +Metaproxy 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, or (at your option) any later +version. + +Metaproxy 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., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA +*/ -%LICENSE% - */ #include "config.hpp" #if HAVE_UNISTD_H #include #endif +#include +#include #ifdef WIN32 #include #else @@ -17,6 +31,8 @@ #include #include #include + +#include #endif #if HAVE_SYS_SOCKET_H @@ -31,24 +47,41 @@ #include #include +#include #include -#include +#include #include #include "pipe.hpp" -namespace yp2 { +namespace mp = metaproxy_1; + +namespace metaproxy_1 { class Pipe::Rep : public boost::noncopyable { friend class Pipe; Rep(); int m_fd[2]; int m_socket; + bool nonblock(int s); + void close(int &fd); }; } -using namespace yp2; +using namespace mp; + +void Pipe::Rep::close(int &fd) +{ +#ifdef WIN32 + if (fd != -1) + ::closesocket(fd); +#else + if (fd != -1) + ::close(fd); +#endif + fd = -1; +} Pipe::Rep::Rep() { @@ -56,54 +89,125 @@ Pipe::Rep::Rep() m_socket = -1; } +bool Pipe::Rep::nonblock(int s) +{ +#ifdef WIN32 + unsigned long tru = 1; + if (ioctlsocket(s, FIONBIO, &tru) < 0) + return false; +#else + if (fcntl(s, F_SETFL, O_NONBLOCK) < 0) + return false; +#ifndef MSG_NOSIGNAL + signal (SIGPIPE, SIG_IGN); +#endif +#endif + return true; +} + Pipe::Pipe(int port_to_use) : m_p(new Rep) { +#ifdef WIN32 + WSADATA wsaData; + WORD wVersionRequested = MAKEWORD(2, 0); + if (WSAStartup( wVersionRequested, &wsaData )) + throw Pipe::Error("WSAStartup failed"); +#else + port_to_use = 0; // we'll just use pipe on Unix +#endif if (port_to_use) { + // create server socket m_p->m_socket = socket(AF_INET, SOCK_STREAM, 0); if (m_p->m_socket < 0) throw Pipe::Error("could not create socket"); - - m_p->m_fd[1] = socket(AF_INET, SOCK_STREAM, 0); - if (m_p->m_fd[1] < 0) - throw Pipe::Error("could not create socket"); - +#ifndef WIN32 + unsigned long one = 1; + if (setsockopt(m_p->m_socket, SOL_SOCKET, SO_REUSEADDR, (char*) + &one, sizeof(one)) < 0) + throw Pipe::Error("setsockopt error"); +#endif + // bind server socket struct sockaddr_in add; add.sin_family = AF_INET; add.sin_port = htons(port_to_use); add.sin_addr.s_addr = INADDR_ANY; struct sockaddr *addr = ( struct sockaddr *) &add; - + if (bind(m_p->m_socket, addr, sizeof(struct sockaddr_in))) throw Pipe::Error("could not bind on socket"); - + if (listen(m_p->m_socket, 3) < 0) throw Pipe::Error("could not listen on socket"); + // client socket + unsigned int tmpadd; + tmpadd = (unsigned) inet_addr("127.0.0.1"); + if (tmpadd) + memcpy(&add.sin_addr.s_addr, &tmpadd, sizeof(struct in_addr)); + else + throw Pipe::Error("inet_addr failed"); + + m_p->m_fd[1] = socket(AF_INET, SOCK_STREAM, 0); + if (m_p->m_fd[1] < 0) + throw Pipe::Error("could not create socket"); + + m_p->nonblock(m_p->m_fd[1]); + + if (connect(m_p->m_fd[1], addr, sizeof(*addr)) < 0) + { +#ifdef WIN32 + if (WSAGetLastError() != WSAEWOULDBLOCK) + throw Pipe::Error("could not connect to socket"); +#else + if (errno != EINPROGRESS) + throw Pipe::Error("could not connect to socket"); +#endif + } + + // server accept struct sockaddr caddr; +#ifdef WIN32 + int caddr_len = sizeof(caddr); +#else socklen_t caddr_len = sizeof(caddr); +#endif m_p->m_fd[0] = accept(m_p->m_socket, &caddr, &caddr_len); if (m_p->m_fd[0] < 0) throw Pipe::Error("could not accept on socket"); - - if (connect(m_p->m_fd[1], addr, sizeof(addr)) < 0) - throw Pipe::Error("could not connect to socket"); + + // complete connect + fd_set write_set; + FD_ZERO(&write_set); + FD_SET(m_p->m_fd[1], &write_set); + int r = select(m_p->m_fd[1]+1, 0, &write_set, 0, 0); + if (r != 1) + throw Pipe::Error("could not complete connect"); + + m_p->close(m_p->m_socket); } else { - m_p->m_socket = 0; - pipe(m_p->m_fd); +#ifndef WIN32 + if (pipe(m_p->m_fd)) + throw Pipe::Error("pipe failed"); + else + { + assert(m_p->m_fd[0] >= 0); + assert(m_p->m_fd[1] >= 0); + } +#endif } } Pipe::~Pipe() { - if (m_p->m_fd[0] != -1) - close(m_p->m_fd[0]); - if (m_p->m_fd[1] != -1) - close(m_p->m_fd[1]); - if (m_p->m_socket != -1) - close(m_p->m_socket); + m_p->close(m_p->m_fd[0]); + m_p->close(m_p->m_fd[1]); + m_p->close(m_p->m_socket); +#ifdef WIN32 + WSACleanup(); +#endif } int &Pipe::read_fd() const @@ -119,8 +223,8 @@ int &Pipe::write_fd() const /* * Local variables: * c-basic-offset: 4 + * c-file-style: "Stroustrup" * indent-tabs-mode: nil - * c-file-style: "stroustrup" * End: * vim: shiftwidth=4 tabstop=8 expandtab */