udp: refactor udp code
[vpp.git] / src / vnet / ip / ip.c
1 /*
2  * Copyright (c) 2017 Cisco and/or its affiliates.
3  * Licensed under the Apache License, Version 2.0 (the "License");
4  * you may not use this file except in compliance with the License.
5  * You may obtain a copy of the License at:
6  *
7  *     http://www.apache.org/licenses/LICENSE-2.0
8  *
9  * Unless required by applicable law or agreed to in writing, software
10  * distributed under the License is distributed on an "AS IS" BASIS,
11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12  * See the License for the specific language governing permissions and
13  * limitations under the License.
14  */
15
16 #include <vnet/ip/ip.h>
17 #include <vnet/fib/fib_table.h>
18
19 u8
20 ip_is_zero (ip46_address_t * ip46_address, u8 is_ip4)
21 {
22   if (is_ip4)
23     return (ip46_address->ip4.as_u32 == 0);
24   else
25     return (ip46_address->as_u64[0] == 0 && ip46_address->as_u64[1] == 0);
26 }
27
28 u8
29 ip_is_local_host (ip46_address_t * ip46_address, u8 is_ip4)
30 {
31   if (is_ip4)
32     return (ip46_address->ip4.as_u8[0] == 127);
33   else
34     return (ip46_address->as_u64[0] == 0 && ip46_address->as_u64[1] == 1);
35 }
36
37 /**
38  * Checks that an ip is local to the requested fib
39  */
40 u8
41 ip_is_local (u32 fib_index, ip46_address_t * ip46_address, u8 is_ip4)
42 {
43   fib_node_index_t fei;
44   fib_entry_flag_t flags;
45   fib_prefix_t prefix;
46
47   /* Check if requester is local */
48   if (is_ip4)
49     {
50       prefix.fp_len = 32;
51       prefix.fp_proto = FIB_PROTOCOL_IP4;
52     }
53   else
54     {
55       prefix.fp_len = 128;
56       prefix.fp_proto = FIB_PROTOCOL_IP6;
57     }
58
59   clib_memcpy (&prefix.fp_addr, ip46_address, sizeof (ip46_address_t));
60   fei = fib_table_lookup (0, &prefix);
61   flags = fib_entry_get_flags (fei);
62
63   return (flags & FIB_ENTRY_FLAG_LOCAL);
64 }
65
66 void
67 ip_copy (ip46_address_t * dst, ip46_address_t * src, u8 is_ip4)
68 {
69   if (is_ip4)
70     dst->ip4.as_u32 = src->ip4.as_u32;
71   else
72     clib_memcpy (&dst->ip6, &src->ip6, sizeof (ip6_address_t));
73 }
74
75 void
76 ip_set (ip46_address_t * dst, void *src, u8 is_ip4)
77 {
78   if (is_ip4)
79     dst->ip4.as_u32 = ((ip4_address_t *) src)->as_u32;
80   else
81     clib_memcpy (&dst->ip6, (ip6_address_t *) src, sizeof (ip6_address_t));
82 }
83
84 u8
85 ip_interface_has_address (u32 sw_if_index, ip46_address_t * ip, u8 is_ip4)
86 {
87   ip_interface_address_t *ia = 0;
88
89   if (is_ip4)
90     {
91       ip_lookup_main_t *lm4 = &ip4_main.lookup_main;
92       ip4_address_t *ip4;
93       /* *INDENT-OFF* */
94       foreach_ip_interface_address (lm4, ia, sw_if_index, 1 /* unnumbered */ ,
95       ({
96         ip4 = ip_interface_address_get_address (lm4, ia);
97         if (ip4_address_compare (ip4, &ip->ip4) == 0)
98           return 1;
99       }));
100       /* *INDENT-ON* */
101     }
102   else
103     {
104       ip_lookup_main_t *lm6 = &ip6_main.lookup_main;
105       ip6_address_t *ip6;
106       /* *INDENT-OFF* */
107       foreach_ip_interface_address (lm6, ia, sw_if_index, 1 /* unnumbered */ ,
108       ({
109         ip6 = ip_interface_address_get_address (lm6, ia);
110         if (ip6_address_compare (ip6, &ip->ip6) == 0)
111           return 1;
112       }));
113       /* *INDENT-ON* */
114     }
115   return 0;
116 }
117
118 void *
119 ip_interface_get_first_ip (u32 sw_if_index, u8 is_ip4)
120 {
121   ip_lookup_main_t *lm4 = &ip4_main.lookup_main;
122   ip_lookup_main_t *lm6 = &ip6_main.lookup_main;
123   ip_interface_address_t *ia = 0;
124
125   if (is_ip4)
126     {
127       /* *INDENT-OFF* */
128       foreach_ip_interface_address (lm4, ia, sw_if_index, 1 /* unnumbered */ ,
129       ({
130         return ip_interface_address_get_address (lm4, ia);
131       }));
132       /* *INDENT-ON* */
133     }
134   else
135     {
136       /* *INDENT-OFF* */
137       foreach_ip_interface_address (lm6, ia, sw_if_index, 1 /* unnumbered */ ,
138       ({
139         ip6_address_t *rv;
140         rv = ip_interface_address_get_address (lm6, ia);
141         /* Trying to use a link-local ip6 src address is a fool's errand */
142         if (!ip6_address_is_link_local_unicast (rv))
143           return rv;
144       }));
145       /* *INDENT-ON* */
146     }
147
148   return 0;
149 }
150
151 /*
152  * fd.io coding-style-patch-verification: ON
153  *
154  * Local Variables:
155  * eval: (c-set-style "gnu")
156  * End:
157  */