metaproxy  1.13.0
pipe.cpp
Go to the documentation of this file.
1 /* This file is part of Metaproxy.
2  Copyright (C) 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 #include <string.h>
51 
52 #include <deque>
53 
54 #include <yazpp/socket-observer.h>
55 #include <yaz/log.h>
56 
57 #include "pipe.hpp"
58 
59 namespace mp = metaproxy_1;
60 
61 namespace metaproxy_1 {
62  class Pipe::Rep : public boost::noncopyable {
63  friend class Pipe;
64  Rep();
65  int m_fd[2];
66  int m_socket;
67  bool nonblock(int s);
68  void close(int &fd);
69  };
70 }
71 
72 using namespace mp;
73 
74 void Pipe::Rep::close(int &fd)
75 {
76 #ifdef WIN32
77  if (fd != -1)
78  ::closesocket(fd);
79 #else
80  if (fd != -1)
81  ::close(fd);
82 #endif
83  fd = -1;
84 }
85 
87 {
88  m_fd[0] = m_fd[1] = -1;
89  m_socket = -1;
90 }
91 
93 {
94 #ifdef WIN32
95  unsigned long tru = 1;
96  if (ioctlsocket(s, FIONBIO, &tru) < 0)
97  return false;
98 #else
99  if (fcntl(s, F_SETFL, O_NONBLOCK) < 0)
100  return false;
101 #ifndef MSG_NOSIGNAL
102  signal (SIGPIPE, SIG_IGN);
103 #endif
104 #endif
105  return true;
106 }
107 
108 Pipe::Pipe(int port_to_use) : m_p(new Rep)
109 {
110 #ifdef WIN32
111  WSADATA wsaData;
112  WORD wVersionRequested = MAKEWORD(2, 0);
113  if (WSAStartup( wVersionRequested, &wsaData ))
114  throw Pipe::Error("WSAStartup failed");
115 #else
116  port_to_use = 0; // we'll just use pipe on Unix
117 #endif
118  if (port_to_use)
119  {
120  // create server socket
121  m_p->m_socket = socket(AF_INET, SOCK_STREAM, 0);
122  if (m_p->m_socket < 0)
123  throw Pipe::Error("could not create socket");
124 #ifndef WIN32
125  unsigned long one = 1;
126  if (setsockopt(m_p->m_socket, SOL_SOCKET, SO_REUSEADDR, (char*)
127  &one, sizeof(one)) < 0)
128  throw Pipe::Error("setsockopt error");
129 #endif
130  // bind server socket
131  struct sockaddr_in add;
132  add.sin_family = AF_INET;
133  add.sin_port = htons(port_to_use);
134  add.sin_addr.s_addr = INADDR_ANY;
135  struct sockaddr *addr = ( struct sockaddr *) &add;
136 
137  if (bind(m_p->m_socket, addr, sizeof(struct sockaddr_in)))
138  throw Pipe::Error("could not bind on socket");
139 
140  if (listen(m_p->m_socket, 3) < 0)
141  throw Pipe::Error("could not listen on socket");
142 
143  // client socket
144  unsigned int tmpadd;
145  tmpadd = (unsigned) inet_addr("127.0.0.1");
146  if (tmpadd)
147  memcpy(&add.sin_addr.s_addr, &tmpadd, sizeof(struct in_addr));
148  else
149  throw Pipe::Error("inet_addr failed");
150 
151  m_p->m_fd[1] = socket(AF_INET, SOCK_STREAM, 0);
152  if (m_p->m_fd[1] < 0)
153  throw Pipe::Error("could not create socket");
154 
155  m_p->nonblock(m_p->m_fd[1]);
156 
157  if (connect(m_p->m_fd[1], addr, sizeof(*addr)) < 0)
158  {
159 #ifdef WIN32
160  if (WSAGetLastError() != WSAEWOULDBLOCK)
161  throw Pipe::Error("could not connect to socket");
162 #else
163  if (errno != EINPROGRESS)
164  throw Pipe::Error("could not connect to socket");
165 #endif
166  }
167 
168  // server accept
169  struct sockaddr caddr;
170 #ifdef WIN32
171  int caddr_len = sizeof(caddr);
172 #else
173  socklen_t caddr_len = sizeof(caddr);
174 #endif
175  m_p->m_fd[0] = accept(m_p->m_socket, &caddr, &caddr_len);
176  if (m_p->m_fd[0] < 0)
177  throw Pipe::Error("could not accept on socket");
178 
179  // complete connect
180  fd_set write_set;
181  FD_ZERO(&write_set);
182  FD_SET(m_p->m_fd[1], &write_set);
183  int r = select(m_p->m_fd[1]+1, 0, &write_set, 0, 0);
184  if (r != 1)
185  throw Pipe::Error("could not complete connect");
186 
187  m_p->close(m_p->m_socket);
188  }
189  else
190  {
191 #ifndef WIN32
192  if (pipe(m_p->m_fd))
193  throw Pipe::Error("pipe failed");
194  else
195  {
196  assert(m_p->m_fd[0] >= 0);
197  assert(m_p->m_fd[1] >= 0);
198  }
199 #endif
200  }
201 }
202 
204 {
205  m_p->close(m_p->m_fd[0]);
206  m_p->close(m_p->m_fd[1]);
207  m_p->close(m_p->m_socket);
208 #ifdef WIN32
209  WSACleanup();
210 #endif
211 }
212 
213 int &Pipe::read_fd() const
214 {
215  return m_p->m_fd[0];
216 }
217 
218 int &Pipe::write_fd() const
219 {
220  return m_p->m_fd[1];
221 }
222 
223 /*
224  * Local variables:
225  * c-basic-offset: 4
226  * c-file-style: "Stroustrup"
227  * indent-tabs-mode: nil
228  * End:
229  * vim: shiftwidth=4 tabstop=8 expandtab
230  */
231 
Pipe(int port_to_use)
Definition: pipe.cpp:108
int & read_fd() const
Definition: pipe.cpp:213
int & write_fd() const
Definition: pipe.cpp:218
void close(int &fd)
Definition: pipe.cpp:74
bool nonblock(int s)
Definition: pipe.cpp:92
boost::scoped_ptr< Rep > m_p
Definition: pipe.hpp:42