]> git.ipfire.org Git - thirdparty/systemd.git/blame - src/libsystemd/sd-netlink/netlink-message.c
sd-netlink: make a couple of helper functions static
[thirdparty/systemd.git] / src / libsystemd / sd-netlink / netlink-message.c
CommitLineData
65f568bb
TG
1/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
2
3/***
4 This file is part of systemd.
5
6 Copyright 2013 Tom Gundersen <teg@jklm.no>
7
8 systemd is free software; you can redistribute it and/or modify it
9 under the terms of the GNU Lesser General Public License as published by
10 the Free Software Foundation; either version 2.1 of the License, or
11 (at your option) any later version.
12
13 systemd is distributed in the hope that it will be useful, but
14 WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 Lesser General Public License for more details.
17
18 You should have received a copy of the GNU Lesser General Public License
19 along with systemd; If not, see <http://www.gnu.org/licenses/>.
20***/
21
22#include <netinet/in.h>
65f568bb
TG
23#include <stdbool.h>
24#include <unistd.h>
25
26#include "util.h"
2ce84835 27#include "socket-util.h"
6482f626 28#include "formats-util.h"
65f568bb 29#include "refcnt.h"
d5eff740 30#include "missing.h"
65f568bb 31
1c4baffc
TG
32#include "sd-netlink.h"
33#include "netlink-util.h"
34#include "netlink-internal.h"
35#include "netlink-types.h"
65f568bb 36
31a4e153 37#define GET_CONTAINER(m, i) ((i) < (m)->n_containers ? (struct rtattr*)((uint8_t*)(m)->hdr + (m)->container_offsets[i]) : NULL)
e5c4350b 38#define PUSH_CONTAINER(m, new) (m)->container_offsets[(m)->n_containers ++] = (uint8_t*)(new) - (uint8_t*)(m)->hdr;
4ebe732c 39
0a2478a9
TG
40#define RTA_TYPE(rta) ((rta)->rta_type & NLA_TYPE_MASK)
41
89489ef7 42int message_new_empty(sd_netlink *rtnl, sd_netlink_message **ret) {
1c4baffc 43 sd_netlink_message *m;
65f568bb
TG
44
45 assert_return(ret, -EINVAL);
65f568bb 46
f131770b 47 /* Note that 'rtnl' is currently unused, if we start using it internally
8c578303 48 we must take care to avoid problems due to mutual references between
ff9b60f3 49 buses and their queued messages. See sd-bus.
8c578303
TG
50 */
51
1c4baffc 52 m = new0(sd_netlink_message, 1);
65f568bb
TG
53 if (!m)
54 return -ENOMEM;
55
65f568bb
TG
56 m->n_ref = REFCNT_INIT;
57
65f568bb
TG
58 m->sealed = false;
59
60 *ret = m;
61
62 return 0;
63}
64
1c4baffc
TG
65int message_new(sd_netlink *rtnl, sd_netlink_message **ret, uint16_t type) {
66 _cleanup_netlink_message_unref_ sd_netlink_message *m = NULL;
d8e538ec
TG
67 const NLType *nl_type;
68 size_t size;
69 int r;
70
846a6b3d 71 r = type_system_get_type(&type_system_root, &nl_type, type);
d8e538ec
TG
72 if (r < 0)
73 return r;
74
12b7dff4
DH
75 if (type_get_type(nl_type) != NETLINK_TYPE_NESTED)
76 return -EINVAL;
77
1b89cf56 78 r = message_new_empty(rtnl, &m);
d8e538ec
TG
79 if (r < 0)
80 return r;
81
817d1cd8 82 size = NLMSG_SPACE(type_get_size(nl_type));
1b89cf56
TG
83
84 assert(size >= sizeof(struct nlmsghdr));
85 m->hdr = malloc0(size);
86 if (!m->hdr)
87 return -ENOMEM;
88
89 m->hdr->nlmsg_flags = NLM_F_REQUEST | NLM_F_ACK;
90
12b7dff4 91 type_get_type_system(nl_type, &m->container_type_system[0]);
1b89cf56
TG
92 m->hdr->nlmsg_len = size;
93 m->hdr->nlmsg_type = type;
94
95 *ret = m;
96 m = NULL;
d8e538ec
TG
97
98 return 0;
99}
100
1c4baffc 101int sd_netlink_message_request_dump(sd_netlink_message *m, int dump) {
6e20c8f8
TG
102 assert_return(m, -EINVAL);
103 assert_return(m->hdr, -EINVAL);
20dff6c4
TG
104 assert_return(m->hdr->nlmsg_type == RTM_GETLINK ||
105 m->hdr->nlmsg_type == RTM_GETADDR ||
106 m->hdr->nlmsg_type == RTM_GETROUTE ||
107 m->hdr->nlmsg_type == RTM_GETNEIGH,
6e20c8f8
TG
108 -EINVAL);
109
110 if (dump)
111 m->hdr->nlmsg_flags |= NLM_F_DUMP;
112 else
113 m->hdr->nlmsg_flags &= ~NLM_F_DUMP;
114
115 return 0;
116}
117
1c4baffc 118sd_netlink_message *sd_netlink_message_ref(sd_netlink_message *m) {
65f568bb
TG
119 if (m)
120 assert_se(REFCNT_INC(m->n_ref) >= 2);
121
122 return m;
123}
124
1c4baffc 125sd_netlink_message *sd_netlink_message_unref(sd_netlink_message *m) {
f0c4b1c3 126 if (m && REFCNT_DEC(m->n_ref) == 0) {
3dd215e0
TG
127 unsigned i;
128
65f568bb 129 free(m->hdr);
3dd215e0 130
9f5bbfe3 131 for (i = 0; i <= m->n_containers; i++)
3dd215e0
TG
132 free(m->rta_offset_tb[i]);
133
1c4baffc 134 sd_netlink_message_unref(m->next);
1403f45a 135
65f568bb
TG
136 free(m);
137 }
138
139 return NULL;
140}
141
1c4baffc 142int sd_netlink_message_get_type(sd_netlink_message *m, uint16_t *type) {
65f568bb
TG
143 assert_return(m, -EINVAL);
144 assert_return(type, -EINVAL);
145
146 *type = m->hdr->nlmsg_type;
147
148 return 0;
149}
150
1c4baffc 151int sd_netlink_message_is_broadcast(sd_netlink_message *m) {
1f0db3ed
TG
152 assert_return(m, -EINVAL);
153
3f42446d 154 return m->broadcast;
1f0db3ed
TG
155}
156
4ebe732c
ZJS
157/* If successful the updated message will be correctly aligned, if
158 unsuccessful the old message is untouched. */
1c4baffc 159static int add_rtattr(sd_netlink_message *m, unsigned short type, const void *data, size_t data_length) {
7ca1d319
TG
160 uint32_t rta_length;
161 size_t message_length, padding_length;
65f568bb
TG
162 struct nlmsghdr *new_hdr;
163 struct rtattr *rta;
8e337e64 164 char *padding;
5a081409 165 unsigned i;
7ca1d319 166 int offset;
65f568bb 167
33125ac5
TG
168 assert(m);
169 assert(m->hdr);
e5c4350b 170 assert(!m->sealed);
33125ac5 171 assert(NLMSG_ALIGN(m->hdr->nlmsg_len) == m->hdr->nlmsg_len);
7ca1d319
TG
172 assert(!data || data_length);
173
174 /* get offset of the new attribute */
175 offset = m->hdr->nlmsg_len;
65f568bb 176
8e337e64 177 /* get the size of the new rta attribute (with padding at the end) */
65f568bb 178 rta_length = RTA_LENGTH(data_length);
4ebe732c
ZJS
179
180 /* get the new message size (with padding at the end) */
7ca1d319 181 message_length = offset + RTA_ALIGN(rta_length);
65f568bb
TG
182
183 /* realloc to fit the new attribute */
184 new_hdr = realloc(m->hdr, message_length);
185 if (!new_hdr)
186 return -ENOMEM;
187 m->hdr = new_hdr;
188
189 /* get pointer to the attribute we are about to add */
7ca1d319 190 rta = (struct rtattr *) ((uint8_t *) m->hdr + offset);
65f568bb 191
5a081409
TG
192 /* if we are inside containers, extend them */
193 for (i = 0; i < m->n_containers; i++)
7ca1d319 194 GET_CONTAINER(m, i)->rta_len += message_length - offset;
33125ac5 195
65f568bb
TG
196 /* fill in the attribute */
197 rta->rta_type = type;
198 rta->rta_len = rta_length;
7ca1d319 199 if (data)
33125ac5
TG
200 /* we don't deal with the case where the user lies about the type
201 * and gives us too little data (so don't do that)
7ca1d319 202 */
33125ac5 203 padding = mempcpy(RTA_DATA(rta), data, data_length);
7ca1d319
TG
204 else {
205 /* if no data was passed, make sure we still initialize the padding
206 note that we can have data_length > 0 (used by some containers) */
207 padding = RTA_DATA(rta);
33125ac5 208 }
65f568bb 209
7ca1d319
TG
210 /* make sure also the padding at the end of the message is initialized */
211 padding_length = (uint8_t*)m->hdr + message_length - (uint8_t*)padding;
212 memzero(padding, padding_length);
213
4ebe732c
ZJS
214 /* update message size */
215 m->hdr->nlmsg_len = message_length;
216
7ca1d319 217 return offset;
65f568bb
TG
218}
219
6c14ad61 220static int message_attribute_has_type(sd_netlink_message *m, size_t *out_size, uint16_t attribute_type, uint16_t data_type) {
d8e538ec
TG
221 const NLType *type;
222 int r;
223
6c14ad61
DH
224 assert(m);
225
d8e538ec
TG
226 r = type_system_get_type(m->container_type_system[m->n_containers], &type, attribute_type);
227 if (r < 0)
228 return r;
229
817d1cd8 230 if (type_get_type(type) != data_type)
d8e538ec
TG
231 return -EINVAL;
232
6c14ad61
DH
233 if (out_size)
234 *out_size = type_get_size(type);
235 return 0;
d8e538ec
TG
236}
237
1c4baffc 238int sd_netlink_message_append_string(sd_netlink_message *m, unsigned short type, const char *data) {
d8e538ec 239 size_t length, size;
0a0dc69b 240 int r;
65f568bb
TG
241
242 assert_return(m, -EINVAL);
e5c4350b 243 assert_return(!m->sealed, -EPERM);
65f568bb
TG
244 assert_return(data, -EINVAL);
245
6c14ad61 246 r = message_attribute_has_type(m, &size, type, NETLINK_TYPE_STRING);
0a0dc69b
TG
247 if (r < 0)
248 return r;
65f568bb 249
d8e538ec 250 if (size) {
3072eecf
LP
251 length = strnlen(data, size+1);
252 if (length > size)
d8e538ec
TG
253 return -EINVAL;
254 } else
255 length = strlen(data);
33125ac5 256
d8e538ec 257 r = add_rtattr(m, type, data, length + 1);
0a0dc69b
TG
258 if (r < 0)
259 return r;
260
261 return 0;
262}
263
1c4baffc 264int sd_netlink_message_append_u8(sd_netlink_message *m, unsigned short type, uint8_t data) {
7b179640
SS
265 int r;
266
267 assert_return(m, -EINVAL);
268 assert_return(!m->sealed, -EPERM);
269
6c14ad61 270 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_U8);
7b179640
SS
271 if (r < 0)
272 return r;
273
7b179640
SS
274 r = add_rtattr(m, type, &data, sizeof(uint8_t));
275 if (r < 0)
276 return r;
277
278 return 0;
279}
280
281
1c4baffc 282int sd_netlink_message_append_u16(sd_netlink_message *m, unsigned short type, uint16_t data) {
01b36069
TG
283 int r;
284
285 assert_return(m, -EINVAL);
e5c4350b 286 assert_return(!m->sealed, -EPERM);
01b36069 287
6c14ad61 288 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_U16);
01b36069
TG
289 if (r < 0)
290 return r;
291
01b36069
TG
292 r = add_rtattr(m, type, &data, sizeof(uint16_t));
293 if (r < 0)
294 return r;
295
296 return 0;
297}
298
1c4baffc 299int sd_netlink_message_append_u32(sd_netlink_message *m, unsigned short type, uint32_t data) {
0a0dc69b
TG
300 int r;
301
302 assert_return(m, -EINVAL);
e5c4350b 303 assert_return(!m->sealed, -EPERM);
0a0dc69b 304
6c14ad61 305 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_U32);
0a0dc69b
TG
306 if (r < 0)
307 return r;
308
4d47756b 309 r = add_rtattr(m, type, &data, sizeof(uint32_t));
0a0dc69b
TG
310 if (r < 0)
311 return r;
312
313 return 0;
314}
315
1c4baffc 316int sd_netlink_message_append_in_addr(sd_netlink_message *m, unsigned short type, const struct in_addr *data) {
0a0dc69b
TG
317 int r;
318
319 assert_return(m, -EINVAL);
e5c4350b 320 assert_return(!m->sealed, -EPERM);
0a0dc69b
TG
321 assert_return(data, -EINVAL);
322
6c14ad61 323 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_IN_ADDR);
0a0dc69b
TG
324 if (r < 0)
325 return r;
326
4d47756b 327 r = add_rtattr(m, type, data, sizeof(struct in_addr));
0a0dc69b
TG
328 if (r < 0)
329 return r;
330
331 return 0;
332}
333
1c4baffc 334int sd_netlink_message_append_in6_addr(sd_netlink_message *m, unsigned short type, const struct in6_addr *data) {
0a0dc69b
TG
335 int r;
336
337 assert_return(m, -EINVAL);
e5c4350b 338 assert_return(!m->sealed, -EPERM);
0a0dc69b
TG
339 assert_return(data, -EINVAL);
340
6c14ad61 341 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_IN_ADDR);
0a0dc69b
TG
342 if (r < 0)
343 return r;
344
4d47756b 345 r = add_rtattr(m, type, data, sizeof(struct in6_addr));
0a0dc69b
TG
346 if (r < 0)
347 return r;
348
349 return 0;
350}
351
1c4baffc 352int sd_netlink_message_append_ether_addr(sd_netlink_message *m, unsigned short type, const struct ether_addr *data) {
0a0dc69b
TG
353 int r;
354
355 assert_return(m, -EINVAL);
e5c4350b 356 assert_return(!m->sealed, -EPERM);
0a0dc69b
TG
357 assert_return(data, -EINVAL);
358
6c14ad61 359 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_ETHER_ADDR);
d8e538ec
TG
360 if (r < 0)
361 return r;
0a0dc69b 362
b9eaf3d1 363 r = add_rtattr(m, type, data, ETH_ALEN);
0a0dc69b
TG
364 if (r < 0)
365 return r;
366
367 return 0;
65f568bb
TG
368}
369
1c4baffc 370int sd_netlink_message_append_cache_info(sd_netlink_message *m, unsigned short type, const struct ifa_cacheinfo *info) {
aba496a5
UTL
371 int r;
372
373 assert_return(m, -EINVAL);
374 assert_return(!m->sealed, -EPERM);
375 assert_return(info, -EINVAL);
376
6c14ad61 377 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_CACHE_INFO);
aba496a5
UTL
378 if (r < 0)
379 return r;
380
381 r = add_rtattr(m, type, info, sizeof(struct ifa_cacheinfo));
382 if (r < 0)
383 return r;
384
385 return 0;
386}
387
1c4baffc 388int sd_netlink_message_open_container(sd_netlink_message *m, unsigned short type) {
d8e538ec
TG
389 size_t size;
390 int r;
33125ac5 391
65f568bb 392 assert_return(m, -EINVAL);
e5c4350b 393 assert_return(!m->sealed, -EPERM);
7ca1d319 394 assert_return(m->n_containers < RTNL_CONTAINER_DEPTH, -ERANGE);
33125ac5 395
6c14ad61 396 r = message_attribute_has_type(m, &size, type, NETLINK_TYPE_NESTED);
4af7b60d
TG
397 if (r < 0) {
398 const NLTypeSystemUnion *type_system_union;
399 int family;
400
6c14ad61 401 r = message_attribute_has_type(m, &size, type, NETLINK_TYPE_UNION);
4af7b60d
TG
402 if (r < 0)
403 return r;
4af7b60d 404
89489ef7 405 r = sd_rtnl_message_get_family(m, &family);
4af7b60d
TG
406 if (r < 0)
407 return r;
408
409 r = type_system_get_type_system_union(m->container_type_system[m->n_containers], &type_system_union, type);
410 if (r < 0)
411 return r;
412
413 r = type_system_union_protocol_get_type_system(type_system_union,
414 &m->container_type_system[m->n_containers + 1],
415 family);
416 if (r < 0)
417 return r;
418 } else {
4af7b60d
TG
419 r = type_system_get_type_system(m->container_type_system[m->n_containers],
420 &m->container_type_system[m->n_containers + 1],
421 type);
422 if (r < 0)
423 return r;
424 }
31a4e153 425
fcf81a54 426 r = add_rtattr(m, type | NLA_F_NESTED, NULL, size);
d8e538ec
TG
427 if (r < 0)
428 return r;
429
7ca1d319
TG
430 m->container_offsets[m->n_containers ++] = r;
431
d8e538ec
TG
432 return 0;
433}
434
1c4baffc 435int sd_netlink_message_open_container_union(sd_netlink_message *m, unsigned short type, const char *key) {
d8e538ec
TG
436 const NLTypeSystemUnion *type_system_union;
437 int r;
438
439 assert_return(m, -EINVAL);
440 assert_return(!m->sealed, -EPERM);
441
442 r = type_system_get_type_system_union(m->container_type_system[m->n_containers], &type_system_union, type);
443 if (r < 0)
444 return r;
445
446 r = type_system_union_get_type_system(type_system_union,
447 &m->container_type_system[m->n_containers + 1],
448 key);
449 if (r < 0)
450 return r;
33125ac5 451
1c4baffc 452 r = sd_netlink_message_append_string(m, type_system_union->match, key);
d8e538ec
TG
453 if (r < 0)
454 return r;
455
456 /* do we evere need non-null size */
da041d69 457 r = add_rtattr(m, type | NLA_F_NESTED, NULL, 0);
d8e538ec
TG
458 if (r < 0)
459 return r;
460
7ca1d319
TG
461 m->container_offsets[m->n_containers ++] = r;
462
d8e538ec 463 return 0;
33125ac5
TG
464}
465
d8e538ec 466
1c4baffc 467int sd_netlink_message_close_container(sd_netlink_message *m) {
33125ac5 468 assert_return(m, -EINVAL);
e5c4350b 469 assert_return(!m->sealed, -EPERM);
5a081409 470 assert_return(m->n_containers > 0, -EINVAL);
33125ac5 471
d8e538ec 472 m->container_type_system[m->n_containers] = NULL;
5a081409 473 m->n_containers --;
33125ac5
TG
474
475 return 0;
476}
477
4203fc8b 478static int netlink_message_read_internal(sd_netlink_message *m, unsigned short type, void **data) {
f66eeb6b
TG
479 struct rtattr *rta;
480
44caa5e7
SS
481 assert_return(m, -EINVAL);
482 assert_return(m->sealed, -EPERM);
483 assert_return(data, -EINVAL);
d8e538ec
TG
484 assert(m->n_containers <= RTNL_CONTAINER_DEPTH);
485 assert(m->rta_offset_tb[m->n_containers]);
486 assert(type < m->rta_tb_size[m->n_containers]);
44caa5e7 487
3dd215e0 488 if(!m->rta_offset_tb[m->n_containers][type])
44caa5e7
SS
489 return -ENODATA;
490
3dd215e0 491 rta = (struct rtattr*)((uint8_t *) m->hdr + m->rta_offset_tb[m->n_containers][type]);
44caa5e7 492
f66eeb6b
TG
493 *data = RTA_DATA(rta);
494
495 return RTA_PAYLOAD(rta);
44caa5e7
SS
496}
497
1c4baffc 498int sd_netlink_message_read_string(sd_netlink_message *m, unsigned short type, const char **data) {
44caa5e7
SS
499 int r;
500 void *attr_data;
501
73ae2b7d
TG
502 assert_return(m, -EINVAL);
503
6c14ad61 504 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_STRING);
d8e538ec
TG
505 if (r < 0)
506 return r;
44caa5e7 507
4203fc8b 508 r = netlink_message_read_internal(m, type, &attr_data);
f66eeb6b 509 if (r < 0)
44caa5e7 510 return r;
f66eeb6b
TG
511 else if (strnlen(attr_data, r) >= (size_t) r)
512 return -EIO;
44caa5e7 513
73ae2b7d
TG
514 if (data)
515 *data = (const char *) attr_data;
44caa5e7
SS
516
517 return 0;
518}
519
1c4baffc 520int sd_netlink_message_read_u8(sd_netlink_message *m, unsigned short type, uint8_t *data) {
44caa5e7
SS
521 int r;
522 void *attr_data;
523
73ae2b7d
TG
524 assert_return(m, -EINVAL);
525
6c14ad61 526 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_U8);
d8e538ec
TG
527 if (r < 0)
528 return r;
44caa5e7 529
4203fc8b 530 r = netlink_message_read_internal(m, type, &attr_data);
f66eeb6b 531 if (r < 0)
44caa5e7 532 return r;
f66eeb6b
TG
533 else if ((size_t) r < sizeof(uint8_t))
534 return -EIO;
44caa5e7 535
73ae2b7d
TG
536 if (data)
537 *data = *(uint8_t *) attr_data;
44caa5e7
SS
538
539 return 0;
540}
541
1c4baffc 542int sd_netlink_message_read_u16(sd_netlink_message *m, unsigned short type, uint16_t *data) {
44caa5e7
SS
543 int r;
544 void *attr_data;
545
73ae2b7d
TG
546 assert_return(m, -EINVAL);
547
6c14ad61 548 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_U16);
d8e538ec
TG
549 if (r < 0)
550 return r;
44caa5e7 551
4203fc8b 552 r = netlink_message_read_internal(m, type, &attr_data);
f66eeb6b 553 if (r < 0)
44caa5e7 554 return r;
f66eeb6b
TG
555 else if ((size_t) r < sizeof(uint16_t))
556 return -EIO;
44caa5e7 557
73ae2b7d
TG
558 if (data)
559 *data = *(uint16_t *) attr_data;
44caa5e7
SS
560
561 return 0;
562}
563
1c4baffc 564int sd_netlink_message_read_u32(sd_netlink_message *m, unsigned short type, uint32_t *data) {
44caa5e7
SS
565 int r;
566 void *attr_data;
567
73ae2b7d
TG
568 assert_return(m, -EINVAL);
569
6c14ad61 570 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_U32);
d8e538ec
TG
571 if (r < 0)
572 return r;
44caa5e7 573
4203fc8b 574 r = netlink_message_read_internal(m, type, &attr_data);
f66eeb6b 575 if (r < 0)
44caa5e7 576 return r;
f66eeb6b
TG
577 else if ((size_t)r < sizeof(uint32_t))
578 return -EIO;
44caa5e7 579
73ae2b7d
TG
580 if (data)
581 *data = *(uint32_t *) attr_data;
44caa5e7
SS
582
583 return 0;
584}
585
1c4baffc 586int sd_netlink_message_read_ether_addr(sd_netlink_message *m, unsigned short type, struct ether_addr *data) {
4e9e7f18
SS
587 int r;
588 void *attr_data;
589
73ae2b7d
TG
590 assert_return(m, -EINVAL);
591
6c14ad61 592 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_ETHER_ADDR);
d8e538ec
TG
593 if (r < 0)
594 return r;
4e9e7f18 595
4203fc8b 596 r = netlink_message_read_internal(m, type, &attr_data);
f66eeb6b 597 if (r < 0)
4e9e7f18 598 return r;
f66eeb6b
TG
599 else if ((size_t)r < sizeof(struct ether_addr))
600 return -EIO;
4e9e7f18 601
73ae2b7d
TG
602 if (data)
603 memcpy(data, attr_data, sizeof(struct ether_addr));
4e9e7f18
SS
604
605 return 0;
606}
607
1c4baffc 608int sd_netlink_message_read_cache_info(sd_netlink_message *m, unsigned short type, struct ifa_cacheinfo *info) {
aba496a5
UTL
609 int r;
610 void *attr_data;
611
73ae2b7d
TG
612 assert_return(m, -EINVAL);
613
6c14ad61 614 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_CACHE_INFO);
aba496a5
UTL
615 if (r < 0)
616 return r;
617
4203fc8b 618 r = netlink_message_read_internal(m, type, &attr_data);
aba496a5
UTL
619 if (r < 0)
620 return r;
621 else if ((size_t)r < sizeof(struct ifa_cacheinfo))
622 return -EIO;
623
73ae2b7d
TG
624 if (info)
625 memcpy(info, attr_data, sizeof(struct ifa_cacheinfo));
aba496a5
UTL
626
627 return 0;
628}
629
1c4baffc 630int sd_netlink_message_read_in_addr(sd_netlink_message *m, unsigned short type, struct in_addr *data) {
4e9e7f18
SS
631 int r;
632 void *attr_data;
633
73ae2b7d
TG
634 assert_return(m, -EINVAL);
635
6c14ad61 636 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_IN_ADDR);
d8e538ec
TG
637 if (r < 0)
638 return r;
4e9e7f18 639
4203fc8b 640 r = netlink_message_read_internal(m, type, &attr_data);
f66eeb6b 641 if (r < 0)
4e9e7f18 642 return r;
f66eeb6b
TG
643 else if ((size_t)r < sizeof(struct in_addr))
644 return -EIO;
4e9e7f18 645
73ae2b7d
TG
646 if (data)
647 memcpy(data, attr_data, sizeof(struct in_addr));
4e9e7f18
SS
648
649 return 0;
650}
651
1c4baffc 652int sd_netlink_message_read_in6_addr(sd_netlink_message *m, unsigned short type, struct in6_addr *data) {
4e9e7f18
SS
653 int r;
654 void *attr_data;
655
73ae2b7d
TG
656 assert_return(m, -EINVAL);
657
6c14ad61 658 r = message_attribute_has_type(m, NULL, type, NETLINK_TYPE_IN_ADDR);
d8e538ec
TG
659 if (r < 0)
660 return r;
4e9e7f18 661
4203fc8b 662 r = netlink_message_read_internal(m, type, &attr_data);
3dd215e0 663 if (r < 0)
4e9e7f18 664 return r;
3dd215e0
TG
665 else if ((size_t)r < sizeof(struct in6_addr))
666 return -EIO;
4e9e7f18 667
73ae2b7d
TG
668 if (data)
669 memcpy(data, attr_data, sizeof(struct in6_addr));
4e9e7f18
SS
670
671 return 0;
672}
673
4203fc8b
TG
674static int netlink_message_parse(sd_netlink_message *m,
675 size_t **rta_offset_tb,
676 unsigned short *rta_tb_size,
677 int count,
678 struct rtattr *rta,
679 unsigned int rt_len) {
680 unsigned short type;
681 size_t *tb;
682
683 tb = new0(size_t, count);
684 if(!tb)
685 return -ENOMEM;
686
687 *rta_tb_size = count;
688
689 for (; RTA_OK(rta, rt_len); rta = RTA_NEXT(rta, rt_len)) {
690 type = RTA_TYPE(rta);
691
692 /* if the kernel is newer than the headers we used
693 when building, we ignore out-of-range attributes
694 */
695 if (type >= count)
696 continue;
697
698 if (tb[type])
699 log_debug("rtnl: message parse - overwriting repeated attribute");
700
701 tb[type] = (uint8_t *) rta - (uint8_t *) m->hdr;
702 }
703
704 *rta_offset_tb = tb;
705
706 return 0;
707}
708
817d1cd8 709int sd_netlink_message_enter_container(sd_netlink_message *m, unsigned short type_id) {
d8e538ec
TG
710 const NLType *nl_type;
711 const NLTypeSystem *type_system;
3dd215e0 712 void *container;
817d1cd8 713 uint16_t type;
d8e538ec
TG
714 size_t size;
715 int r;
3dd215e0
TG
716
717 assert_return(m, -EINVAL);
718 assert_return(m->n_containers < RTNL_CONTAINER_DEPTH, -EINVAL);
719
d8e538ec
TG
720 r = type_system_get_type(m->container_type_system[m->n_containers],
721 &nl_type,
817d1cd8 722 type_id);
3dd215e0
TG
723 if (r < 0)
724 return r;
3dd215e0 725
817d1cd8
DH
726 type = type_get_type(nl_type);
727
728 if (type == NETLINK_TYPE_NESTED) {
d8e538ec
TG
729 r = type_system_get_type_system(m->container_type_system[m->n_containers],
730 &type_system,
817d1cd8 731 type_id);
d8e538ec
TG
732 if (r < 0)
733 return r;
817d1cd8 734 } else if (type == NETLINK_TYPE_UNION) {
d8e538ec 735 const NLTypeSystemUnion *type_system_union;
d8e538ec
TG
736
737 r = type_system_get_type_system_union(m->container_type_system[m->n_containers],
738 &type_system_union,
817d1cd8 739 type_id);
d8e538ec
TG
740 if (r < 0)
741 return r;
742
4af7b60d
TG
743 switch (type_system_union->match_type) {
744 case NL_MATCH_SIBLING:
745 {
746 const char *key;
747
1c4baffc 748 r = sd_netlink_message_read_string(m, type_system_union->match, &key);
4af7b60d
TG
749 if (r < 0)
750 return r;
751
752 r = type_system_union_get_type_system(type_system_union,
753 &type_system,
754 key);
755 if (r < 0)
756 return r;
757
758 break;
759 }
760 case NL_MATCH_PROTOCOL:
761 {
762 int family;
763
89489ef7 764 r = sd_rtnl_message_get_family(m, &family);
4af7b60d
TG
765 if (r < 0)
766 return r;
767
768 r = type_system_union_protocol_get_type_system(type_system_union,
769 &type_system,
770 family);
771 if (r < 0)
772 return r;
773
774 break;
775 }
776 default:
1c4baffc 777 assert_not_reached("sd-netlink: invalid type system union type");
4af7b60d 778 }
d8e538ec
TG
779 } else
780 return -EINVAL;
781
4203fc8b 782 r = netlink_message_read_internal(m, type_id, &container);
3dd215e0
TG
783 if (r < 0)
784 return r;
d8e538ec
TG
785 else
786 size = (size_t)r;
3dd215e0 787
d8e538ec 788 m->n_containers ++;
3dd215e0 789
4203fc8b
TG
790 r = netlink_message_parse(m,
791 &m->rta_offset_tb[m->n_containers],
792 &m->rta_tb_size[m->n_containers],
793 type_system_get_count(type_system),
794 container,
795 size);
d8e538ec
TG
796 if (r < 0) {
797 m->n_containers --;
3dd215e0 798 return r;
d8e538ec 799 }
3dd215e0 800
d8e538ec 801 m->container_type_system[m->n_containers] = type_system;
3dd215e0
TG
802
803 return 0;
804}
805
1c4baffc 806int sd_netlink_message_exit_container(sd_netlink_message *m) {
e5c4350b
TG
807 assert_return(m, -EINVAL);
808 assert_return(m->sealed, -EINVAL);
809 assert_return(m->n_containers > 0, -EINVAL);
810
3dd215e0
TG
811 free(m->rta_offset_tb[m->n_containers]);
812 m->rta_offset_tb[m->n_containers] = NULL;
d8e538ec 813 m->container_type_system[m->n_containers] = NULL;
3dd215e0 814
e5c4350b
TG
815 m->n_containers --;
816
817 return 0;
818}
819
1c4baffc 820uint32_t rtnl_message_get_serial(sd_netlink_message *m) {
65f568bb 821 assert(m);
9d0db178 822 assert(m->hdr);
65f568bb
TG
823
824 return m->hdr->nlmsg_seq;
825}
826
1c4baffc 827int sd_netlink_message_is_error(sd_netlink_message *m) {
45af44d4
TG
828 assert_return(m, 0);
829 assert_return(m->hdr, 0);
830
831 return m->hdr->nlmsg_type == NLMSG_ERROR;
832}
833
1c4baffc 834int sd_netlink_message_get_errno(sd_netlink_message *m) {
65f568bb
TG
835 struct nlmsgerr *err;
836
e16bcf98 837 assert_return(m, -EINVAL);
9d0db178 838 assert_return(m->hdr, -EINVAL);
65f568bb 839
1c4baffc 840 if (!sd_netlink_message_is_error(m))
65f568bb
TG
841 return 0;
842
843 err = NLMSG_DATA(m->hdr);
844
845 return err->error;
846}
847
1c4baffc 848int sd_netlink_message_rewind(sd_netlink_message *m) {
817d1cd8
DH
849 const NLType *nl_type;
850 uint16_t type;
851 size_t size;
3dd215e0
TG
852 unsigned i;
853 int r;
0fc7531b
TG
854
855 assert_return(m, -EINVAL);
0fc7531b 856
3dd215e0
TG
857 /* don't allow appending to message once parsed */
858 if (!m->sealed)
859 rtnl_message_seal(m);
860
861 for (i = 1; i <= m->n_containers; i++) {
862 free(m->rta_offset_tb[i]);
863 m->rta_offset_tb[i] = NULL;
864 m->rta_tb_size[i] = 0;
d8e538ec 865 m->container_type_system[i] = NULL;
3dd215e0
TG
866 }
867
868 m->n_containers = 0;
869
870 if (m->rta_offset_tb[0]) {
871 /* top-level attributes have already been parsed */
872 return 0;
873 }
874
d8e538ec
TG
875 assert(m->hdr);
876
846a6b3d 877 r = type_system_get_type(&type_system_root, &nl_type, m->hdr->nlmsg_type);
d8e538ec
TG
878 if (r < 0)
879 return r;
880
817d1cd8
DH
881 type = type_get_type(nl_type);
882 size = type_get_size(nl_type);
883
884 if (type == NETLINK_TYPE_NESTED) {
c658008f 885 const NLTypeSystem *type_system;
d8e538ec 886
817d1cd8 887 type_get_type_system(nl_type, &type_system);
d8e538ec
TG
888
889 m->container_type_system[0] = type_system;
890
4203fc8b
TG
891 r = netlink_message_parse(m,
892 &m->rta_offset_tb[m->n_containers],
893 &m->rta_tb_size[m->n_containers],
894 type_system_get_count(type_system),
895 (struct rtattr*)((uint8_t*)NLMSG_DATA(m->hdr) + NLMSG_ALIGN(size)),
896 NLMSG_PAYLOAD(m->hdr, size));
d8e538ec
TG
897 if (r < 0)
898 return r;
0fc7531b
TG
899 }
900
901 return 0;
902}
3dd215e0 903
1c4baffc 904void rtnl_message_seal(sd_netlink_message *m) {
3dd215e0
TG
905 assert(m);
906 assert(!m->sealed);
907
908 m->sealed = true;
909}
1403f45a 910
1c4baffc 911sd_netlink_message *sd_netlink_message_next(sd_netlink_message *m) {
1403f45a
TG
912 assert_return(m, NULL);
913
914 return m->next;
915}