src/basic/socket-util.h

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2 #pragma once
   3
   4 /***
   5   This file is part of systemd.
   6
   7   Copyright 2010 Lennart Poettering
   8 ***/
   9
  10 #include <netinet/ether.h>
  11 #include <netinet/in.h>
  12 #include <stdbool.h>
  13 #include <stddef.h>
  14 #include <sys/socket.h>
  15 #include <sys/types.h>
  16 #include <sys/un.h>
  17 #include <linux/netlink.h>
  18 #include <linux/if_infiniband.h>
  19 #include <linux/if_packet.h>
  20
  21 #include "macro.h"
  22 #include "missing.h"
  23 #include "util.h"
  24
  25 union sockaddr_union {
  26         /* The minimal, abstract version */
  27         struct sockaddr sa;
  28
  29         /* The libc provided version that allocates "enough room" for every protocol */
  30         struct sockaddr_storage storage;
  31
  32         /* Protoctol-specific implementations */
  33         struct sockaddr_in in;
  34         struct sockaddr_in6 in6;
  35         struct sockaddr_un un;
  36 #if 0 /// UNNEEDED by elogind.
  37         struct sockaddr_nl nl;
  38         struct sockaddr_ll ll;
  39 #endif // 0
  40         struct sockaddr_vm vm;
  41
  42         /* Ensure there is enough space to store Infiniband addresses */
  43         uint8_t ll_buffer[offsetof(struct sockaddr_ll, sll_addr) + CONST_MAX(ETH_ALEN, INFINIBAND_ALEN)];
  44
  45         /* Ensure there is enough space after the AF_UNIX sun_path for one more NUL byte, just to be sure that the path
  46          * component is always followed by at least one NUL byte. */
  47         uint8_t un_buffer[sizeof(struct sockaddr_un) + 1];
  48 };
  49
  50 #if 0 /// UNNEEDED by elogind
  51 typedef struct SocketAddress {
  52         union sockaddr_union sockaddr;
  53
  54         /* We store the size here explicitly due to the weird
  55          * sockaddr_un semantics for abstract sockets */
  56         socklen_t size;
  57
  58         /* Socket type, i.e. SOCK_STREAM, SOCK_DGRAM, ... */
  59         int type;
  60
  61         /* Socket protocol, IPPROTO_xxx, usually 0, except for netlink */
  62         int protocol;
  63 } SocketAddress;
  64
  65 typedef enum SocketAddressBindIPv6Only {
  66         SOCKET_ADDRESS_DEFAULT,
  67         SOCKET_ADDRESS_BOTH,
  68         SOCKET_ADDRESS_IPV6_ONLY,
  69         _SOCKET_ADDRESS_BIND_IPV6_ONLY_MAX,
  70         _SOCKET_ADDRESS_BIND_IPV6_ONLY_INVALID = -1
  71 } SocketAddressBindIPv6Only;
  72
  73 #define socket_address_family(a) ((a)->sockaddr.sa.sa_family)
  74
  75 const char* socket_address_type_to_string(int t) _const_;
  76 int socket_address_type_from_string(const char *s) _pure_;
  77
  78 int socket_address_parse(SocketAddress *a, const char *s);
  79 int socket_address_parse_and_warn(SocketAddress *a, const char *s);
  80 int socket_address_parse_netlink(SocketAddress *a, const char *s);
  81 int socket_address_print(const SocketAddress *a, char **p);
  82 int socket_address_verify(const SocketAddress *a) _pure_;
  83 int socket_address_unlink(SocketAddress *a);
  84
  85 bool socket_address_can_accept(const SocketAddress *a) _pure_;
  86
  87 int socket_address_listen(
  88                 const SocketAddress *a,
  89                 int flags,
  90                 int backlog,
  91                 SocketAddressBindIPv6Only only,
  92                 const char *bind_to_device,
  93                 bool reuse_port,
  94                 bool free_bind,
  95                 bool transparent,
  96                 mode_t directory_mode,
  97                 mode_t socket_mode,
  98                 const char *label);
  99 int make_socket_fd(int log_level, const char* address, int type, int flags);
 100
 101 bool socket_address_is(const SocketAddress *a, const char *s, int type);
 102 bool socket_address_is_netlink(const SocketAddress *a, const char *s);
 103
 104 bool socket_address_matches_fd(const SocketAddress *a, int fd);
 105
 106 bool socket_address_equal(const SocketAddress *a, const SocketAddress *b) _pure_;
 107
 108 const char* socket_address_get_path(const SocketAddress *a);
 109
 110 bool socket_ipv6_is_supported(void);
 111 #endif // 0
 112
 113 int sockaddr_port(const struct sockaddr *_sa, unsigned *port);
 114
 115 #if 0 /// UNNEEDED by elogind
 116 int sockaddr_pretty(const struct sockaddr *_sa, socklen_t salen, bool translate_ipv6, bool include_port, char **ret);
 117 int getpeername_pretty(int fd, bool include_port, char **ret);
 118 int getsockname_pretty(int fd, char **ret);
 119
 120 int socknameinfo_pretty(union sockaddr_union *sa, socklen_t salen, char **_ret);
 121
 122 const char* socket_address_bind_ipv6_only_to_string(SocketAddressBindIPv6Only b) _const_;
 123 SocketAddressBindIPv6Only socket_address_bind_ipv6_only_from_string(const char *s) _pure_;
 124 SocketAddressBindIPv6Only socket_address_bind_ipv6_only_or_bool_from_string(const char *s);
 125
 126 int netlink_family_to_string_alloc(int b, char **s);
 127 int netlink_family_from_string(const char *s) _pure_;
 128
 129 bool sockaddr_equal(const union sockaddr_union *a, const union sockaddr_union *b);
 130 #endif // 0
 131
 132 int fd_inc_sndbuf(int fd, size_t n);
 133 int fd_inc_rcvbuf(int fd, size_t n);
 134 #if 0 /// UNNEEDED by elogind
 135
 136 int ip_tos_to_string_alloc(int i, char **s);
 137 int ip_tos_from_string(const char *s);
 138
 139 bool ifname_valid(const char *p);
 140 bool address_label_valid(const char *p);
 141 #endif // 0
 142
 143 int getpeercred(int fd, struct ucred *ucred);
 144 int getpeersec(int fd, char **ret);
 145 int getpeergroups(int fd, gid_t **ret);
 146
 147 int send_one_fd_sa(int transport_fd,
 148                    int fd,
 149                    const struct sockaddr *sa, socklen_t len,
 150                    int flags);
 151 #define send_one_fd(transport_fd, fd, flags) send_one_fd_sa(transport_fd, fd, NULL, 0, flags)
 152 #if 0 /// UNNEEDED by elogind
 153 int receive_one_fd(int transport_fd, int flags);
 154
 155 ssize_t next_datagram_size_fd(int fd);
 156
 157 int flush_accept(int fd);
 158 #endif // 0
 159
 160 #define CMSG_FOREACH(cmsg, mh)                                          \
 161         for ((cmsg) = CMSG_FIRSTHDR(mh); (cmsg); (cmsg) = CMSG_NXTHDR((mh), (cmsg)))
 162
 163 #if 0 /// UNNEEDED by elogind
 164 struct cmsghdr* cmsg_find(struct msghdr *mh, int level, int type, socklen_t length);
 165 #endif // 0
 166
 167 /*
 168  * Certain hardware address types (e.g Infiniband) do not fit into sll_addr
 169  * (8 bytes) and run over the structure. This macro returns the correct size that
 170  * must be passed to kernel.
 171  */
 172 #define SOCKADDR_LL_LEN(sa)                                             \
 173         ({                                                              \
 174                 const struct sockaddr_ll *_sa = &(sa);                  \
 175                 size_t _mac_len = sizeof(_sa->sll_addr);                \
 176                 assert(_sa->sll_family == AF_PACKET);                   \
 177                 if (be16toh(_sa->sll_hatype) == ARPHRD_ETHER)           \
 178                         _mac_len = MAX(_mac_len, (size_t) ETH_ALEN);    \
 179                 if (be16toh(_sa->sll_hatype) == ARPHRD_INFINIBAND)      \
 180                         _mac_len = MAX(_mac_len, (size_t) INFINIBAND_ALEN); \
 181                 offsetof(struct sockaddr_ll, sll_addr) + _mac_len;      \
 182         })
 183
 184 /* Covers only file system and abstract AF_UNIX socket addresses, but not unnamed socket addresses. */
 185 #define SOCKADDR_UN_LEN(sa)                                             \
 186         ({                                                              \
 187                 const struct sockaddr_un *_sa = &(sa);                  \
 188                 assert(_sa->sun_family == AF_UNIX);                     \
 189                 offsetof(struct sockaddr_un, sun_path) +                \
 190                         (_sa->sun_path[0] == 0 ?                        \
 191                          1 + strnlen(_sa->sun_path+1, sizeof(_sa->sun_path)-1) : \
 192                          strnlen(_sa->sun_path, sizeof(_sa->sun_path))); \
 193         })
 194
 195 #if 0 /// UNNEEDED by elogind
 196 int socket_ioctl_fd(void);
 197 #endif // 0