Style(9) cleanup.
[dragonfly.git] / usr.sbin / setkey / test-policy.c
1 /*
2  * Copyright (C) 1995, 1996, 1997, 1998, and 1999 WIDE Project.
3  * All rights reserved.
4  * 
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  * 3. Neither the name of the project nor the names of its contributors
14  *    may be used to endorse or promote products derived from this software
15  *    without specific prior written permission.
16  * 
17  * THIS SOFTWARE IS PROVIDED BY THE PROJECT AND CONTRIBUTORS ``AS IS'' AND
18  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
19  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
20  * ARE DISCLAIMED.  IN NO EVENT SHALL THE PROJECT OR CONTRIBUTORS BE LIABLE
21  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
22  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
23  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
24  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
25  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
26  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
27  * SUCH DAMAGE.
28  *
29  * $FreeBSD: src/usr.sbin/setkey/test-policy.c,v 1.1 2000/01/06 12:40:53 shin Exp $
30  * $DragonFly: src/usr.sbin/setkey/test-policy.c,v 1.3 2004/03/24 18:23:47 cpressey Exp $
31  */
32
33 #include <sys/types.h>
34 #include <sys/param.h>
35 #include <sys/socket.h>
36 #include <netinet/in.h>
37 #include <netinet6/in6.h>
38 #include <netkey/keyv2.h>
39 #include <stdlib.h>
40 #include <string.h>
41
42 #include <netinet6/ipsec.h>
43
44 char *requests[] = {
45 "must_error",           /* must be error */
46 "ipsec must_error",     /* must be error */
47 "ipsec esp/must_error", /* must be error */
48 "discard",
49 "none",
50 "entrust",
51 "bypass",               /* may be error */
52 "ipsec esp",            /* must be error */
53 "ipsec ah/require",
54 "ipsec ah/use/",
55 "ipsec esp/require ah/default/203.178.141.194",
56 "ipsec ah/use/203.178.141.195 esp/use/203.178.141.194",
57 "ipsec esp/elf.wide.ydc.co.jp esp/www.wide.ydc.co.jp"
58 "
59 ipsec esp/require ah/use esp/require/10.0.0.1
60 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1
61 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1
62 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1
63 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1
64 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1
65 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1
66 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1 ah/use/3ffe:501:481d::1
67 ah/use/3ffe:501:481d::1  ah/use/3ffe:501:481d::1ah/use/3ffe:501:481d::1
68 ",
69 };
70
71 u_char  *p_secpolicy;
72
73 int     test(char *buf, int family);
74 char    *setpolicy(char *req);
75
76 int
77 main(int argc __unused, char **argv __unused)
78 {
79         int i;
80         char *buf;
81
82         for (i = 0; i < sizeof(requests)/sizeof(requests[0]); i++) {
83                 printf("* requests:[%s]\n", requests[i]);
84                 if ((buf = setpolicy(requests[i])) == NULL)
85                         continue;
86                 printf("\tsetlen:%d\n", PFKEY_EXTLEN(buf));
87
88                 printf("\tPF_INET:\n");
89                 test(buf, PF_INET);
90
91                 printf("\tPF_INET6:\n");
92                 test(buf, PF_INET6);
93                 free(buf);
94         }
95 }
96
97 int
98 test(char *policy, int family)
99 {
100         int so, proto, optname;
101         int len;
102         char getbuf[1024];
103         char *buf = NULL;
104
105         switch (family) {
106         case PF_INET:
107                 proto = IPPROTO_IP;
108                 optname = IP_IPSEC_POLICY;
109                 break;
110         case PF_INET6:
111                 proto = IPPROTO_IPV6;
112                 optname = IPV6_IPSEC_POLICY;
113                 break;
114         }
115
116         if ((so = socket(family, SOCK_DGRAM, 0)) < 0)
117                 perror("socket");
118
119         if (setsockopt(so, proto, optname, policy, PFKEY_EXTLEN(policy)) < 0)
120                 perror("setsockopt");
121
122         len = sizeof(getbuf);
123         memset(getbuf, 0, sizeof(getbuf));
124         if (getsockopt(so, proto, optname, getbuf, &len) < 0)
125                 perror("getsockopt");
126
127         printf("\tgetlen:%d\n", len);
128
129         if ((buf = ipsec_dump_policy(getbuf, NULL)) == NULL)
130                 ipsec_strerror();
131         else
132                 printf("\t[%s]\n", buf);
133
134         free(buf);
135
136         close (so);
137 }
138
139 char *
140 setpolicy(char *req)
141 {
142         int len;
143         char *buf;
144
145         if ((len = ipsec_get_policylen(req)) < 0) {
146                 printf("ipsec_get_policylen: %s\n", ipsec_strerror());
147                 return NULL;
148         }
149
150         if ((buf = malloc(len)) == NULL) {
151                 perror("malloc");
152                 return NULL;
153         }
154
155         if ((len = ipsec_set_policy(buf, len, req)) < 0) {
156                 printf("ipsec_set_policy: %s\n", ipsec_strerror());
157                 free(buf);
158                 return NULL;
159         }
160
161         return buf;
162 }