1 /*-
2 * SPDX-License-Identifier: BSD-2-Clause
3 *
4 * Copyright (c) 2020 Ryan Moeller <freqlabs@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 AUTHORS 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 AUTHORS 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/cdefs.h>
29 #include <sys/dnv.h>
30 #include <sys/nv.h>
31 #include <netinet/in.h>
32
33 #include <assert.h>
34 #include <errno.h>
35 #include <netdb.h>
36 #include <stdlib.h>
37 #include <string.h>
38 #include <unistd.h>
39
40 #include <libcasper.h>
41 #include <libcasper_service.h>
42
43 #include "cap_netdb.h"
44
45 static struct protoent *
protoent_unpack(nvlist_t * nvl)46 protoent_unpack(nvlist_t *nvl)
47 {
48 struct protoent *pp;
49 char **aliases;
50 size_t n;
51
52 pp = malloc(sizeof(*pp));
53 if (pp == NULL) {
54 nvlist_destroy(nvl);
55 return (NULL);
56 }
57
58 pp->p_name = nvlist_take_string(nvl, "name");
59
60 aliases = nvlist_take_string_array(nvl, "aliases", &n);
61 pp->p_aliases = realloc(aliases, sizeof(char *) * (n + 1));
62 if (pp->p_aliases == NULL) {
63 while (n-- > 0)
64 free(aliases[n]);
65 free(aliases);
66 free(pp->p_name);
67 free(pp);
68 nvlist_destroy(nvl);
69 return (NULL);
70 }
71 pp->p_aliases[n] = NULL;
72
73 pp->p_proto = (int)nvlist_take_number(nvl, "proto");
74
75 nvlist_destroy(nvl);
76 return (pp);
77 }
78
79 struct protoent *
cap_getprotobyname(cap_channel_t * chan,const char * name)80 cap_getprotobyname(cap_channel_t *chan, const char *name)
81 {
82 nvlist_t *nvl;
83
84 nvl = nvlist_create(0);
85 nvlist_add_string(nvl, "cmd", "getprotobyname");
86 nvlist_add_string(nvl, "name", name);
87 nvl = cap_xfer_nvlist(chan, nvl);
88 if (nvl == NULL)
89 return (NULL);
90 if (dnvlist_get_number(nvl, "error", 0) != 0) {
91 nvlist_destroy(nvl);
92 return (NULL);
93 }
94 return (protoent_unpack(nvl));
95 }
96
97 static void
protoent_pack(const struct protoent * pp,nvlist_t * nvl)98 protoent_pack(const struct protoent *pp, nvlist_t *nvl)
99 {
100 int n = 0;
101
102 nvlist_add_string(nvl, "name", pp->p_name);
103
104 while (pp->p_aliases[n] != NULL)
105 ++n;
106 nvlist_add_string_array(nvl, "aliases",
107 (const char * const *)pp->p_aliases, n);
108
109 nvlist_add_number(nvl, "proto", (uint64_t)pp->p_proto);
110 }
111
112 static int
netdb_getprotobyname(const nvlist_t * limits __unused,const nvlist_t * nvlin,nvlist_t * nvlout)113 netdb_getprotobyname(const nvlist_t *limits __unused, const nvlist_t *nvlin,
114 nvlist_t *nvlout)
115 {
116 const char *name;
117 struct protoent *pp;
118
119 name = dnvlist_get_string(nvlin, "name", NULL);
120 if (name == NULL)
121 return (EDOOFUS);
122
123 pp = getprotobyname(name);
124 if (pp == NULL)
125 return (EINVAL);
126
127 protoent_pack(pp, nvlout);
128 return (0);
129 }
130
131 static int
netdb_limit(const nvlist_t * oldlimits __unused,const nvlist_t * newlimits __unused)132 netdb_limit(const nvlist_t *oldlimits __unused,
133 const nvlist_t *newlimits __unused)
134 {
135
136 return (0);
137 }
138
139 static int
netdb_command(const char * cmd,const nvlist_t * limits,nvlist_t * nvlin,nvlist_t * nvlout)140 netdb_command(const char *cmd, const nvlist_t *limits, nvlist_t *nvlin,
141 nvlist_t *nvlout)
142 {
143 int error;
144
145 if (strcmp(cmd, "getprotobyname") == 0)
146 error = netdb_getprotobyname(limits, nvlin, nvlout);
147 else
148 error = NO_RECOVERY;
149
150 return (error);
151 }
152
153 CREATE_SERVICE("system.netdb", netdb_limit, netdb_command, 0);
154