4 // Fraunhofer Institute for Open Communication Systems (FOKUS)
5 // Competence Center NETwork research (NET), St. Augustin, GERMANY
6 // Stefan Bund <g0dil@berlios.de>
8 // This program is free software; you can redistribute it and/or modify
9 // it under the terms of the GNU General Public License as published by
10 // the Free Software Foundation; either version 2 of the License, or
11 // (at your option) any later version.
13 // This program is distributed in the hope that it will be useful,
14 // but WITHOUT ANY WARRANTY; without even the implied warranty of
15 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 // GNU General Public License for more details.
18 // You should have received a copy of the GNU General Public License
19 // along with this program; if not, write to the
20 // Free Software Foundation, Inc.,
21 // 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
24 \brief TCPSocketHandle unit tests */
26 //#include "TCPSocketHandle.test.hh"
27 //#include "TCPSocketHandle.test.ih"
30 #include <sys/types.h>
31 #include <sys/ioctl.h>
34 #include <sys/socket.h>
35 #include <netinet/in.h>
37 #include "TCPSocketHandle.hh"
39 #include "../../../Utils/auto_unit_test.hh"
40 #include <boost/test/test_tools.hpp>
43 ///////////////////////////////cc.p////////////////////////////////////////
47 void error(char const * fn, char const * proc="")
49 std::cerr << "\n" << proc << ((*proc)?": ":"") << fn << ": " << strerror(errno) << std::endl;
52 void fail(char const * proc, char const * fn)
60 void start(void (*fn)())
62 server_pid = ::fork();
63 if (server_pid < 0) BOOST_FAIL("fork()");
64 if (server_pid == 0) {
65 signal(SIGCHLD, SIG_IGN);
69 signal(SIGCHLD, SIG_DFL);
76 if (waitpid(server_pid,&status,0)<0)
77 BOOST_FAIL("waitpid()");
78 BOOST_CHECK_EQUAL( status , 0 );
92 ///////////////////////////////////////////////////////////////////////////
98 int serv = socket(PF_INET,SOCK_STREAM,0);
99 if (serv<0) fail("server_v4","socket()");
101 if (setsockopt(serv,SOL_SOCKET,SO_REUSEADDR,&v,sizeof(v))<0)
102 fail("server_v4","setsockopt()");
103 struct sockaddr_in sin;
104 ::memset(&sin,0,sizeof(sin));
105 sin.sin_family = AF_INET;
106 sin.sin_port = htons(12345);
107 sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
108 if (bind(serv,(struct sockaddr *)&sin,sizeof(sin))<0) fail("server_v4","bind()");
109 if (listen(serv,1)<0) fail("server_v4","listen()");
110 int sock = accept(serv,0,0);
111 if (sock < 0) fail("server_v4","accept()");
115 int n = read(sock,buffer,1024);
116 if (n == 4 && strncmp(buffer,"QUIT",4) == 0)
118 write(sock,buffer,n);
121 if (shutdown(sock, SHUT_RDWR) < 0) fail("server_v4","shutdown()");
122 if (close(sock) < 0) fail("server_v4","close()");
123 if (close(serv) < 0) fail("server_v4","close()");
128 int serv = socket(PF_INET6,SOCK_STREAM,0);
129 if (serv<0) fail("server_v6","socket()");
131 if (setsockopt(serv,SOL_SOCKET,SO_REUSEADDR,&v,sizeof(v))<0)
132 fail("server_v6","setsockopt()");
133 struct sockaddr_in6 sin;
134 ::memset(&sin,0,sizeof(sin));
135 sin.sin6_family = AF_INET6;
136 sin.sin6_port = htons(12345);
137 sin.sin6_addr = in6addr_loopback;
138 if (bind(serv,(struct sockaddr *)&sin,sizeof(sin))<0) fail("server_v6","bind()");
139 if (listen(serv,1)<0) fail("server_v6","listen()");
140 int sock = accept(serv,0,0);
141 if (sock < 0) fail("server_v6","accept()");
145 int n = read(sock,buffer,1024);
146 if (n == 4 && strncmp(buffer,"QUIT",4) == 0)
148 write(sock,buffer,n);
151 if (shutdown(sock, SHUT_RDWR) < 0) fail("server_v6","shutdown()");
152 if (close(sock) < 0) fail("server_v6","close()");
153 if (close(serv) < 0) fail("server_v6","close()");
158 BOOST_AUTO_UNIT_TEST(tcpv4ClientSocketHandle)
161 senf::TCPv4ClientSocketHandle sock;
163 BOOST_CHECK_THROW( sock.connect(senf::INet4SocketAddress("127.0.0.1:12345")),
164 senf::SystemException );
170 senf::TCPv4ClientSocketHandle sock;
171 SENF_CHECK_NO_THROW( sock.bind(senf::INet4SocketAddress("127.0.0.1:23456")) );
172 SENF_CHECK_NO_THROW( sock.connect(senf::INet4SocketAddress("127.0.0.1:12345")) );
173 BOOST_CHECK( sock.peer() == senf::INet4SocketAddress("127.0.0.1:12345") );
174 BOOST_CHECK( sock.local() == senf::INet4SocketAddress("127.0.0.1:23456") );
175 BOOST_CHECK( sock.blocking() );
176 SENF_CHECK_NO_THROW( sock.protocol().rcvbuf(2048) );
177 BOOST_CHECK_EQUAL( sock.protocol().rcvbuf(), 2048u );
178 SENF_CHECK_NO_THROW( sock.protocol().sndbuf(2048) );
179 BOOST_CHECK_EQUAL( sock.protocol().sndbuf(), 2048u );
180 SENF_CHECK_NO_THROW( sock.write(std::string("TEST-WRITE")) );
181 BOOST_CHECK_EQUAL( sock.read(), std::string("TEST-WRITE") );
182 BOOST_CHECK( !sock.eof() );
183 sock.write(std::string("QUIT"));
187 BOOST_CHECK_EQUAL( sock.read(), "" );
188 BOOST_CHECK( sock.eof() );
189 BOOST_CHECK( !sock );
200 senf::TCPv4ClientSocketHandle sock;
202 // Since this is a TCP socket, most of the calls will fail or
203 // are at least not sensible ...
204 // I'll have to move those to a UDPSocket test ... they should
205 // realy only be in the UDP Protocol implementation
206 // SENF_CHECK_NO_THROW( sock.protocol().mcTTL() );
207 // BOOST_CHECK_THROW( sock.protocol().mcTTL(1), senf::SystemException );
208 // SENF_CHECK_NO_THROW( sock.protocol().mcLoop() );
209 // SENF_CHECK_NO_THROW( sock.protocol().mcLoop(false) );
210 // SENF_CHECK_NO_THROW( sock.protocol().mcAddMembership("224.0.0.1:0") );
211 // SENF_CHECK_NO_THROW( sock.protocol().mcAddMembership("224.0.0.1:0","127.0.0.1:0") );
212 // SENF_CHECK_NO_THROW( sock.protocol().mcDropMembership("224.0.0.1:0","127.0.0.1:0") );
213 // SENF_CHECK_NO_THROW( sock.protocol().mcDropMembership("224.0.0.1:0") );
214 // BOOST_CHECK_THROW( sock.protocol().mcIface("lo"), senf::SystemException );
216 // The following setsockopts are hard to REALLY test ...
217 SENF_CHECK_NO_THROW( sock.protocol().nodelay(true) );
218 BOOST_CHECK( sock.protocol().nodelay() );
219 BOOST_CHECK_EQUAL( sock.protocol().siocinq(), 0u );
220 BOOST_CHECK_EQUAL( sock.protocol().siocoutq(), 0u );
222 SENF_CHECK_NO_THROW( sock.protocol().reuseaddr(true) );
223 BOOST_CHECK( sock.protocol().reuseaddr() );
224 SENF_CHECK_NO_THROW( sock.protocol().linger(true,0) );
225 BOOST_CHECK( sock.protocol().linger() == std::make_pair(true, 0u) );
229 BOOST_AUTO_UNIT_TEST(tcpv6ClientSocketHandle)
232 senf::TCPv6ClientSocketHandle sock;
234 BOOST_CHECK_THROW( sock.connect(senf::INet6SocketAddress("[::1]:12345")),
235 senf::SystemException );
241 senf::TCPv6ClientSocketHandle sock;
242 SENF_CHECK_NO_THROW( sock.bind(senf::INet6SocketAddress("[::1]:23456")) );
243 SENF_CHECK_NO_THROW( sock.connect(senf::INet6SocketAddress("[::1]:12345")) );
244 BOOST_CHECK( sock.peer() == senf::INet6SocketAddress("[::1]:12345") );
245 BOOST_CHECK( sock.local() == senf::INet6SocketAddress("[::1]:23456") );
246 BOOST_CHECK( sock.blocking() );
247 SENF_CHECK_NO_THROW( sock.protocol().rcvbuf(2048) );
248 BOOST_CHECK_EQUAL( sock.protocol().rcvbuf(), 2048u );
249 SENF_CHECK_NO_THROW( sock.protocol().sndbuf(2048) );
250 BOOST_CHECK_EQUAL( sock.protocol().sndbuf(), 2048u );
251 SENF_CHECK_NO_THROW( sock.write(std::string("TEST-WRITE")) );
252 BOOST_CHECK_EQUAL( sock.read(), std::string("TEST-WRITE") );
253 // this fails with ENOFILE ... why ????
254 // SENF_CHECK_NO_THROW( sock.protocol().timestamp() );
255 BOOST_CHECK( !sock.eof() );
256 sock.write(std::string("QUIT"));
260 BOOST_CHECK_EQUAL( sock.read(), "" );
261 BOOST_CHECK( sock.eof() );
262 BOOST_CHECK( !sock );
273 senf::TCPv6ClientSocketHandle sock;
275 // The following setsockopts are hard to REALLY test ...
276 SENF_CHECK_NO_THROW( sock.protocol().nodelay(true) );
277 BOOST_CHECK( sock.protocol().nodelay() );
278 BOOST_CHECK_EQUAL( sock.protocol().siocinq(), 0u );
279 BOOST_CHECK_EQUAL( sock.protocol().siocoutq(), 0u );
281 SENF_CHECK_NO_THROW( sock.protocol().reuseaddr(true) );
282 BOOST_CHECK( sock.protocol().reuseaddr() );
283 SENF_CHECK_NO_THROW( sock.protocol().linger(true,0) );
284 BOOST_CHECK( sock.protocol().linger() == std::make_pair(true, 0u) );
288 ///////////////////////////////////////////////////////////////////////////
294 int sock = socket(PF_INET,SOCK_STREAM,0);
295 if (sock<0) fail("client_v4","socket()");
296 struct sockaddr_in sin;
297 ::memset(&sin,0,sizeof(sin));
298 sin.sin_family = AF_INET;
299 sin.sin_port = htons(12346);
300 sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
301 if (connect(sock,(struct sockaddr *)&sin,sizeof(sin)) < 0)
302 fail("client_v4","connect()");
306 int n = read(sock,buffer,1024);
307 if (n == 4 && strncmp(buffer,"QUIT",4) == 0)
309 write(sock,buffer,n);
312 if (shutdown(sock, SHUT_RDWR) < 0) fail("client_v4","shutdown()");
313 if (close(sock) < 0) fail("client_v4","close()");
318 int sock = socket(PF_INET6,SOCK_STREAM,0);
319 if (sock<0) fail("client_v6","socket()");
320 struct sockaddr_in6 sin;
321 ::memset(&sin,0,sizeof(sin));
322 sin.sin6_family = AF_INET6;
323 sin.sin6_port = htons(12347);
324 sin.sin6_addr = in6addr_loopback;
325 if (connect(sock,(struct sockaddr *)&sin,sizeof(sin)) < 0)
326 fail("client_v6","connect()");
330 int n = read(sock,buffer,1024);
331 if (n == 4 && strncmp(buffer,"QUIT",4) == 0)
333 write(sock,buffer,n);
336 if (shutdown(sock, SHUT_RDWR) < 0) fail("client_v6","shutdown()");
337 if (close(sock) < 0) fail("client_v6","close()");
342 BOOST_AUTO_UNIT_TEST(tcpv4ServerSocketHandle)
346 BOOST_CHECKPOINT("Opening server socket");
347 senf::TCPv4ServerSocketHandle server (senf::INet4SocketAddress("127.0.0.1:12346"));
348 BOOST_CHECKPOINT("Starting client");
351 BOOST_CHECKPOINT("Accepting connection");
352 senf::TCPv4ClientSocketHandle client = server.accept();
353 SENF_CHECK_NO_THROW(client.write(std::string("QUIT")));
355 BOOST_CHECKPOINT("Stopping client");
368 BOOST_AUTO_UNIT_TEST(tcpv6ServerSocketHandle)
372 BOOST_CHECKPOINT("Opening server socket");
373 senf::TCPv6ServerSocketHandle server (senf::INet6SocketAddress("[::1]:12347"));
374 BOOST_CHECKPOINT("Starting client");
377 BOOST_CHECKPOINT("Accepting connection");
378 senf::TCPv6ClientSocketHandle client = server.accept();
379 SENF_CHECK_NO_THROW(client.write(std::string("QUIT")));
381 BOOST_CHECKPOINT("Stopping client");
394 ///////////////////////////////cc.e////////////////////////////////////////
401 // c-file-style: "senf"
402 // indent-tabs-mode: nil
403 // ispell-local-dictionary: "american"
404 // compile-command: "scons -u test"
405 // comment-column: 40