xref: /freebsd/tests/sys/netgraph/socket.c (revision a8089ea5aee578e08acab2438e82fc9a9ae50ed8)
1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause
3  *
4  * Copyright (c) 2023 Gleb Smirnoff <glebius@FreeBSD.org>
5  *
6  * Redistribution and use in source and binary forms, with or without
7  * modification, are permitted provided that the following conditions
8  * are met:
9  * 1. Redistributions of source code must retain the above copyright
10  *    notice, this list of conditions and the following disclaimer.
11  * 2. Redistributions in binary form must reproduce the above copyright
12  *    notice, this list of conditions and the following disclaimer in the
13  *    documentation and/or other materials provided with the distribution.
14  *
15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
16  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
17  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
18  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
19  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
20  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
21  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
22  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
23  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
24  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
25  * SUCH DAMAGE.
26  */
27 
28 #include <sys/socket.h>
29 #include <netgraph.h>
30 #include <netgraph/ng_socket.h>
31 
32 #include <errno.h>
33 
34 #include <atf-c.h>
35 
36 ATF_TC_WITHOUT_HEAD(getsockname);
37 ATF_TC_BODY(getsockname, tc)
38 {
39 	struct sockaddr_ng sg;
40 	socklen_t len = sizeof(struct sockaddr_ng);
41 #define	NAME	"0123456789012345678901234567891"	/* 31 chars */
42 	char name[NG_NODESIZ] = NAME;
43 	int cs;
44 
45 	/* Unnamed node returns its ID as name. */
46 	ATF_REQUIRE(NgMkSockNode(NULL, &cs, NULL) == 0);
47 	ATF_REQUIRE(getsockname(cs, (struct sockaddr *)&sg, &len) == 0);
48 	ATF_REQUIRE(strspn(sg.sg_data, "[0123456789abcdef]") >= 3 &&
49 	    sg.sg_data[strspn(sg.sg_data, "[0123456789abcdef]")] == '\0');
50 	close(cs);
51 
52 	/* Named node. */
53 	ATF_REQUIRE(NgMkSockNode(name, &cs, NULL) == 0);
54 	ATF_REQUIRE(getsockname(cs, (struct sockaddr *)&sg, &len) == 0);
55 	ATF_REQUIRE(strcmp(sg.sg_data, NAME) == 0);
56 }
57 
58 ATF_TP_ADD_TCS(tp)
59 {
60 	ATF_TP_ADD_TC(tp, getsockname);
61 
62 	return (atf_no_error());
63 }
64