1 /* SPDX-License-Identifier: LGPL-2.1+ */
3 This file is part of systemd.
5 Copyright 2010 Lennart Poettering
13 #include <netinet/ip.h>
22 #include "alloc-util.h"
25 #include "format-util.h"
29 #include "parse-util.h"
30 #include "path-util.h"
31 #include "process-util.h"
32 #include "socket-util.h"
33 #include "string-table.h"
34 #include "string-util.h"
36 #include "user-util.h"
41 # define IDN_FLAGS NI_IDN
46 static const char* const socket_address_type_table
[] = {
47 [SOCK_STREAM
] = "Stream",
48 [SOCK_DGRAM
] = "Datagram",
50 [SOCK_RDM
] = "ReliableDatagram",
51 [SOCK_SEQPACKET
] = "SequentialPacket",
52 [SOCK_DCCP
] = "DatagramCongestionControl",
55 DEFINE_STRING_TABLE_LOOKUP(socket_address_type
, int);
57 int socket_address_parse(SocketAddress
*a
, const char *s
) {
65 a
->type
= SOCK_STREAM
;
70 /* IPv6 in [x:.....:z]:p notation */
76 n
= strndupa(s
+1, e
-s
-1);
79 if (inet_pton(AF_INET6
, n
, &a
->sockaddr
.in6
.sin6_addr
) <= 0)
80 return errno
> 0 ? -errno
: -EINVAL
;
87 r
= parse_ip_port(e
, &port
);
91 a
->sockaddr
.in6
.sin6_family
= AF_INET6
;
92 a
->sockaddr
.in6
.sin6_port
= htobe16(port
);
93 a
->size
= sizeof(struct sockaddr_in6
);
95 } else if (*s
== '/') {
101 if (l
>= sizeof(a
->sockaddr
.un
.sun_path
))
104 a
->sockaddr
.un
.sun_family
= AF_UNIX
;
105 memcpy(a
->sockaddr
.un
.sun_path
, s
, l
);
106 a
->size
= offsetof(struct sockaddr_un
, sun_path
) + l
+ 1;
108 } else if (*s
== '@') {
109 /* Abstract AF_UNIX socket */
113 if (l
>= sizeof(a
->sockaddr
.un
.sun_path
) - 1)
116 a
->sockaddr
.un
.sun_family
= AF_UNIX
;
117 memcpy(a
->sockaddr
.un
.sun_path
+1, s
+1, l
);
118 a
->size
= offsetof(struct sockaddr_un
, sun_path
) + 1 + l
;
120 } else if (startswith(s
, "vsock:")) {
121 /* AF_VSOCK socket in vsock:cid:port notation */
122 const char *cid_start
= s
+ STRLEN("vsock:");
125 e
= strchr(cid_start
, ':');
129 r
= safe_atou(e
+1, &port
);
133 n
= strndupa(cid_start
, e
- cid_start
);
135 r
= safe_atou(n
, &a
->sockaddr
.vm
.svm_cid
);
139 a
->sockaddr
.vm
.svm_cid
= VMADDR_CID_ANY
;
141 a
->sockaddr
.vm
.svm_family
= AF_VSOCK
;
142 a
->sockaddr
.vm
.svm_port
= port
;
143 a
->size
= sizeof(struct sockaddr_vm
);
150 r
= parse_ip_port(e
+ 1, &port
);
154 n
= strndupa(s
, e
-s
);
156 /* IPv4 in w.x.y.z:p notation? */
157 r
= inet_pton(AF_INET
, n
, &a
->sockaddr
.in
.sin_addr
);
162 /* Gotcha, it's a traditional IPv4 address */
163 a
->sockaddr
.in
.sin_family
= AF_INET
;
164 a
->sockaddr
.in
.sin_port
= htobe16(port
);
165 a
->size
= sizeof(struct sockaddr_in
);
169 if (strlen(n
) > IF_NAMESIZE
-1)
172 /* Uh, our last resort, an interface name */
173 idx
= if_nametoindex(n
);
177 a
->sockaddr
.in6
.sin6_family
= AF_INET6
;
178 a
->sockaddr
.in6
.sin6_port
= htobe16(port
);
179 a
->sockaddr
.in6
.sin6_scope_id
= idx
;
180 a
->sockaddr
.in6
.sin6_addr
= in6addr_any
;
181 a
->size
= sizeof(struct sockaddr_in6
);
186 r
= parse_ip_port(s
, &port
);
190 if (socket_ipv6_is_supported()) {
191 a
->sockaddr
.in6
.sin6_family
= AF_INET6
;
192 a
->sockaddr
.in6
.sin6_port
= htobe16(port
);
193 a
->sockaddr
.in6
.sin6_addr
= in6addr_any
;
194 a
->size
= sizeof(struct sockaddr_in6
);
196 a
->sockaddr
.in
.sin_family
= AF_INET
;
197 a
->sockaddr
.in
.sin_port
= htobe16(port
);
198 a
->sockaddr
.in
.sin_addr
.s_addr
= INADDR_ANY
;
199 a
->size
= sizeof(struct sockaddr_in
);
207 int socket_address_parse_and_warn(SocketAddress
*a
, const char *s
) {
211 /* Similar to socket_address_parse() but warns for IPv6 sockets when we don't support them. */
213 r
= socket_address_parse(&b
, s
);
217 if (!socket_ipv6_is_supported() && b
.sockaddr
.sa
.sa_family
== AF_INET6
) {
218 log_warning("Binding to IPv6 address not available since kernel does not support IPv6.");
219 return -EAFNOSUPPORT
;
226 int socket_address_parse_netlink(SocketAddress
*a
, const char *s
) {
229 _cleanup_free_
char *sfamily
= NULL
;
237 if (sscanf(s
, "%ms %u", &sfamily
, &group
) < 1)
238 return errno
> 0 ? -errno
: -EINVAL
;
240 family
= netlink_family_from_string(sfamily
);
244 a
->sockaddr
.nl
.nl_family
= AF_NETLINK
;
245 a
->sockaddr
.nl
.nl_groups
= group
;
248 a
->size
= sizeof(struct sockaddr_nl
);
249 a
->protocol
= family
;
254 int socket_address_verify(const SocketAddress
*a
) {
257 switch (socket_address_family(a
)) {
260 if (a
->size
!= sizeof(struct sockaddr_in
))
263 if (a
->sockaddr
.in
.sin_port
== 0)
266 if (!IN_SET(a
->type
, SOCK_STREAM
, SOCK_DGRAM
))
272 if (a
->size
!= sizeof(struct sockaddr_in6
))
275 if (a
->sockaddr
.in6
.sin6_port
== 0)
278 if (!IN_SET(a
->type
, SOCK_STREAM
, SOCK_DGRAM
))
284 if (a
->size
< offsetof(struct sockaddr_un
, sun_path
))
287 if (a
->size
> offsetof(struct sockaddr_un
, sun_path
)) {
289 if (a
->sockaddr
.un
.sun_path
[0] != 0) {
293 e
= memchr(a
->sockaddr
.un
.sun_path
, 0, sizeof(a
->sockaddr
.un
.sun_path
));
297 if (a
->size
!= offsetof(struct sockaddr_un
, sun_path
) + (e
- a
->sockaddr
.un
.sun_path
) + 1)
302 if (!IN_SET(a
->type
, SOCK_STREAM
, SOCK_DGRAM
, SOCK_SEQPACKET
))
309 if (a
->size
!= sizeof(struct sockaddr_nl
))
312 if (!IN_SET(a
->type
, SOCK_RAW
, SOCK_DGRAM
))
318 if (a
->size
!= sizeof(struct sockaddr_vm
))
321 if (!IN_SET(a
->type
, SOCK_STREAM
, SOCK_DGRAM
))
327 return -EAFNOSUPPORT
;
331 int socket_address_print(const SocketAddress
*a
, char **ret
) {
337 r
= socket_address_verify(a
);
341 if (socket_address_family(a
) == AF_NETLINK
) {
342 _cleanup_free_
char *sfamily
= NULL
;
344 r
= netlink_family_to_string_alloc(a
->protocol
, &sfamily
);
348 r
= asprintf(ret
, "%s %u", sfamily
, a
->sockaddr
.nl
.nl_groups
);
355 return sockaddr_pretty(&a
->sockaddr
.sa
, a
->size
, false, true, ret
);
358 bool socket_address_can_accept(const SocketAddress
*a
) {
362 IN_SET(a
->type
, SOCK_STREAM
, SOCK_SEQPACKET
);
365 bool socket_address_equal(const SocketAddress
*a
, const SocketAddress
*b
) {
369 /* Invalid addresses are unequal to all */
370 if (socket_address_verify(a
) < 0 ||
371 socket_address_verify(b
) < 0)
374 if (a
->type
!= b
->type
)
377 if (socket_address_family(a
) != socket_address_family(b
))
380 switch (socket_address_family(a
)) {
383 if (a
->sockaddr
.in
.sin_addr
.s_addr
!= b
->sockaddr
.in
.sin_addr
.s_addr
)
386 if (a
->sockaddr
.in
.sin_port
!= b
->sockaddr
.in
.sin_port
)
392 if (memcmp(&a
->sockaddr
.in6
.sin6_addr
, &b
->sockaddr
.in6
.sin6_addr
, sizeof(a
->sockaddr
.in6
.sin6_addr
)) != 0)
395 if (a
->sockaddr
.in6
.sin6_port
!= b
->sockaddr
.in6
.sin6_port
)
401 if (a
->size
<= offsetof(struct sockaddr_un
, sun_path
) ||
402 b
->size
<= offsetof(struct sockaddr_un
, sun_path
))
405 if ((a
->sockaddr
.un
.sun_path
[0] == 0) != (b
->sockaddr
.un
.sun_path
[0] == 0))
408 if (a
->sockaddr
.un
.sun_path
[0]) {
409 if (!path_equal_or_files_same(a
->sockaddr
.un
.sun_path
, b
->sockaddr
.un
.sun_path
, 0))
412 if (a
->size
!= b
->size
)
415 if (memcmp(a
->sockaddr
.un
.sun_path
, b
->sockaddr
.un
.sun_path
, a
->size
) != 0)
422 if (a
->protocol
!= b
->protocol
)
425 if (a
->sockaddr
.nl
.nl_groups
!= b
->sockaddr
.nl
.nl_groups
)
431 if (a
->sockaddr
.vm
.svm_cid
!= b
->sockaddr
.vm
.svm_cid
)
434 if (a
->sockaddr
.vm
.svm_port
!= b
->sockaddr
.vm
.svm_port
)
440 /* Cannot compare, so we assume the addresses are different */
447 bool socket_address_is(const SocketAddress
*a
, const char *s
, int type
) {
448 struct SocketAddress b
;
453 if (socket_address_parse(&b
, s
) < 0)
458 return socket_address_equal(a
, &b
);
461 bool socket_address_is_netlink(const SocketAddress
*a
, const char *s
) {
462 struct SocketAddress b
;
467 if (socket_address_parse_netlink(&b
, s
) < 0)
470 return socket_address_equal(a
, &b
);
473 const char* socket_address_get_path(const SocketAddress
*a
) {
476 if (socket_address_family(a
) != AF_UNIX
)
479 if (a
->sockaddr
.un
.sun_path
[0] == 0)
482 return a
->sockaddr
.un
.sun_path
;
485 bool socket_ipv6_is_supported(void) {
486 if (access("/proc/net/if_inet6", F_OK
) != 0)
492 bool socket_address_matches_fd(const SocketAddress
*a
, int fd
) {
499 b
.size
= sizeof(b
.sockaddr
);
500 if (getsockname(fd
, &b
.sockaddr
.sa
, &b
.size
) < 0)
503 if (b
.sockaddr
.sa
.sa_family
!= a
->sockaddr
.sa
.sa_family
)
506 solen
= sizeof(b
.type
);
507 if (getsockopt(fd
, SOL_SOCKET
, SO_TYPE
, &b
.type
, &solen
) < 0)
510 if (b
.type
!= a
->type
)
513 if (a
->protocol
!= 0) {
514 solen
= sizeof(b
.protocol
);
515 if (getsockopt(fd
, SOL_SOCKET
, SO_PROTOCOL
, &b
.protocol
, &solen
) < 0)
518 if (b
.protocol
!= a
->protocol
)
522 return socket_address_equal(a
, &b
);
525 int sockaddr_port(const struct sockaddr
*_sa
, unsigned *ret_port
) {
526 union sockaddr_union
*sa
= (union sockaddr_union
*) _sa
;
528 /* Note, this returns the port as 'unsigned' rather than 'uint16_t', as AF_VSOCK knows larger ports */
532 switch (sa
->sa
.sa_family
) {
535 *ret_port
= be16toh(sa
->in
.sin_port
);
539 *ret_port
= be16toh(sa
->in6
.sin6_port
);
543 *ret_port
= sa
->vm
.svm_port
;
547 return -EAFNOSUPPORT
;
551 int sockaddr_pretty(const struct sockaddr
*_sa
, socklen_t salen
, bool translate_ipv6
, bool include_port
, char **ret
) {
552 union sockaddr_union
*sa
= (union sockaddr_union
*) _sa
;
557 assert(salen
>= sizeof(sa
->sa
.sa_family
));
559 switch (sa
->sa
.sa_family
) {
564 a
= be32toh(sa
->in
.sin_addr
.s_addr
);
569 a
>> 24, (a
>> 16) & 0xFF, (a
>> 8) & 0xFF, a
& 0xFF,
570 be16toh(sa
->in
.sin_port
));
574 a
>> 24, (a
>> 16) & 0xFF, (a
>> 8) & 0xFF, a
& 0xFF);
581 static const unsigned char ipv4_prefix
[] = {
582 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0xFF, 0xFF
585 if (translate_ipv6
&&
586 memcmp(&sa
->in6
.sin6_addr
, ipv4_prefix
, sizeof(ipv4_prefix
)) == 0) {
587 const uint8_t *a
= sa
->in6
.sin6_addr
.s6_addr
+12;
591 a
[0], a
[1], a
[2], a
[3],
592 be16toh(sa
->in6
.sin6_port
));
596 a
[0], a
[1], a
[2], a
[3]);
600 char a
[INET6_ADDRSTRLEN
];
602 inet_ntop(AF_INET6
, &sa
->in6
.sin6_addr
, a
, sizeof(a
));
608 be16toh(sa
->in6
.sin6_port
));
622 if (salen
<= offsetof(struct sockaddr_un
, sun_path
)) {
623 p
= strdup("<unnamed>");
627 } else if (sa
->un
.sun_path
[0] == 0) {
630 /* FIXME: We assume we can print the
631 * socket path here and that it hasn't
632 * more than one NUL byte. That is
633 * actually an invalid assumption */
635 p
= new(char, sizeof(sa
->un
.sun_path
)+1);
640 memcpy(p
+1, sa
->un
.sun_path
+1, sizeof(sa
->un
.sun_path
)-1);
641 p
[sizeof(sa
->un
.sun_path
)] = 0;
644 p
= strndup(sa
->un
.sun_path
, sizeof(sa
->un
.sun_path
));
658 r
= asprintf(&p
, "vsock:%u", sa
->vm
.svm_cid
);
671 int getpeername_pretty(int fd
, bool include_port
, char **ret
) {
672 union sockaddr_union sa
;
673 socklen_t salen
= sizeof(sa
);
679 if (getpeername(fd
, &sa
.sa
, &salen
) < 0)
682 if (sa
.sa
.sa_family
== AF_UNIX
) {
683 struct ucred ucred
= {};
685 /* UNIX connection sockets are anonymous, so let's use
686 * PID/UID as pretty credentials instead */
688 r
= getpeercred(fd
, &ucred
);
692 if (asprintf(ret
, "PID "PID_FMT
"/UID "UID_FMT
, ucred
.pid
, ucred
.uid
) < 0)
698 /* For remote sockets we translate IPv6 addresses back to IPv4
699 * if applicable, since that's nicer. */
701 return sockaddr_pretty(&sa
.sa
, salen
, true, include_port
, ret
);
704 int getsockname_pretty(int fd
, char **ret
) {
705 union sockaddr_union sa
;
706 socklen_t salen
= sizeof(sa
);
711 if (getsockname(fd
, &sa
.sa
, &salen
) < 0)
714 /* For local sockets we do not translate IPv6 addresses back
715 * to IPv6 if applicable, since this is usually used for
716 * listening sockets where the difference between IPv4 and
719 return sockaddr_pretty(&sa
.sa
, salen
, false, true, ret
);
722 int socknameinfo_pretty(union sockaddr_union
*sa
, socklen_t salen
, char **_ret
) {
724 char host
[NI_MAXHOST
], *ret
;
728 r
= getnameinfo(&sa
->sa
, salen
, host
, sizeof(host
), NULL
, 0, IDN_FLAGS
);
730 int saved_errno
= errno
;
732 r
= sockaddr_pretty(&sa
->sa
, salen
, true, true, &ret
);
736 log_debug_errno(saved_errno
, "getnameinfo(%s) failed: %m", ret
);
747 int socket_address_unlink(SocketAddress
*a
) {
750 if (socket_address_family(a
) != AF_UNIX
)
753 if (a
->sockaddr
.un
.sun_path
[0] == 0)
756 if (unlink(a
->sockaddr
.un
.sun_path
) < 0)
762 static const char* const netlink_family_table
[] = {
763 [NETLINK_ROUTE
] = "route",
764 [NETLINK_FIREWALL
] = "firewall",
765 [NETLINK_INET_DIAG
] = "inet-diag",
766 [NETLINK_NFLOG
] = "nflog",
767 [NETLINK_XFRM
] = "xfrm",
768 [NETLINK_SELINUX
] = "selinux",
769 [NETLINK_ISCSI
] = "iscsi",
770 [NETLINK_AUDIT
] = "audit",
771 [NETLINK_FIB_LOOKUP
] = "fib-lookup",
772 [NETLINK_CONNECTOR
] = "connector",
773 [NETLINK_NETFILTER
] = "netfilter",
774 [NETLINK_IP6_FW
] = "ip6-fw",
775 [NETLINK_DNRTMSG
] = "dnrtmsg",
776 [NETLINK_KOBJECT_UEVENT
] = "kobject-uevent",
777 [NETLINK_GENERIC
] = "generic",
778 [NETLINK_SCSITRANSPORT
] = "scsitransport",
779 [NETLINK_ECRYPTFS
] = "ecryptfs",
780 [NETLINK_RDMA
] = "rdma",
783 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(netlink_family
, int, INT_MAX
);
785 static const char* const socket_address_bind_ipv6_only_table
[_SOCKET_ADDRESS_BIND_IPV6_ONLY_MAX
] = {
786 [SOCKET_ADDRESS_DEFAULT
] = "default",
787 [SOCKET_ADDRESS_BOTH
] = "both",
788 [SOCKET_ADDRESS_IPV6_ONLY
] = "ipv6-only"
791 DEFINE_STRING_TABLE_LOOKUP(socket_address_bind_ipv6_only
, SocketAddressBindIPv6Only
);
793 SocketAddressBindIPv6Only
parse_socket_address_bind_ipv6_only_or_bool(const char *n
) {
796 r
= parse_boolean(n
);
798 return SOCKET_ADDRESS_IPV6_ONLY
;
800 return SOCKET_ADDRESS_BOTH
;
802 return socket_address_bind_ipv6_only_from_string(n
);
805 bool sockaddr_equal(const union sockaddr_union
*a
, const union sockaddr_union
*b
) {
809 if (a
->sa
.sa_family
!= b
->sa
.sa_family
)
812 if (a
->sa
.sa_family
== AF_INET
)
813 return a
->in
.sin_addr
.s_addr
== b
->in
.sin_addr
.s_addr
;
815 if (a
->sa
.sa_family
== AF_INET6
)
816 return memcmp(&a
->in6
.sin6_addr
, &b
->in6
.sin6_addr
, sizeof(a
->in6
.sin6_addr
)) == 0;
818 if (a
->sa
.sa_family
== AF_VSOCK
)
819 return a
->vm
.svm_cid
== b
->vm
.svm_cid
;
824 int fd_inc_sndbuf(int fd
, size_t n
) {
826 socklen_t l
= sizeof(value
);
828 r
= getsockopt(fd
, SOL_SOCKET
, SO_SNDBUF
, &value
, &l
);
829 if (r
>= 0 && l
== sizeof(value
) && (size_t) value
>= n
*2)
832 /* If we have the privileges we will ignore the kernel limit. */
835 if (setsockopt(fd
, SOL_SOCKET
, SO_SNDBUFFORCE
, &value
, sizeof(value
)) < 0)
836 if (setsockopt(fd
, SOL_SOCKET
, SO_SNDBUF
, &value
, sizeof(value
)) < 0)
842 int fd_inc_rcvbuf(int fd
, size_t n
) {
844 socklen_t l
= sizeof(value
);
846 r
= getsockopt(fd
, SOL_SOCKET
, SO_RCVBUF
, &value
, &l
);
847 if (r
>= 0 && l
== sizeof(value
) && (size_t) value
>= n
*2)
850 /* If we have the privileges we will ignore the kernel limit. */
853 if (setsockopt(fd
, SOL_SOCKET
, SO_RCVBUFFORCE
, &value
, sizeof(value
)) < 0)
854 if (setsockopt(fd
, SOL_SOCKET
, SO_RCVBUF
, &value
, sizeof(value
)) < 0)
859 static const char* const ip_tos_table
[] = {
860 [IPTOS_LOWDELAY
] = "low-delay",
861 [IPTOS_THROUGHPUT
] = "throughput",
862 [IPTOS_RELIABILITY
] = "reliability",
863 [IPTOS_LOWCOST
] = "low-cost",
866 DEFINE_STRING_TABLE_LOOKUP_WITH_FALLBACK(ip_tos
, int, 0xff);
868 bool ifname_valid(const char *p
) {
871 /* Checks whether a network interface name is valid. This is inspired by dev_valid_name() in the kernel sources
872 * but slightly stricter, as we only allow non-control, non-space ASCII characters in the interface name. We
873 * also don't permit names that only container numbers, to avoid confusion with numeric interface indexes. */
878 if (strlen(p
) >= IFNAMSIZ
)
881 if (dot_or_dot_dot(p
))
885 if ((unsigned char) *p
>= 127U)
888 if ((unsigned char) *p
<= 32U)
891 if (IN_SET(*p
, ':', '/'))
894 numeric
= numeric
&& (*p
>= '0' && *p
<= '9');
904 bool address_label_valid(const char *p
) {
909 if (strlen(p
) >= IFNAMSIZ
)
913 if ((uint8_t) *p
>= 127U)
916 if ((uint8_t) *p
<= 31U)
924 int getpeercred(int fd
, struct ucred
*ucred
) {
925 socklen_t n
= sizeof(struct ucred
);
932 r
= getsockopt(fd
, SOL_SOCKET
, SO_PEERCRED
, &u
, &n
);
936 if (n
!= sizeof(struct ucred
))
939 /* Check if the data is actually useful and not suppressed due to namespacing issues */
940 if (!pid_is_valid(u
.pid
))
943 /* Note that we don't check UID/GID here, as namespace translation works differently there: instead of
944 * receiving in "invalid" user/group we get the overflow UID/GID. */
950 int getpeersec(int fd
, char **ret
) {
951 _cleanup_free_
char *s
= NULL
;
962 if (getsockopt(fd
, SOL_SOCKET
, SO_PEERSEC
, s
, &n
) >= 0)
979 int getpeergroups(int fd
, gid_t
**ret
) {
980 socklen_t n
= sizeof(gid_t
) * 64;
981 _cleanup_free_ gid_t
*d
= NULL
;
991 if (getsockopt(fd
, SOL_SOCKET
, SO_PEERGROUPS
, d
, &n
) >= 0)
1000 assert_se(n
% sizeof(gid_t
) == 0);
1003 if ((socklen_t
) (int) n
!= n
)
1014 const struct sockaddr
*sa
, socklen_t len
,
1018 struct cmsghdr cmsghdr
;
1019 uint8_t buf
[CMSG_SPACE(sizeof(int))];
1021 struct msghdr mh
= {
1022 .msg_name
= (struct sockaddr
*) sa
,
1024 .msg_control
= &control
,
1025 .msg_controllen
= sizeof(control
),
1027 struct cmsghdr
*cmsg
;
1029 assert(transport_fd
>= 0);
1032 cmsg
= CMSG_FIRSTHDR(&mh
);
1033 cmsg
->cmsg_level
= SOL_SOCKET
;
1034 cmsg
->cmsg_type
= SCM_RIGHTS
;
1035 cmsg
->cmsg_len
= CMSG_LEN(sizeof(int));
1036 memcpy(CMSG_DATA(cmsg
), &fd
, sizeof(int));
1038 mh
.msg_controllen
= CMSG_SPACE(sizeof(int));
1039 if (sendmsg(transport_fd
, &mh
, MSG_NOSIGNAL
| flags
) < 0)
1045 int receive_one_fd(int transport_fd
, int flags
) {
1047 struct cmsghdr cmsghdr
;
1048 uint8_t buf
[CMSG_SPACE(sizeof(int))];
1050 struct msghdr mh
= {
1051 .msg_control
= &control
,
1052 .msg_controllen
= sizeof(control
),
1054 struct cmsghdr
*cmsg
, *found
= NULL
;
1056 assert(transport_fd
>= 0);
1059 * Receive a single FD via @transport_fd. We don't care for
1060 * the transport-type. We retrieve a single FD at most, so for
1061 * packet-based transports, the caller must ensure to send
1062 * only a single FD per packet. This is best used in
1063 * combination with send_one_fd().
1066 if (recvmsg(transport_fd
, &mh
, MSG_NOSIGNAL
| MSG_CMSG_CLOEXEC
| flags
) < 0)
1069 CMSG_FOREACH(cmsg
, &mh
) {
1070 if (cmsg
->cmsg_level
== SOL_SOCKET
&&
1071 cmsg
->cmsg_type
== SCM_RIGHTS
&&
1072 cmsg
->cmsg_len
== CMSG_LEN(sizeof(int))) {
1080 cmsg_close_all(&mh
);
1084 return *(int*) CMSG_DATA(found
);
1087 ssize_t
next_datagram_size_fd(int fd
) {
1091 /* This is a bit like FIONREAD/SIOCINQ, however a bit more powerful. The difference being: recv(MSG_PEEK) will
1092 * actually cause the next datagram in the queue to be validated regarding checksums, which FIONREAD doesn't
1093 * do. This difference is actually of major importance as we need to be sure that the size returned here
1094 * actually matches what we will read with recvmsg() next, as otherwise we might end up allocating a buffer of
1095 * the wrong size. */
1097 l
= recv(fd
, NULL
, 0, MSG_PEEK
|MSG_TRUNC
);
1099 if (IN_SET(errno
, EOPNOTSUPP
, EFAULT
))
1112 /* Some sockets (AF_PACKET) do not support null-sized recv() with MSG_TRUNC set, let's fall back to FIONREAD
1113 * for them. Checksums don't matter for raw sockets anyway, hence this should be fine. */
1115 if (ioctl(fd
, FIONREAD
, &k
) < 0)
1121 int flush_accept(int fd
) {
1123 struct pollfd pollfd
= {
1129 /* Similar to flush_fd() but flushes all incoming connection by accepting them and immediately closing them. */
1134 r
= poll(&pollfd
, 1, 0);
1144 cfd
= accept4(fd
, NULL
, NULL
, SOCK_NONBLOCK
|SOCK_CLOEXEC
);
1149 if (errno
== EAGAIN
)
1159 struct cmsghdr
* cmsg_find(struct msghdr
*mh
, int level
, int type
, socklen_t length
) {
1160 struct cmsghdr
*cmsg
;
1164 CMSG_FOREACH(cmsg
, mh
)
1165 if (cmsg
->cmsg_level
== level
&&
1166 cmsg
->cmsg_type
== type
&&
1167 (length
== (socklen_t
) -1 || length
== cmsg
->cmsg_len
))
1173 int socket_ioctl_fd(void) {
1176 /* Create a socket to invoke the various network interface ioctl()s on. Traditionally only AF_INET was good for
1177 * that. Since kernel 4.6 AF_NETLINK works for this too. We first try to use AF_INET hence, but if that's not
1178 * available (for example, because it is made unavailable via SECCOMP or such), we'll fall back to the more
1179 * generic AF_NETLINK. */
1181 fd
= socket(AF_INET
, SOCK_DGRAM
|SOCK_CLOEXEC
, 0);
1183 fd
= socket(AF_NETLINK
, SOCK_RAW
|SOCK_CLOEXEC
, NETLINK_GENERIC
);