1a9643ea8Slogwang /*-
2*22ce4affSfengbojiang  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3*22ce4affSfengbojiang  *
4a9643ea8Slogwang  * Copyright (c) 2005 Paolo Pisati <[email protected]>
5a9643ea8Slogwang  * All rights reserved.
6a9643ea8Slogwang  *
7a9643ea8Slogwang  * Redistribution and use in source and binary forms, with or without
8a9643ea8Slogwang  * modification, are permitted provided that the following conditions
9a9643ea8Slogwang  * are met:
10a9643ea8Slogwang  * 1. Redistributions of source code must retain the above copyright
11a9643ea8Slogwang  *    notice, this list of conditions and the following disclaimer.
12a9643ea8Slogwang  * 2. Redistributions in binary form must reproduce the above copyright
13a9643ea8Slogwang  *    notice, this list of conditions and the following disclaimer in the
14a9643ea8Slogwang  *    documentation and/or other materials provided with the distribution.
15a9643ea8Slogwang  *
16a9643ea8Slogwang  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
17a9643ea8Slogwang  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18a9643ea8Slogwang  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19a9643ea8Slogwang  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
20a9643ea8Slogwang  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
21a9643ea8Slogwang  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
22a9643ea8Slogwang  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
23a9643ea8Slogwang  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
24a9643ea8Slogwang  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
25a9643ea8Slogwang  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
26a9643ea8Slogwang  * SUCH DAMAGE.
27a9643ea8Slogwang  */
28a9643ea8Slogwang 
29a9643ea8Slogwang #include <sys/cdefs.h>
30a9643ea8Slogwang __FBSDID("$FreeBSD$");
31a9643ea8Slogwang 
32a9643ea8Slogwang /*
33a9643ea8Slogwang  * Alias_dummy is just an empty skeleton used to demostrate how to write
34a9643ea8Slogwang  * a module for libalias, that will run unalterated in userland or in
35a9643ea8Slogwang  * kernel land.
36a9643ea8Slogwang  */
37a9643ea8Slogwang 
38a9643ea8Slogwang #ifdef _KERNEL
39a9643ea8Slogwang #include <sys/param.h>
40a9643ea8Slogwang #include <sys/kernel.h>
41a9643ea8Slogwang #include <sys/module.h>
42a9643ea8Slogwang #else
43a9643ea8Slogwang #include <errno.h>
44a9643ea8Slogwang #include <sys/types.h>
45a9643ea8Slogwang #include <stdio.h>
46a9643ea8Slogwang #endif
47a9643ea8Slogwang 
48a9643ea8Slogwang #include <netinet/in_systm.h>
49a9643ea8Slogwang #include <netinet/in.h>
50a9643ea8Slogwang #include <netinet/ip.h>
51a9643ea8Slogwang #include <netinet/udp.h>
52a9643ea8Slogwang 
53a9643ea8Slogwang #ifdef _KERNEL
54a9643ea8Slogwang #include <netinet/libalias/alias_local.h>
55a9643ea8Slogwang #include <netinet/libalias/alias_mod.h>
56a9643ea8Slogwang #else
57a9643ea8Slogwang #include "alias_local.h"
58a9643ea8Slogwang #include "alias_mod.h"
59a9643ea8Slogwang #endif
60a9643ea8Slogwang 
61a9643ea8Slogwang static void
62a9643ea8Slogwang AliasHandleDummy(struct libalias *la, struct ip *ip, struct alias_data *ah);
63a9643ea8Slogwang 
64a9643ea8Slogwang static int
fingerprint(struct libalias * la,struct alias_data * ah)65a9643ea8Slogwang fingerprint(struct libalias *la, struct alias_data *ah)
66a9643ea8Slogwang {
67a9643ea8Slogwang 
68a9643ea8Slogwang 	/*
69a9643ea8Slogwang 	 * Check here all the data that will be used later, if any field
70a9643ea8Slogwang 	 * is empy/NULL, return a -1 value.
71a9643ea8Slogwang 	 */
72a9643ea8Slogwang 	if (ah->dport == NULL || ah->sport == NULL || ah->lnk == NULL ||
73a9643ea8Slogwang 		ah->maxpktsize == 0)
74a9643ea8Slogwang 		return (-1);
75a9643ea8Slogwang 	/*
76a9643ea8Slogwang 	 * Fingerprint the incoming packet, if it matches any conditions
77a9643ea8Slogwang 	 * return an OK value.
78a9643ea8Slogwang 	 */
79a9643ea8Slogwang 	if (ntohs(*ah->dport) == 123
80a9643ea8Slogwang 	    || ntohs(*ah->sport) == 456)
81a9643ea8Slogwang 		return (0); /* I know how to handle it. */
82a9643ea8Slogwang 	return (-1); /* I don't recognize this packet. */
83a9643ea8Slogwang }
84a9643ea8Slogwang 
85a9643ea8Slogwang /*
86a9643ea8Slogwang  * Wrap in this general purpose function, the real function used to alias the
87a9643ea8Slogwang  * packets.
88a9643ea8Slogwang  */
89a9643ea8Slogwang 
90a9643ea8Slogwang static int
protohandler(struct libalias * la,struct ip * pip,struct alias_data * ah)91a9643ea8Slogwang protohandler(struct libalias *la, struct ip *pip, struct alias_data *ah)
92a9643ea8Slogwang {
93a9643ea8Slogwang 
94a9643ea8Slogwang 	AliasHandleDummy(la, pip, ah);
95a9643ea8Slogwang 	return (0);
96a9643ea8Slogwang }
97a9643ea8Slogwang 
98a9643ea8Slogwang /*
99a9643ea8Slogwang  * NOTA BENE: the next variable MUST NOT be renamed in any case if you want
100a9643ea8Slogwang  * your module to work in userland, cause it's used to find and use all
101a9643ea8Slogwang  * the protocol handlers present in every module.
102a9643ea8Slogwang  * So WATCH OUT, your module needs this variables and it needs it with
103a9643ea8Slogwang  * ITS EXACT NAME: handlers.
104a9643ea8Slogwang  */
105a9643ea8Slogwang 
106a9643ea8Slogwang struct proto_handler handlers [] = {
107a9643ea8Slogwang 	{
108a9643ea8Slogwang 	  .pri = 666,
109a9643ea8Slogwang 	  .dir = IN|OUT,
110a9643ea8Slogwang 	  .proto = UDP|TCP,
111a9643ea8Slogwang 	  .fingerprint = &fingerprint,
112a9643ea8Slogwang 	  .protohandler = &protohandler
113a9643ea8Slogwang 	},
114a9643ea8Slogwang 	{ EOH }
115a9643ea8Slogwang };
116a9643ea8Slogwang 
117a9643ea8Slogwang static int
mod_handler(module_t mod,int type,void * data)118a9643ea8Slogwang mod_handler(module_t mod, int type, void *data)
119a9643ea8Slogwang {
120a9643ea8Slogwang 	int error;
121a9643ea8Slogwang 
122a9643ea8Slogwang 	switch (type) {
123a9643ea8Slogwang 	case MOD_LOAD:
124a9643ea8Slogwang 		error = 0;
125a9643ea8Slogwang 		LibAliasAttachHandlers(handlers);
126a9643ea8Slogwang 		break;
127a9643ea8Slogwang 	case MOD_UNLOAD:
128a9643ea8Slogwang 		error = 0;
129a9643ea8Slogwang 		LibAliasDetachHandlers(handlers);
130a9643ea8Slogwang 		break;
131a9643ea8Slogwang 	default:
132a9643ea8Slogwang 		error = EINVAL;
133a9643ea8Slogwang 	}
134a9643ea8Slogwang 	return (error);
135a9643ea8Slogwang }
136a9643ea8Slogwang 
137a9643ea8Slogwang #ifdef _KERNEL
138a9643ea8Slogwang static
139a9643ea8Slogwang #endif
140a9643ea8Slogwang moduledata_t alias_mod = {
141a9643ea8Slogwang        "alias_dummy", mod_handler, NULL
142a9643ea8Slogwang };
143a9643ea8Slogwang 
144a9643ea8Slogwang #ifdef	_KERNEL
145a9643ea8Slogwang DECLARE_MODULE(alias_dummy, alias_mod, SI_SUB_DRIVERS, SI_ORDER_SECOND);
146a9643ea8Slogwang MODULE_VERSION(alias_dummy, 1);
147a9643ea8Slogwang MODULE_DEPEND(alias_dummy, libalias, 1, 1, 1);
148a9643ea8Slogwang #endif
149a9643ea8Slogwang 
150a9643ea8Slogwang static void
AliasHandleDummy(struct libalias * la,struct ip * ip,struct alias_data * ah)151a9643ea8Slogwang AliasHandleDummy(struct libalias *la, struct ip *ip, struct alias_data *ah)
152a9643ea8Slogwang {
153a9643ea8Slogwang 	; /* Dummy. */
154a9643ea8Slogwang }
155