1/*
2 * linux/include/linux/sunrpc/addr.h
3 *
4 * Various routines for copying and comparing sockaddrs and for
5 * converting them to and from presentation format.
6 */
7#ifndef _LINUX_SUNRPC_ADDR_H
8#define _LINUX_SUNRPC_ADDR_H
9
10#include <linux/socket.h>
11#include <linux/in.h>
12#include <linux/in6.h>
13#include <net/ipv6.h>
14
15size_t		rpc_ntop(const struct sockaddr *, char *, const size_t);
16size_t		rpc_pton(struct net *, const char *, const size_t,
17			 struct sockaddr *, const size_t);
18char *		rpc_sockaddr2uaddr(const struct sockaddr *, gfp_t);
19size_t		rpc_uaddr2sockaddr(struct net *, const char *, const size_t,
20				   struct sockaddr *, const size_t);
21
22static inline unsigned short rpc_get_port(const struct sockaddr *sap)
23{
24	switch (sap->sa_family) {
25	case AF_INET:
26		return ntohs(((struct sockaddr_in *)sap)->sin_port);
27	case AF_INET6:
28		return ntohs(((struct sockaddr_in6 *)sap)->sin6_port);
29	}
30	return 0;
31}
32
33static inline void rpc_set_port(struct sockaddr *sap,
34				const unsigned short port)
35{
36	switch (sap->sa_family) {
37	case AF_INET:
38		((struct sockaddr_in *)sap)->sin_port = htons(port);
39		break;
40	case AF_INET6:
41		((struct sockaddr_in6 *)sap)->sin6_port = htons(port);
42		break;
43	}
44}
45
46#define IPV6_SCOPE_DELIMITER		'%'
47#define IPV6_SCOPE_ID_LEN		sizeof("%nnnnnnnnnn")
48
49static inline bool __rpc_cmp_addr4(const struct sockaddr *sap1,
50				   const struct sockaddr *sap2)
51{
52	const struct sockaddr_in *sin1 = (const struct sockaddr_in *)sap1;
53	const struct sockaddr_in *sin2 = (const struct sockaddr_in *)sap2;
54
55	return sin1->sin_addr.s_addr == sin2->sin_addr.s_addr;
56}
57
58static inline bool __rpc_copy_addr4(struct sockaddr *dst,
59				    const struct sockaddr *src)
60{
61	const struct sockaddr_in *ssin = (struct sockaddr_in *) src;
62	struct sockaddr_in *dsin = (struct sockaddr_in *) dst;
63
64	dsin->sin_family = ssin->sin_family;
65	dsin->sin_addr.s_addr = ssin->sin_addr.s_addr;
66	return true;
67}
68
69#if IS_ENABLED(CONFIG_IPV6)
70static inline bool __rpc_cmp_addr6(const struct sockaddr *sap1,
71				   const struct sockaddr *sap2)
72{
73	const struct sockaddr_in6 *sin1 = (const struct sockaddr_in6 *)sap1;
74	const struct sockaddr_in6 *sin2 = (const struct sockaddr_in6 *)sap2;
75
76	if (!ipv6_addr_equal(&sin1->sin6_addr, &sin2->sin6_addr))
77		return false;
78	else if (ipv6_addr_type(&sin1->sin6_addr) & IPV6_ADDR_LINKLOCAL)
79		return sin1->sin6_scope_id == sin2->sin6_scope_id;
80
81	return true;
82}
83
84static inline bool __rpc_copy_addr6(struct sockaddr *dst,
85				    const struct sockaddr *src)
86{
87	const struct sockaddr_in6 *ssin6 = (const struct sockaddr_in6 *) src;
88	struct sockaddr_in6 *dsin6 = (struct sockaddr_in6 *) dst;
89
90	dsin6->sin6_family = ssin6->sin6_family;
91	dsin6->sin6_addr = ssin6->sin6_addr;
92	dsin6->sin6_scope_id = ssin6->sin6_scope_id;
93	return true;
94}
95#else	/* !(IS_ENABLED(CONFIG_IPV6) */
96static inline bool __rpc_cmp_addr6(const struct sockaddr *sap1,
97				   const struct sockaddr *sap2)
98{
99	return false;
100}
101
102static inline bool __rpc_copy_addr6(struct sockaddr *dst,
103				    const struct sockaddr *src)
104{
105	return false;
106}
107#endif	/* !(IS_ENABLED(CONFIG_IPV6) */
108
109/**
110 * rpc_cmp_addr - compare the address portion of two sockaddrs.
111 * @sap1: first sockaddr
112 * @sap2: second sockaddr
113 *
114 * Just compares the family and address portion. Ignores port, but
115 * compares the scope if it's a link-local address.
116 *
117 * Returns true if the addrs are equal, false if they aren't.
118 */
119static inline bool rpc_cmp_addr(const struct sockaddr *sap1,
120				const struct sockaddr *sap2)
121{
122	if (sap1->sa_family == sap2->sa_family) {
123		switch (sap1->sa_family) {
124		case AF_INET:
125			return __rpc_cmp_addr4(sap1, sap2);
126		case AF_INET6:
127			return __rpc_cmp_addr6(sap1, sap2);
128		}
129	}
130	return false;
131}
132
133/**
134 * rpc_copy_addr - copy the address portion of one sockaddr to another
135 * @dst: destination sockaddr
136 * @src: source sockaddr
137 *
138 * Just copies the address portion and family. Ignores port, scope, etc.
139 * Caller is responsible for making certain that dst is large enough to hold
140 * the address in src. Returns true if address family is supported. Returns
141 * false otherwise.
142 */
143static inline bool rpc_copy_addr(struct sockaddr *dst,
144				 const struct sockaddr *src)
145{
146	switch (src->sa_family) {
147	case AF_INET:
148		return __rpc_copy_addr4(dst, src);
149	case AF_INET6:
150		return __rpc_copy_addr6(dst, src);
151	}
152	return false;
153}
154
155/**
156 * rpc_get_scope_id - return scopeid for a given sockaddr
157 * @sa: sockaddr to get scopeid from
158 *
159 * Returns the value of the sin6_scope_id for AF_INET6 addrs, or 0 if
160 * not an AF_INET6 address.
161 */
162static inline u32 rpc_get_scope_id(const struct sockaddr *sa)
163{
164	if (sa->sa_family != AF_INET6)
165		return 0;
166
167	return ((struct sockaddr_in6 *) sa)->sin6_scope_id;
168}
169
170#endif /* _LINUX_SUNRPC_ADDR_H */
171