chiark / gitweb /
sd-rtnl: socket_read - use a read buffer
[elogind.git] / src / libsystemd / sd-rtnl / rtnl-internal.h
1 /*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
2
3 #pragma once
4
5 /***
6   This file is part of systemd.
7
8   Copyright 2013 Tom Gundersen <teg@jklm.no>
9
10   systemd is free software; you can redistribute it and/or modify it
11   under the terms of the GNU Lesser General Public License as published by
12   the Free Software Foundation; either version 2.1 of the License, or
13   (at your option) any later version.
14
15   systemd is distributed in the hope that it will be useful, but
16   WITHOUT ANY WARRANTY; without even the implied warranty of
17   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18   Lesser General Public License for more details.
19
20   You should have received a copy of the GNU Lesser General Public License
21   along with systemd; If not, see <http://www.gnu.org/licenses/>.
22 ***/
23
24 #include <linux/netlink.h>
25
26 #include "refcnt.h"
27 #include "prioq.h"
28 #include "list.h"
29
30 #include "sd-rtnl.h"
31
32 #include "rtnl-types.h"
33
34 #define RTNL_DEFAULT_TIMEOUT ((usec_t) (25 * USEC_PER_SEC))
35
36 #define RTNL_WQUEUE_MAX 1024
37 #define RTNL_RQUEUE_MAX 64*1024
38
39 #define RTNL_CONTAINER_DEPTH 32
40
41 struct reply_callback {
42         sd_rtnl_message_handler_t callback;
43         void *userdata;
44         usec_t timeout;
45         uint64_t serial;
46         unsigned prioq_idx;
47 };
48
49 struct match_callback {
50         sd_rtnl_message_handler_t callback;
51         uint16_t type;
52         void *userdata;
53
54         LIST_FIELDS(struct match_callback, match_callbacks);
55 };
56
57 struct sd_rtnl {
58         RefCount n_ref;
59
60         int fd;
61
62         union {
63                 struct sockaddr sa;
64                 struct sockaddr_nl nl;
65         } sockaddr;
66
67         sd_rtnl_message **rqueue;
68         unsigned rqueue_size;
69         size_t rqueue_allocated;
70
71         sd_rtnl_message **wqueue;
72         unsigned wqueue_size;
73         size_t wqueue_allocated;
74
75         struct nlmsghdr *rbuffer;
76         size_t rbuffer_allocated;
77
78         bool processing:1;
79
80         uint32_t serial;
81
82         struct Prioq *reply_callbacks_prioq;
83         Hashmap *reply_callbacks;
84
85         LIST_HEAD(struct match_callback, match_callbacks);
86
87         pid_t original_pid;
88
89         sd_event_source *io_event_source;
90         sd_event_source *time_event_source;
91         sd_event_source *exit_event_source;
92         sd_event *event;
93 };
94
95 struct sd_rtnl_message {
96         RefCount n_ref;
97
98         sd_rtnl *rtnl;
99
100         struct nlmsghdr *hdr;
101         const struct NLTypeSystem *(container_type_system[RTNL_CONTAINER_DEPTH]); /* the type of the container and all its parents */
102         size_t container_offsets[RTNL_CONTAINER_DEPTH]; /* offset from hdr to each container's start */
103         unsigned n_containers; /* number of containers */
104         size_t next_rta_offset; /* offset from hdr to next rta */
105         size_t *rta_offset_tb[RTNL_CONTAINER_DEPTH];
106         unsigned short rta_tb_size[RTNL_CONTAINER_DEPTH];
107         bool sealed:1;
108
109         sd_rtnl_message *next; /* next in a chain of multi-part messages */
110 };
111
112 int message_new(sd_rtnl *rtnl, sd_rtnl_message **ret, uint16_t type);
113
114 int socket_write_message(sd_rtnl *nl, sd_rtnl_message *m);
115 int socket_read_message(sd_rtnl *nl);
116
117 int rtnl_rqueue_make_room(sd_rtnl *rtnl);
118
119 int rtnl_message_read_internal(sd_rtnl_message *m, unsigned short type, void **data);
120 int rtnl_message_parse(sd_rtnl_message *m,
121                        size_t **rta_offset_tb,
122                        unsigned short *rta_tb_size,
123                        int max,
124                        struct rtattr *rta,
125                        unsigned int rt_len);
126
127 /* Make sure callbacks don't destroy the rtnl connection */
128 #define RTNL_DONT_DESTROY(rtnl) \
129         _cleanup_rtnl_unref_ _unused_ sd_rtnl *_dont_destroy_##rtnl = sd_rtnl_ref(rtnl)