1d56e3eca280ebfbe39c6a76fe7a75770f1a29bd
[metaproxy-moved-to-github.git] / src / pipe.cpp
1 /* This file is part of Metaproxy.
2    Copyright (C) 2005-2008 Index Data
3
4 Metaproxy is free software; you can redistribute it and/or modify it under
5 the terms of the GNU General Public License as published by the Free
6 Software Foundation; either version 2, or (at your option) any later
7 version.
8
9 Metaproxy is distributed in the hope that it will be useful, but WITHOUT ANY
10 WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 for more details.
13
14 You should have received a copy of the GNU General Public License
15 along with this program; if not, write to the Free Software
16 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
17 */
18
19 #include "config.hpp"
20
21 #if HAVE_UNISTD_H
22 #include <unistd.h>
23 #endif
24
25 #include <signal.h>
26 #include <errno.h>
27 #ifdef WIN32
28 #include <winsock.h>
29 #else
30 #include <netinet/in.h>
31 #include <netdb.h>
32 #include <arpa/inet.h>
33 #include <netinet/tcp.h>
34
35 #include <fcntl.h>
36 #endif
37
38 #if HAVE_SYS_SOCKET_H
39 #include <sys/socket.h>
40 #endif
41 #if HAVE_SYS_SELECT_H
42 #include <sys/select.h>
43 #endif
44
45 #include <boost/thread/thread.hpp>
46 #include <boost/thread/mutex.hpp>
47 #include <boost/thread/condition.hpp>
48
49 #include <stdio.h>
50
51 #include <deque>
52
53 #include <yazpp/socket-observer.h>
54 #include <yaz/log.h>
55
56 #include "pipe.hpp"
57
58 namespace mp = metaproxy_1;
59
60 namespace metaproxy_1 {
61     class Pipe::Rep : public boost::noncopyable {
62         friend class Pipe;
63         Rep();
64         int m_fd[2];
65         int m_socket;
66         bool nonblock(int s);
67         void close(int &fd);
68     };
69 }
70
71 using namespace mp;
72
73 void Pipe::Rep::close(int &fd)
74 {
75 #ifdef WIN32
76     if (fd != -1)
77         ::closesocket(fd);
78 #else
79     if (fd != -1)
80         ::close(fd);
81 #endif
82     fd = -1;
83 }
84
85 Pipe::Rep::Rep()
86 {
87     m_fd[0] = m_fd[1] = -1;
88     m_socket = -1;
89 }
90
91 bool Pipe::Rep::nonblock(int s)
92 {
93 #ifdef WIN32
94     unsigned long tru = 1;
95     if (ioctlsocket(s, FIONBIO, &tru) < 0)
96         return false;
97 #else
98     if (fcntl(s, F_SETFL, O_NONBLOCK) < 0)
99         return false;
100 #ifndef MSG_NOSIGNAL
101     signal (SIGPIPE, SIG_IGN);
102 #endif
103 #endif
104     return true;
105 }
106
107 Pipe::Pipe(int port_to_use) : m_p(new Rep)
108 {
109 #ifdef WIN32
110     WSADATA wsaData;
111     WORD wVersionRequested = MAKEWORD(2, 0);
112     if (WSAStartup( wVersionRequested, &wsaData ))
113         throw Pipe::Error("WSAStartup failed");
114 #else
115     port_to_use = 0;  // we'll just use pipe on Unix
116 #endif
117     if (port_to_use)
118     {
119         // create server socket
120         m_p->m_socket = socket(AF_INET, SOCK_STREAM, 0);
121         if (m_p->m_socket < 0)
122             throw Pipe::Error("could not create socket");
123 #ifndef WIN32
124         unsigned long one = 1;
125         if (setsockopt(m_p->m_socket, SOL_SOCKET, SO_REUSEADDR, (char*) 
126                        &one, sizeof(one)) < 0)
127             throw Pipe::Error("setsockopt error");
128 #endif
129         // bind server socket
130         struct sockaddr_in add;
131         add.sin_family = AF_INET;
132         add.sin_port = htons(port_to_use);
133         add.sin_addr.s_addr = INADDR_ANY;
134         struct sockaddr *addr = ( struct sockaddr *) &add;
135       
136         if (bind(m_p->m_socket, addr, sizeof(struct sockaddr_in)))
137             throw Pipe::Error("could not bind on socket");
138         
139         if (listen(m_p->m_socket, 3) < 0)
140             throw Pipe::Error("could not listen on socket");
141
142         // client socket
143         unsigned int tmpadd;
144         tmpadd = (unsigned) inet_addr("127.0.0.1");
145         if (tmpadd)
146             memcpy(&add.sin_addr.s_addr, &tmpadd, sizeof(struct in_addr));
147         else
148             throw Pipe::Error("inet_addr failed");
149             
150         m_p->m_fd[1] = socket(AF_INET, SOCK_STREAM, 0);
151         if (m_p->m_fd[1] < 0)
152             throw Pipe::Error("could not create socket");
153         
154         m_p->nonblock(m_p->m_fd[1]);
155
156         if (connect(m_p->m_fd[1], addr, sizeof(*addr)) < 0)
157         {
158 #ifdef WIN32
159             if (WSAGetLastError() != WSAEWOULDBLOCK)
160                 throw Pipe::Error("could not connect to socket");
161 #else
162             if (errno != EINPROGRESS)
163                 throw Pipe::Error("could not connect to socket");
164 #endif
165         }
166
167         // server accept
168         struct sockaddr caddr;
169 #ifdef WIN32
170         int caddr_len = sizeof(caddr);
171 #else
172         socklen_t caddr_len = sizeof(caddr);
173 #endif
174         m_p->m_fd[0] = accept(m_p->m_socket, &caddr, &caddr_len);
175         if (m_p->m_fd[0] < 0)
176             throw Pipe::Error("could not accept on socket");
177
178         // complete connect
179         fd_set write_set;
180         FD_ZERO(&write_set);
181         FD_SET(m_p->m_fd[1], &write_set);
182         int r = select(m_p->m_fd[1]+1, 0, &write_set, 0, 0);
183         if (r != 1)
184             throw Pipe::Error("could not complete connect");
185
186         m_p->close(m_p->m_socket);
187     }
188     else
189     {
190 #ifndef WIN32
191         if (pipe(m_p->m_fd))
192             throw Pipe::Error("pipe failed");
193         else
194         {
195             assert(m_p->m_fd[0] >= 0);
196             assert(m_p->m_fd[1] >= 0);
197         }
198 #endif
199     }
200 }
201
202 Pipe::~Pipe()
203 {
204     m_p->close(m_p->m_fd[0]);
205     m_p->close(m_p->m_fd[1]);
206     m_p->close(m_p->m_socket);
207 #ifdef WIN32
208     WSACleanup();
209 #endif
210 }
211
212 int &Pipe::read_fd() const
213 {
214     return m_p->m_fd[0];
215 }
216
217 int &Pipe::write_fd() const
218 {
219     return m_p->m_fd[1];
220 }
221
222 /*
223  * Local variables:
224  * c-basic-offset: 4
225  * indent-tabs-mode: nil
226  * c-file-style: "stroustrup"
227  * End:
228  * vim: shiftwidth=4 tabstop=8 expandtab
229  */
230