2 * Copyright (c) 2020 Doc.ai and/or its affiliates.
3 * Copyright (c) 2020 Cisco and/or its affiliates.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at:
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
17 #include <vnet/vnet.h>
18 #include <vnet/ip/ip6_link.h>
19 #include <vppinfra/error.h>
20 #include <vlibmemory/api.h>
21 #include <wireguard/wireguard.h>
22 #include <wireguard/wireguard_send.h>
25 ip46_enqueue_packet (vlib_main_t *vm, u32 bi0, int is_ip4)
28 u32 lookup_node_index =
29 is_ip4 ? ip4_lookup_node.index : ip6_lookup_node.index;
31 f = vlib_get_frame_to_node (vm, lookup_node_index);
32 /* f can not be NULL here - frame allocation failure causes panic */
34 u32 *to_next = vlib_frame_vector_args (f);
38 vlib_put_frame_to_node (vm, lookup_node_index, f);
44 wg_buffer_prepend_rewrite (vlib_main_t *vm, vlib_buffer_t *b0,
45 const u8 *rewrite, u8 is_ip4)
49 ip4_udp_header_t *hdr4;
51 vlib_buffer_advance (b0, -sizeof (*hdr4));
53 hdr4 = vlib_buffer_get_current (b0);
55 /* copy only ip4 and udp header; wireguard header not needed */
56 clib_memcpy (hdr4, rewrite, sizeof (ip4_udp_header_t));
59 clib_host_to_net_u16 (b0->current_length - sizeof (ip4_header_t));
60 ip4_header_set_len_w_chksum (&hdr4->ip4,
61 clib_host_to_net_u16 (b0->current_length));
65 ip6_udp_header_t *hdr6;
67 vlib_buffer_advance (b0, -sizeof (*hdr6));
69 hdr6 = vlib_buffer_get_current (b0);
71 /* copy only ip6 and udp header; wireguard header not needed */
72 clib_memcpy (hdr6, rewrite, sizeof (ip6_udp_header_t));
74 hdr6->ip6.payload_length = hdr6->udp.length =
75 clib_host_to_net_u16 (b0->current_length - sizeof (ip6_header_t));
77 /* IPv6 UDP checksum is mandatory */
79 ip6_header_t *ip6_0 = &(hdr6->ip6);
81 ip6_tcp_udp_icmp_compute_checksum (vm, b0, ip6_0, &bogus);
87 wg_create_buffer (vlib_main_t *vm, const u8 *rewrite, const u8 *packet,
88 u32 packet_len, u32 *bi, u8 is_ip4)
93 n_buf0 = vlib_buffer_alloc (vm, bi, 1);
97 b0 = vlib_get_buffer (vm, *bi);
99 u8 *payload = vlib_buffer_get_current (b0);
100 clib_memcpy (payload, packet, packet_len);
102 b0->current_length = packet_len;
104 wg_buffer_prepend_rewrite (vm, b0, rewrite, is_ip4);
110 wg_build_rewrite (ip46_address_t *src_addr, u16 src_port,
111 ip46_address_t *dst_addr, u16 dst_port, u8 is_ip4)
113 if (ip46_address_is_zero (dst_addr) || 0 == dst_port)
119 ip4_udp_header_t *hdr;
121 /* reserve space for ip4, udp and wireguard headers */
122 vec_validate (rewrite, sizeof (ip4_udp_wg_header_t) - 1);
123 hdr = (ip4_udp_header_t *) rewrite;
125 hdr->ip4.ip_version_and_header_length = 0x45;
127 hdr->ip4.src_address = src_addr->ip4;
128 hdr->ip4.dst_address = dst_addr->ip4;
129 hdr->ip4.protocol = IP_PROTOCOL_UDP;
130 hdr->ip4.checksum = ip4_header_checksum (&hdr->ip4);
132 hdr->udp.src_port = clib_host_to_net_u16 (src_port);
133 hdr->udp.dst_port = clib_host_to_net_u16 (dst_port);
134 hdr->udp.checksum = 0;
138 ip6_udp_header_t *hdr;
140 /* reserve space for ip6, udp and wireguard headers */
141 vec_validate (rewrite, sizeof (ip6_udp_wg_header_t) - 1);
142 hdr = (ip6_udp_header_t *) rewrite;
144 hdr->ip6.ip_version_traffic_class_and_flow_label = 0x60;
145 ip6_address_copy (&hdr->ip6.src_address, &src_addr->ip6);
146 ip6_address_copy (&hdr->ip6.dst_address, &dst_addr->ip6);
147 hdr->ip6.protocol = IP_PROTOCOL_UDP;
148 hdr->ip6.hop_limit = 64;
150 hdr->udp.src_port = clib_host_to_net_u16 (src_port);
151 hdr->udp.dst_port = clib_host_to_net_u16 (dst_port);
152 hdr->udp.checksum = 0;
159 wg_send_handshake (vlib_main_t * vm, wg_peer_t * peer, bool is_retry)
161 ASSERT (vm->thread_index == 0);
163 if (!wg_peer_can_send (peer))
166 message_handshake_initiation_t packet;
169 peer->timer_handshake_attempts = 0;
171 if (!wg_birthdate_has_expired (peer->last_sent_handshake, REKEY_TIMEOUT) ||
172 wg_peer_is_dead (peer))
175 if (noise_create_initiation (vm,
177 &packet.sender_index,
178 packet.unencrypted_ephemeral,
179 packet.encrypted_static,
180 packet.encrypted_timestamp))
182 packet.header.type = MESSAGE_HANDSHAKE_INITIATION;
183 cookie_maker_mac (&peer->cookie_maker, &packet.macs, &packet,
185 wg_timers_any_authenticated_packet_sent (peer);
186 wg_timers_handshake_initiated (peer);
187 wg_timers_any_authenticated_packet_traversal (peer);
189 peer->last_sent_handshake = vlib_time_now (vm);
194 u8 is_ip4 = ip46_address_is_ip4 (&peer->dst.addr);
196 if (!wg_create_buffer (vm, peer->rewrite, (u8 *) &packet, sizeof (packet),
200 ip46_enqueue_packet (vm, bi0, is_ip4);
211 wg_send_handshake_thread_fn (void *arg)
213 wg_send_args_t *a = arg;
215 wg_main_t *wmp = &wg_main;
216 wg_peer_t *peer = wg_peer_get (a->peer_idx);
218 wg_send_handshake (wmp->vlib_main, peer, a->is_retry);
223 wg_send_handshake_from_mt (u32 peer_idx, bool is_retry)
226 .peer_idx = peer_idx,
227 .is_retry = is_retry,
230 vl_api_rpc_call_main_thread (wg_send_handshake_thread_fn,
231 (u8 *) & a, sizeof (a));
235 wg_send_keepalive (vlib_main_t * vm, wg_peer_t * peer)
237 ASSERT (vm->thread_index == 0);
239 if (!wg_peer_can_send (peer))
242 u32 size_of_packet = message_data_len (0);
243 message_data_t *packet =
244 (message_data_t *) wg_main.per_thread_data[vm->thread_index].data;
247 enum noise_state_crypt state;
249 if (!peer->remote.r_current)
251 wg_send_handshake (vm, peer, false);
256 noise_remote_encrypt (vm,
258 &packet->receiver_index,
259 &packet->counter, NULL, 0, packet->encrypted_data);
261 if (PREDICT_FALSE (state == SC_KEEP_KEY_FRESH))
263 wg_send_handshake (vm, peer, false);
265 else if (PREDICT_FALSE (state == SC_FAILED))
267 wg_peer_update_flags (peer - wg_peer_pool, WG_PEER_ESTABLISHED, false);
272 u8 is_ip4 = ip46_address_is_ip4 (&peer->dst.addr);
273 packet->header.type = MESSAGE_DATA;
275 if (!wg_create_buffer (vm, peer->rewrite, (u8 *) packet, size_of_packet,
282 ip46_enqueue_packet (vm, bi0, is_ip4);
284 wg_timers_any_authenticated_packet_sent (peer);
285 wg_timers_any_authenticated_packet_traversal (peer);
292 wg_send_handshake_response (vlib_main_t * vm, wg_peer_t * peer)
294 message_handshake_response_t packet;
296 if (!wg_peer_can_send (peer))
299 if (noise_create_response (vm,
301 &packet.sender_index,
302 &packet.receiver_index,
303 packet.unencrypted_ephemeral,
304 packet.encrypted_nothing))
306 packet.header.type = MESSAGE_HANDSHAKE_RESPONSE;
307 cookie_maker_mac (&peer->cookie_maker, &packet.macs, &packet,
310 if (noise_remote_begin_session (vm, &peer->remote))
312 wg_timers_session_derived (peer);
313 wg_timers_any_authenticated_packet_sent (peer);
314 wg_timers_any_authenticated_packet_traversal (peer);
317 u8 is_ip4 = ip46_address_is_ip4 (&peer->dst.addr);
318 if (!wg_create_buffer (vm, peer->rewrite, (u8 *) &packet,
319 sizeof (packet), &bi0, is_ip4))
322 ip46_enqueue_packet (vm, bi0, is_ip4);
331 wg_send_handshake_cookie (vlib_main_t *vm, u32 sender_index,
332 cookie_checker_t *cookie_checker,
333 message_macs_t *macs, ip46_address_t *wg_if_addr,
334 u16 wg_if_port, ip46_address_t *remote_addr,
337 message_handshake_cookie_t packet;
340 packet.header.type = MESSAGE_HANDSHAKE_COOKIE;
341 packet.receiver_index = sender_index;
343 cookie_checker_create_payload (vm, cookie_checker, macs, packet.nonce,
344 packet.encrypted_cookie, remote_addr,
348 u8 is_ip4 = ip46_address_is_ip4 (remote_addr);
350 rewrite = wg_build_rewrite (wg_if_addr, wg_if_port, remote_addr, remote_port,
353 ret = wg_create_buffer (vm, rewrite, (u8 *) &packet, sizeof (packet), &bi0,
359 ip46_enqueue_packet (vm, bi0, is_ip4);
365 * fd.io coding-style-patch-verification: ON
368 * eval: (c-set-style "gnu")