Author: bz
Date: Fri Feb 17 03:25:26 2012
New Revision: 231855
URL: http://svn.freebsd.org/changeset/base/231855

Log:
  Add regression tests for the setsockopt(2) SO_SETFIB socket option.
  
  Check that the expected domain(9) families all handle the socket option
  correctly and do proper bounds checks.  This would catch bugs as fixed
  in (r230938,)r230981.
  
  Sponsored by: Cisco Systems, Inc.

Added:
  head/tools/regression/sockets/so_setfib/
  head/tools/regression/sockets/so_setfib/Makefile   (contents, props changed)
  head/tools/regression/sockets/so_setfib/so_setfib.c   (contents, props 
changed)
  head/tools/regression/sockets/so_setfib/so_setfib.t   (contents, props 
changed)

Added: head/tools/regression/sockets/so_setfib/Makefile
==============================================================================
--- /dev/null   00:00:00 1970   (empty, because file is newly added)
+++ head/tools/regression/sockets/so_setfib/Makefile    Fri Feb 17 03:25:26 
2012        (r231855)
@@ -0,0 +1,14 @@
+# $FreeBSD$
+
+PROG=  so_setfib
+NO_MAN=
+WARNS?=        6
+
+.ifdef INET6
+CFLAGS+=       -DINET6
+.endif
+.ifdef INET
+CFLAGS+=       -DINET
+.endif
+
+.include <bsd.prog.mk>

Added: head/tools/regression/sockets/so_setfib/so_setfib.c
==============================================================================
--- /dev/null   00:00:00 1970   (empty, because file is newly added)
+++ head/tools/regression/sockets/so_setfib/so_setfib.c Fri Feb 17 03:25:26 
2012        (r231855)
@@ -0,0 +1,189 @@
+/*-
+ * Copyright (c) 2012 Cisco Systems, Inc.
+ * All rights reserved.
+ *
+ * This software was developed by Bjoern Zeeb under contract to
+ * Cisco Systems, Inc..
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+ * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
+ * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+ * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+ * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+ * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+ * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+ * SUCH DAMAGE.
+ *
+ * $FreeBSD$
+ */
+
+/*
+ * Regression test on SO_SETFIB setsockopt(2).
+ * 
+ * Check that the expected domain(9) families all handle the socket option
+ * correctly and do proper bounds checks.
+ *
+ * Test plan:
+ * 1. Get system wide number of FIBs from sysctl and convert to index (-= 1).
+ * 2. For each protocol family (INET, INET6, ROUTE and LOCAL) open socketes of
+ *    type (STREAM, DGRAM and RAW) as supported.
+ * 3. Do a sequence of -2, -1, 0, .. n, n+1, n+2 SO_SETFIB sockopt calls,
+ *    expecting the first two and last two to fail (valid 0 ... n).
+ * 4. Try 3 random numbers.  Calculate result based on valid range.
+ * 5. Repeat for next domain family and type from (2) on.
+ */
+
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <sys/sysctl.h>
+
+#include <err.h>
+#include <errno.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+
+static struct t_dom {
+       int             domain;
+       const char      *name;
+} t_dom[] = {
+#ifdef INET6
+       { .domain = PF_INET6, .name = "PF_INET6" },
+#endif
+#ifdef INET
+       { .domain = PF_INET, .name = "PF_INET" },
+#endif
+       { .domain = PF_ROUTE, .name = "PF_ROUTE" },
+       { .domain = PF_LOCAL, .name = "PF_LOCAL" },
+};
+
+static struct t_type {
+       int             type;
+       const char      *name;
+} t_type[] = {
+       { .type = SOCK_STREAM, .name = "SOCK_STREAM" },
+       { .type = SOCK_DGRAM, .name = "SOCK_DGRAM" },
+       { .type = SOCK_RAW, .name = "SOCK_RAW" },
+};
+
+/*
+ * Number of FIBs as read from net.fibs sysctl - 1.  Initialize to clear out of
+ * bounds value to not accidentally run on a limited range. 
+ */
+static int rt_numfibs = -42;
+
+/* Number of test case. */
+static int testno = 1;
+
+
+/*
+ * Try the setsockopt with given FIB number i on socket s.
+ * Handle result given on error and valid range and errno.
+ */
+static void
+so_setfib(int s, int i, u_int dom, u_int type)
+{
+       int error;
+
+       error = setsockopt(s, SOL_SOCKET, SO_SETFIB, &i, sizeof(i));
+       /* For out of bounds we expect an error. */
+       if (error == -1 && (i < 0 || i > rt_numfibs))
+               printf("ok %d %s_%s_%d\n", testno, t_dom[dom].name,
+                   t_type[type].name, i);
+       else if (error != -1 && (i < 0 || i > rt_numfibs))
+               printf("not ok %d %s_%s_%d # setsockopt(%d, SOL_SOCKET, "
+                   "SO_SETFIB, %d, ..) unexpectedly succeeded\n", testno,
+                   t_dom[dom].name, t_type[type].name, i, s, i);
+       else if (error == 0)
+               printf("ok %d %s_%s_%d\n", testno, t_dom[dom].name,
+                   t_type[type].name, i);
+       else if (errno != EINVAL)
+               printf("not ok %d %s_%s_%d # setsockopt(%d, SOL_SOCKET, "
+                   "SO_SETFIB, %d, ..) unexpected error: %s\n", testno,
+                   t_dom[dom].name, t_type[type].name, i, s, i,
+                   strerror(errno));
+       else
+               printf("not ok %d %s_%s_%d\n", testno, t_dom[dom].name,
+                   t_type[type].name, i);
+
+       /* Test run done, next please. */
+       testno++;
+}
+
+/*
+ * Main test.  Open socket given domain family and type.  For each FIB, out of
+ * bounds FIB numbers and 3 random FIB numbers set the socket option.
+ */
+static void
+t(u_int dom, u_int type)
+{
+       int i, s;
+
+       /* PF_ROUTE only supports RAW socket types, while PF_LOCAL does not. */
+       if (t_dom[dom].domain == PF_ROUTE && t_type[type].type != SOCK_RAW)
+               return;
+       if (t_dom[dom].domain == PF_LOCAL && t_type[type].type == SOCK_RAW)
+               return;
+
+       /* Open socket for given combination. */
+       s = socket(t_dom[dom].domain, t_type[type].type, 0);
+       if (s == -1) {
+               printf("not ok %d %s_%s # socket(): %s\n", testno,
+                   t_dom[dom].name, t_type[type].name, strerror(errno));
+               return;
+       }
+       
+       /* Test FIBs -2, -1, 0, .. n, n + 1, n + 2. */
+       for (i = -2; i <= (rt_numfibs + 2); i++)
+               so_setfib(s, i, dom, type);
+
+       /* Test 3 random FIB numbers. */
+       for (i = 0; i < 3; i++)
+               so_setfib(s, (int)random(), dom, type);
+
+       /* Close socket. */
+       close(s);
+}
+
+/*
+ * Returns 0 if no program error, 1 on sysctlbyname error.
+ * Test results are communicated by printf("[not ]ok <n> ..").
+ */
+int
+main(int argc __unused, char *argv[] __unused)
+{
+       u_int i, j;
+       size_t s;
+
+       /* Initalize randomness. */
+       srandomdev();
+
+       /* Get number of FIBs supported by kernel. */
+       s = sizeof(rt_numfibs);
+       if (sysctlbyname("net.fibs", &rt_numfibs, &s, NULL, 0) == -1)
+               err(1, "sysctlbyname(net.fibs, ..)");
+       /* Adjust from number to index. */
+       rt_numfibs -= 1;
+
+       /* Run tests. */
+       for (i = 0; i < sizeof(t_dom) / sizeof(struct t_dom); i++)
+               for (j = 0; j < sizeof(t_type) / sizeof(struct t_type); j++)
+                       t(i, j);
+
+       return (0);
+}
+
+/* end */

Added: head/tools/regression/sockets/so_setfib/so_setfib.t
==============================================================================
--- /dev/null   00:00:00 1970   (empty, because file is newly added)
+++ head/tools/regression/sockets/so_setfib/so_setfib.t Fri Feb 17 03:25:26 
2012        (r231855)
@@ -0,0 +1,59 @@
+#!/bin/sh
+#-
+# Copyright (c) 2012 Cisco Systems, Inc.
+# All rights reserved.
+#
+# This software was developed by Bjoern Zeeb under contract to
+# Cisco Systems, Inc..
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions
+# are met:
+# 1. Redistributions of source code must retain the above copyright
+#    notice, this list of conditions and the following disclaimer.
+# 2. Redistributions in binary form must reproduce the above copyright
+#    notice, this list of conditions and the following disclaimer in the
+#    documentation and/or other materials provided with the distribution.
+#
+# THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
+# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
+# ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
+# OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+# HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
+# OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
+# SUCH DAMAGE.
+#
+# $FreeBSD$
+#
+
+cd `dirname $0`
+
+EXECUTABLE=`basename $0 .t`
+
+FIBS=`sysctl -n net.fibs`
+INET=`sysctl -n kern.features.inet`
+INET6=`sysctl -n kern.features.inet6`
+
+case "${INET}" in
+1)     OPTS="${OPTS} -DINET" ;;
+*)     INET=0 ;;
+esac
+case "${INET6}" in
+1)     OPTS="${OPTS} -DINET6" ;;
+*)     INET6=0 ;;
+esac
+
+make ${EXECUTABLE} ${OPTS} 2>&1 > /dev/null
+
+# two out of bounds on each side + 3 random
+FIBS=$((2 + FIBS + 2 + 3))
+# ROUTE and LOCAL are 1 domain together given 2 or 1 types only
+TESTS=$(((1 + ${INET} + ${INET6}) * 3 * ${FIBS}))
+
+echo "1..${TESTS}"
+
+exec ./${EXECUTABLE}
_______________________________________________
[email protected] mailing list
http://lists.freebsd.org/mailman/listinfo/svn-src-head
To unsubscribe, send any mail to "[email protected]"

Reply via email to