--- /dev/null
+// $Id: RawINetSocketHandle.test.cc 597 2008-01-15 09:16:20Z g0dil $
+//
+// Copyright (C) 2007
+// Fraunhofer Institute for Open Communication Systems (FOKUS)
+// Competence Center NETwork research (NET), St. Augustin, GERMANY
+// David Wagner <dw6@berlios.de>
+//
+// This program 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 of the License, or
+// (at your option) any later version.
+//
+// This program 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.,
+// 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
+
+/** \file
+ \brief RawINetSocketHandle unit tests */
+
+#include <sys/types.h>
+#include <sys/ioctl.h>
+#include <sys/wait.h>
+#include <unistd.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include "RawINetSocketHandle.hh"
+#include <iostream>
+
+#include "../../../Utils/auto_unit_test.hh"
+#include <boost/test/test_tools.hpp>
+
+namespace {
+
+ void error(char const * fn, char const * proc="")
+ {
+ std::cerr << "\n" << proc << ((*proc)?": ":"") << fn << ": " << strerror(errno) << std::endl;
+ }
+
+ void fail(char const * proc, char const * fn)
+ {
+ error(fn,proc);
+ _exit(1);
+ }
+
+ int server_pid = 0;
+
+ void start(void (*fn)())
+ {
+ server_pid = ::fork();
+ if (server_pid < 0) BOOST_FAIL("fork()");
+ if (server_pid == 0) {
+ (*fn)();
+ _exit(0);
+ }
+ }
+
+ void wait()
+ {
+ int status;
+ if (waitpid(server_pid,&status,0)<0)
+ BOOST_FAIL("waitpid()");
+ BOOST_CHECK_EQUAL( status , 0 );
+ }
+
+ void stop()
+ {
+ if (server_pid) {
+ kill(server_pid,9);
+ wait();
+ server_pid = 0;
+ }
+ }
+
+}
+
+///////////////////////////////////////////////////////////////////////////
+
+namespace {
+
+ void server_v4() //listen for packets with proto=47 (GRE) and resend them with proto=48
+ {
+ struct sockaddr_in sin;
+ ::memset(&sin,0,sizeof(sin));
+ sin.sin_family = AF_INET;
+ sin.sin_port = htons(0);
+ sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
+
+ int sockrec = socket(PF_INET,SOCK_RAW,47);
+ if (sockrec<0) fail("server_v4","socket(rec)");
+ int socksend = socket(PF_INET,SOCK_RAW,48);
+ if (socksend<0) fail("server_v4","socket(send)");
+
+ char buffer[1024];
+ while (1) {
+ int n = read(sockrec,buffer,1024);
+ if (n == 20+4 )//&& strncmp(,"QUIT",4) == 0)
+ break;
+ sleep(1);
+ //jaja, fieses gehacke...
+ sendto(socksend,buffer+20,n-20,0,(struct sockaddr *)&sin,sizeof(sin));
+ }
+
+ if (close(sockrec) < 0) fail("server_v4","close(rec)");
+ if (close(socksend) < 0) fail("server_v4","close(send)");
+ }
+ void server_v6() //listen for packets with proto=47 (GRE) and resend them with proto=48
+ {
+ struct sockaddr_in6 sin;
+ ::memset(&sin,0,sizeof(sin));
+ sin.sin6_family = AF_INET6;
+ sin.sin6_port = htons(0);
+ inet_pton(AF_INET6, "::1", &sin.sin6_addr);
+ int sockrec6 = socket(PF_INET6,SOCK_RAW,47);
+ if (sockrec6<0) fail("server_v6","socket(rec)");
+ int socksend6 = socket(PF_INET6,SOCK_RAW,48);
+ if (socksend6<0) fail("server_v6","socket(send)");
+ char buffer[1024];
+ while (1) {
+ int n = read(sockrec6,buffer,1024);
+ if (n<0) fail("server_v6","read(sockrec6)");
+ if (n == 4 && strncmp(buffer,"QUIT",4) == 0)
+ break;
+ sleep(1);
+ //jaja, fieses gehacke...
+ n = sendto(socksend6,buffer,n,0,(struct sockaddr *)&sin,sizeof(sin));
+ if (n<0) fail("server_v6","sendto(socksend6)");
+ }
+ if (close(sockrec6) < 0) fail("server_v6","close(rec)");
+ if (close(socksend6) < 0) fail("server_v6","close(send)");
+ }
+
+}
+
+BOOST_AUTO_UNIT_TEST(RawV4ClientSocketHandle)
+{
+ if (getuid() != 0) {
+ BOOST_WARN_MESSAGE(false, "Cannot test senf::RawV4SocketHandle as non-root user");
+ return;
+ }
+ try {
+ std::string test = "TEST-WRITE";
+ alarm(10);
+ start(server_v4);
+ senf::RawV4ClientSocketHandle sock(47); //IPPROTO_GRE
+ SENF_CHECK_NO_THROW( sock.protocol().rcvbuf(2048) );
+ BOOST_CHECK_EQUAL( sock.protocol().rcvbuf(), 2048u );
+ SENF_CHECK_NO_THROW( sock.protocol().sndbuf(2048) );
+ BOOST_CHECK_EQUAL( sock.protocol().sndbuf(), 2048u );
+ SENF_CHECK_NO_THROW( sock.writeto(senf::INet4SocketAddress("127.0.0.1:0"), test) );
+ senf::RawV4ClientSocketHandle sockrec(48); //IPPROTO_GRE+1
+ std::string in = sockrec.read();
+ BOOST_CHECK_EQUAL(in.substr(20), test);
+ SENF_CHECK_NO_THROW( sock.writeto(senf::INet4SocketAddress("127.0.0.1:0"),"QUIT"));
+ //sock.close();
+ //sockrec.close();
+ alarm(0);
+ } catch (...) {
+ alarm(0);
+ throw;
+ }
+}
+
+BOOST_AUTO_UNIT_TEST(RawV6ClientSocketHandle)
+{
+ if (getuid() != 0) {
+ BOOST_WARN_MESSAGE(false, "Cannot test senf::RawV6SocketHandle as non-root user");
+ return;
+ }
+ try {
+ std::string test = "TEST-WRITE";
+ alarm(5);
+ start(server_v6);
+ sleep(1);
+ senf::RawV6ClientSocketHandle sock(47); //IPPROTO_GRE
+ SENF_CHECK_NO_THROW( sock.protocol().rcvbuf(2048) );
+ BOOST_CHECK_EQUAL( sock.protocol().rcvbuf(), 2048u );
+ SENF_CHECK_NO_THROW( sock.protocol().sndbuf(2048) );
+ BOOST_CHECK_EQUAL( sock.protocol().sndbuf(), 2048u );
+ SENF_CHECK_NO_THROW( sock.writeto(senf::INet6SocketAddress("[::1]:0"), test) );
+ senf::RawV6ClientSocketHandle sockrec(48); //IPPROTO_GRE+1
+ std::string in = sockrec.read();
+ BOOST_CHECK_EQUAL(in, test);
+ SENF_CHECK_NO_THROW( sock.writeto(senf::INet6SocketAddress("[::1]:0"),"QUIT"));
+ alarm(0);
+ } catch (...) {
+ alarm(0);
+ throw;
+ }
+}
+
+///////////////////////////////cc.e////////////////////////////////////////
+#undef prefix_
+
+\f
+// Local Variables:
+// mode: c++
+// fill-column: 100
+// comment-column: 40
+// c-file-style: "senf"
+// indent-tabs-mode: nil
+// ispell-local-dictionary: "american"
+// compile-command: "scons -u test"
+// End:
projects / senf.git / blobdiff