Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 1 | /* |
Sven Eckelmann | 64afe35 | 2011-01-27 10:38:15 +0100 | [diff] [blame] | 2 | * Copyright (C) 2010-2011 B.A.T.M.A.N. contributors: |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 3 | * |
| 4 | * Andreas Langer |
| 5 | * |
| 6 | * This program is free software; you can redistribute it and/or |
| 7 | * modify it under the terms of version 2 of the GNU General Public |
| 8 | * License as published by the Free Software Foundation. |
| 9 | * |
| 10 | * This program is distributed in the hope that it will be useful, but |
| 11 | * WITHOUT ANY WARRANTY; without even the implied warranty of |
| 12 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 13 | * General Public License for more details. |
| 14 | * |
| 15 | * You should have received a copy of the GNU General Public License |
| 16 | * along with this program; if not, write to the Free Software |
| 17 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA |
| 18 | * 02110-1301, USA |
| 19 | * |
| 20 | */ |
| 21 | |
| 22 | #include "main.h" |
| 23 | #include "unicast.h" |
| 24 | #include "send.h" |
| 25 | #include "soft-interface.h" |
| 26 | #include "gateway_client.h" |
| 27 | #include "originator.h" |
| 28 | #include "hash.h" |
| 29 | #include "translation-table.h" |
| 30 | #include "routing.h" |
| 31 | #include "hard-interface.h" |
| 32 | |
| 33 | |
| 34 | static struct sk_buff *frag_merge_packet(struct list_head *head, |
| 35 | struct frag_packet_list_entry *tfp, |
| 36 | struct sk_buff *skb) |
| 37 | { |
| 38 | struct unicast_frag_packet *up = |
| 39 | (struct unicast_frag_packet *)skb->data; |
| 40 | struct sk_buff *tmp_skb; |
| 41 | struct unicast_packet *unicast_packet; |
Marek Lindner | 1406206 | 2011-02-08 12:43:54 +0000 | [diff] [blame] | 42 | int hdr_len = sizeof(struct unicast_packet); |
| 43 | int uni_diff = sizeof(struct unicast_frag_packet) - hdr_len; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 44 | |
| 45 | /* set skb to the first part and tmp_skb to the second part */ |
| 46 | if (up->flags & UNI_FRAG_HEAD) { |
| 47 | tmp_skb = tfp->skb; |
| 48 | } else { |
| 49 | tmp_skb = skb; |
| 50 | skb = tfp->skb; |
| 51 | } |
| 52 | |
Sven Eckelmann | 531c9da | 2011-02-06 23:26:43 +0000 | [diff] [blame] | 53 | if (skb_linearize(skb) < 0 || skb_linearize(tmp_skb) < 0) |
| 54 | goto err; |
| 55 | |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 56 | skb_pull(tmp_skb, sizeof(struct unicast_frag_packet)); |
Sven Eckelmann | 531c9da | 2011-02-06 23:26:43 +0000 | [diff] [blame] | 57 | if (pskb_expand_head(skb, 0, tmp_skb->len, GFP_ATOMIC) < 0) |
| 58 | goto err; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 59 | |
| 60 | /* move free entry to end */ |
| 61 | tfp->skb = NULL; |
| 62 | tfp->seqno = 0; |
| 63 | list_move_tail(&tfp->list, head); |
| 64 | |
| 65 | memcpy(skb_put(skb, tmp_skb->len), tmp_skb->data, tmp_skb->len); |
| 66 | kfree_skb(tmp_skb); |
| 67 | |
| 68 | memmove(skb->data + uni_diff, skb->data, hdr_len); |
| 69 | unicast_packet = (struct unicast_packet *) skb_pull(skb, uni_diff); |
| 70 | unicast_packet->packet_type = BAT_UNICAST; |
| 71 | |
| 72 | return skb; |
Sven Eckelmann | 531c9da | 2011-02-06 23:26:43 +0000 | [diff] [blame] | 73 | |
| 74 | err: |
| 75 | /* free buffered skb, skb will be freed later */ |
| 76 | kfree_skb(tfp->skb); |
| 77 | return NULL; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 78 | } |
| 79 | |
| 80 | static void frag_create_entry(struct list_head *head, struct sk_buff *skb) |
| 81 | { |
| 82 | struct frag_packet_list_entry *tfp; |
| 83 | struct unicast_frag_packet *up = |
| 84 | (struct unicast_frag_packet *)skb->data; |
| 85 | |
| 86 | /* free and oldest packets stand at the end */ |
| 87 | tfp = list_entry((head)->prev, typeof(*tfp), list); |
| 88 | kfree_skb(tfp->skb); |
| 89 | |
| 90 | tfp->seqno = ntohs(up->seqno); |
| 91 | tfp->skb = skb; |
| 92 | list_move(&tfp->list, head); |
| 93 | return; |
| 94 | } |
| 95 | |
| 96 | static int frag_create_buffer(struct list_head *head) |
| 97 | { |
| 98 | int i; |
| 99 | struct frag_packet_list_entry *tfp; |
| 100 | |
| 101 | for (i = 0; i < FRAG_BUFFER_SIZE; i++) { |
| 102 | tfp = kmalloc(sizeof(struct frag_packet_list_entry), |
| 103 | GFP_ATOMIC); |
| 104 | if (!tfp) { |
| 105 | frag_list_free(head); |
| 106 | return -ENOMEM; |
| 107 | } |
| 108 | tfp->skb = NULL; |
| 109 | tfp->seqno = 0; |
| 110 | INIT_LIST_HEAD(&tfp->list); |
| 111 | list_add(&tfp->list, head); |
| 112 | } |
| 113 | |
| 114 | return 0; |
| 115 | } |
| 116 | |
| 117 | static struct frag_packet_list_entry *frag_search_packet(struct list_head *head, |
| 118 | struct unicast_frag_packet *up) |
| 119 | { |
| 120 | struct frag_packet_list_entry *tfp; |
| 121 | struct unicast_frag_packet *tmp_up = NULL; |
| 122 | uint16_t search_seqno; |
| 123 | |
| 124 | if (up->flags & UNI_FRAG_HEAD) |
| 125 | search_seqno = ntohs(up->seqno)+1; |
| 126 | else |
| 127 | search_seqno = ntohs(up->seqno)-1; |
| 128 | |
| 129 | list_for_each_entry(tfp, head, list) { |
| 130 | |
| 131 | if (!tfp->skb) |
| 132 | continue; |
| 133 | |
| 134 | if (tfp->seqno == ntohs(up->seqno)) |
| 135 | goto mov_tail; |
| 136 | |
| 137 | tmp_up = (struct unicast_frag_packet *)tfp->skb->data; |
| 138 | |
| 139 | if (tfp->seqno == search_seqno) { |
| 140 | |
| 141 | if ((tmp_up->flags & UNI_FRAG_HEAD) != |
| 142 | (up->flags & UNI_FRAG_HEAD)) |
| 143 | return tfp; |
| 144 | else |
| 145 | goto mov_tail; |
| 146 | } |
| 147 | } |
| 148 | return NULL; |
| 149 | |
| 150 | mov_tail: |
| 151 | list_move_tail(&tfp->list, head); |
| 152 | return NULL; |
| 153 | } |
| 154 | |
| 155 | void frag_list_free(struct list_head *head) |
| 156 | { |
| 157 | struct frag_packet_list_entry *pf, *tmp_pf; |
| 158 | |
| 159 | if (!list_empty(head)) { |
| 160 | |
| 161 | list_for_each_entry_safe(pf, tmp_pf, head, list) { |
| 162 | kfree_skb(pf->skb); |
| 163 | list_del(&pf->list); |
| 164 | kfree(pf); |
| 165 | } |
| 166 | } |
| 167 | return; |
| 168 | } |
| 169 | |
| 170 | /* frag_reassemble_skb(): |
| 171 | * returns NET_RX_DROP if the operation failed - skb is left intact |
| 172 | * returns NET_RX_SUCCESS if the fragment was buffered (skb_new will be NULL) |
| 173 | * or the skb could be reassembled (skb_new will point to the new packet and |
| 174 | * skb was freed) |
| 175 | */ |
| 176 | int frag_reassemble_skb(struct sk_buff *skb, struct bat_priv *bat_priv, |
| 177 | struct sk_buff **new_skb) |
| 178 | { |
| 179 | struct orig_node *orig_node; |
| 180 | struct frag_packet_list_entry *tmp_frag_entry; |
| 181 | int ret = NET_RX_DROP; |
| 182 | struct unicast_frag_packet *unicast_packet = |
| 183 | (struct unicast_frag_packet *)skb->data; |
| 184 | |
| 185 | *new_skb = NULL; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 186 | |
Marek Lindner | 7aadf88 | 2011-02-18 12:28:09 +0000 | [diff] [blame] | 187 | orig_node = orig_hash_find(bat_priv, unicast_packet->orig); |
| 188 | if (!orig_node) |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 189 | goto out; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 190 | |
| 191 | orig_node->last_frag_packet = jiffies; |
| 192 | |
| 193 | if (list_empty(&orig_node->frag_list) && |
| 194 | frag_create_buffer(&orig_node->frag_list)) { |
| 195 | pr_debug("couldn't create frag buffer\n"); |
| 196 | goto out; |
| 197 | } |
| 198 | |
| 199 | tmp_frag_entry = frag_search_packet(&orig_node->frag_list, |
| 200 | unicast_packet); |
| 201 | |
| 202 | if (!tmp_frag_entry) { |
| 203 | frag_create_entry(&orig_node->frag_list, skb); |
| 204 | ret = NET_RX_SUCCESS; |
| 205 | goto out; |
| 206 | } |
| 207 | |
| 208 | *new_skb = frag_merge_packet(&orig_node->frag_list, tmp_frag_entry, |
| 209 | skb); |
| 210 | /* if not, merge failed */ |
| 211 | if (*new_skb) |
| 212 | ret = NET_RX_SUCCESS; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 213 | |
Marek Lindner | 7aadf88 | 2011-02-18 12:28:09 +0000 | [diff] [blame] | 214 | out: |
| 215 | if (orig_node) |
Marek Lindner | 7b36e8e | 2011-02-18 12:28:10 +0000 | [diff] [blame] | 216 | orig_node_free_ref(orig_node); |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 217 | return ret; |
| 218 | } |
| 219 | |
| 220 | int frag_send_skb(struct sk_buff *skb, struct bat_priv *bat_priv, |
Marek Lindner | e6c10f4 | 2011-02-18 12:33:20 +0000 | [diff] [blame] | 221 | struct hard_iface *hard_iface, uint8_t dstaddr[]) |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 222 | { |
| 223 | struct unicast_packet tmp_uc, *unicast_packet; |
Marek Lindner | 32ae9b2 | 2011-04-20 15:40:58 +0200 | [diff] [blame] | 224 | struct hard_iface *primary_if; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 225 | struct sk_buff *frag_skb; |
| 226 | struct unicast_frag_packet *frag1, *frag2; |
| 227 | int uc_hdr_len = sizeof(struct unicast_packet); |
| 228 | int ucf_hdr_len = sizeof(struct unicast_frag_packet); |
Sven Eckelmann | 5c77d8b | 2011-01-25 21:59:26 +0000 | [diff] [blame] | 229 | int data_len = skb->len - uc_hdr_len; |
Marek Lindner | 32ae9b2 | 2011-04-20 15:40:58 +0200 | [diff] [blame] | 230 | int large_tail = 0, ret = NET_RX_DROP; |
Sven Eckelmann | c2f7f0e | 2011-02-10 14:33:56 +0000 | [diff] [blame] | 231 | uint16_t seqno; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 232 | |
Marek Lindner | 32ae9b2 | 2011-04-20 15:40:58 +0200 | [diff] [blame] | 233 | primary_if = primary_if_get_selected(bat_priv); |
| 234 | if (!primary_if) |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 235 | goto dropped; |
| 236 | |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 237 | frag_skb = dev_alloc_skb(data_len - (data_len / 2) + ucf_hdr_len); |
Jesper Juhl | ed7809d | 2011-01-13 21:53:38 +0100 | [diff] [blame] | 238 | if (!frag_skb) |
| 239 | goto dropped; |
Sven Eckelmann | 5c77d8b | 2011-01-25 21:59:26 +0000 | [diff] [blame] | 240 | skb_reserve(frag_skb, ucf_hdr_len); |
Jesper Juhl | ed7809d | 2011-01-13 21:53:38 +0100 | [diff] [blame] | 241 | |
| 242 | unicast_packet = (struct unicast_packet *) skb->data; |
| 243 | memcpy(&tmp_uc, unicast_packet, uc_hdr_len); |
Sven Eckelmann | 5c77d8b | 2011-01-25 21:59:26 +0000 | [diff] [blame] | 244 | skb_split(skb, frag_skb, data_len / 2 + uc_hdr_len); |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 245 | |
| 246 | if (my_skb_head_push(skb, ucf_hdr_len - uc_hdr_len) < 0 || |
| 247 | my_skb_head_push(frag_skb, ucf_hdr_len) < 0) |
| 248 | goto drop_frag; |
| 249 | |
| 250 | frag1 = (struct unicast_frag_packet *)skb->data; |
| 251 | frag2 = (struct unicast_frag_packet *)frag_skb->data; |
| 252 | |
| 253 | memcpy(frag1, &tmp_uc, sizeof(struct unicast_packet)); |
| 254 | |
| 255 | frag1->ttl--; |
| 256 | frag1->version = COMPAT_VERSION; |
| 257 | frag1->packet_type = BAT_UNICAST_FRAG; |
| 258 | |
Marek Lindner | 32ae9b2 | 2011-04-20 15:40:58 +0200 | [diff] [blame] | 259 | memcpy(frag1->orig, primary_if->net_dev->dev_addr, ETH_ALEN); |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 260 | memcpy(frag2, frag1, sizeof(struct unicast_frag_packet)); |
| 261 | |
Sven Eckelmann | ae361ce | 2011-01-25 22:02:31 +0000 | [diff] [blame] | 262 | if (data_len & 1) |
| 263 | large_tail = UNI_FRAG_LARGETAIL; |
| 264 | |
| 265 | frag1->flags = UNI_FRAG_HEAD | large_tail; |
| 266 | frag2->flags = large_tail; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 267 | |
Marek Lindner | e6c10f4 | 2011-02-18 12:33:20 +0000 | [diff] [blame] | 268 | seqno = atomic_add_return(2, &hard_iface->frag_seqno); |
Sven Eckelmann | c2f7f0e | 2011-02-10 14:33:56 +0000 | [diff] [blame] | 269 | frag1->seqno = htons(seqno - 1); |
| 270 | frag2->seqno = htons(seqno); |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 271 | |
Marek Lindner | e6c10f4 | 2011-02-18 12:33:20 +0000 | [diff] [blame] | 272 | send_skb_packet(skb, hard_iface, dstaddr); |
| 273 | send_skb_packet(frag_skb, hard_iface, dstaddr); |
Marek Lindner | 32ae9b2 | 2011-04-20 15:40:58 +0200 | [diff] [blame] | 274 | ret = NET_RX_SUCCESS; |
| 275 | goto out; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 276 | |
| 277 | drop_frag: |
| 278 | kfree_skb(frag_skb); |
| 279 | dropped: |
| 280 | kfree_skb(skb); |
Marek Lindner | 32ae9b2 | 2011-04-20 15:40:58 +0200 | [diff] [blame] | 281 | out: |
| 282 | if (primary_if) |
| 283 | hardif_free_ref(primary_if); |
| 284 | return ret; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 285 | } |
| 286 | |
| 287 | int unicast_send_skb(struct sk_buff *skb, struct bat_priv *bat_priv) |
| 288 | { |
| 289 | struct ethhdr *ethhdr = (struct ethhdr *)skb->data; |
| 290 | struct unicast_packet *unicast_packet; |
Marek Lindner | 7b36e8e | 2011-02-18 12:28:10 +0000 | [diff] [blame] | 291 | struct orig_node *orig_node; |
Marek Lindner | 44524fc | 2011-02-10 14:33:53 +0000 | [diff] [blame] | 292 | struct neigh_node *neigh_node; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 293 | int data_len = skb->len; |
Marek Lindner | 44524fc | 2011-02-10 14:33:53 +0000 | [diff] [blame] | 294 | int ret = 1; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 295 | |
| 296 | /* get routing information */ |
Linus Lüssing | 43c70ad | 2011-02-13 21:13:04 +0000 | [diff] [blame] | 297 | if (is_multicast_ether_addr(ethhdr->h_dest)) { |
Marek Lindner | c4aac1a | 2011-03-23 11:24:34 +0100 | [diff] [blame] | 298 | orig_node = (struct orig_node *)gw_get_selected_orig(bat_priv); |
Linus Lüssing | 43c70ad | 2011-02-13 21:13:04 +0000 | [diff] [blame] | 299 | if (orig_node) |
Marek Lindner | 44524fc | 2011-02-10 14:33:53 +0000 | [diff] [blame] | 300 | goto find_router; |
| 301 | } |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 302 | |
Antonio Quartulli | 2dafb49 | 2011-05-05 08:42:45 +0200 | [diff] [blame] | 303 | /* check for tt host - increases orig_node refcount */ |
Marek Lindner | 44524fc | 2011-02-10 14:33:53 +0000 | [diff] [blame] | 304 | orig_node = transtable_search(bat_priv, ethhdr->h_dest); |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 305 | |
Marek Lindner | 44524fc | 2011-02-10 14:33:53 +0000 | [diff] [blame] | 306 | find_router: |
Marek Lindner | d007260 | 2011-01-19 20:01:44 +0000 | [diff] [blame] | 307 | /** |
| 308 | * find_router(): |
| 309 | * - if orig_node is NULL it returns NULL |
| 310 | * - increases neigh_nodes refcount if found. |
| 311 | */ |
Marek Lindner | 44524fc | 2011-02-10 14:33:53 +0000 | [diff] [blame] | 312 | neigh_node = find_router(bat_priv, orig_node, NULL); |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 313 | |
Marek Lindner | 44524fc | 2011-02-10 14:33:53 +0000 | [diff] [blame] | 314 | if (!neigh_node) |
Marek Lindner | d007260 | 2011-01-19 20:01:44 +0000 | [diff] [blame] | 315 | goto out; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 316 | |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 317 | if (my_skb_head_push(skb, sizeof(struct unicast_packet)) < 0) |
Marek Lindner | d007260 | 2011-01-19 20:01:44 +0000 | [diff] [blame] | 318 | goto out; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 319 | |
| 320 | unicast_packet = (struct unicast_packet *)skb->data; |
| 321 | |
| 322 | unicast_packet->version = COMPAT_VERSION; |
| 323 | /* batman packet type: unicast */ |
| 324 | unicast_packet->packet_type = BAT_UNICAST; |
| 325 | /* set unicast ttl */ |
| 326 | unicast_packet->ttl = TTL; |
| 327 | /* copy the destination for faster routing */ |
| 328 | memcpy(unicast_packet->dest, orig_node->orig, ETH_ALEN); |
| 329 | |
| 330 | if (atomic_read(&bat_priv->fragmentation) && |
| 331 | data_len + sizeof(struct unicast_packet) > |
Marek Lindner | d007260 | 2011-01-19 20:01:44 +0000 | [diff] [blame] | 332 | neigh_node->if_incoming->net_dev->mtu) { |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 333 | /* send frag skb decreases ttl */ |
| 334 | unicast_packet->ttl++; |
Marek Lindner | d007260 | 2011-01-19 20:01:44 +0000 | [diff] [blame] | 335 | ret = frag_send_skb(skb, bat_priv, |
| 336 | neigh_node->if_incoming, neigh_node->addr); |
Marek Lindner | 44524fc | 2011-02-10 14:33:53 +0000 | [diff] [blame] | 337 | goto out; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 338 | } |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 339 | |
Marek Lindner | d007260 | 2011-01-19 20:01:44 +0000 | [diff] [blame] | 340 | send_skb_packet(skb, neigh_node->if_incoming, neigh_node->addr); |
Marek Lindner | 44524fc | 2011-02-10 14:33:53 +0000 | [diff] [blame] | 341 | ret = 0; |
| 342 | goto out; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 343 | |
Marek Lindner | 44524fc | 2011-02-10 14:33:53 +0000 | [diff] [blame] | 344 | out: |
| 345 | if (neigh_node) |
| 346 | neigh_node_free_ref(neigh_node); |
| 347 | if (orig_node) |
Marek Lindner | 7b36e8e | 2011-02-18 12:28:10 +0000 | [diff] [blame] | 348 | orig_node_free_ref(orig_node); |
Marek Lindner | 44524fc | 2011-02-10 14:33:53 +0000 | [diff] [blame] | 349 | if (ret == 1) |
| 350 | kfree_skb(skb); |
| 351 | return ret; |
Sven Eckelmann | c6c8fea | 2010-12-13 11:19:28 +0000 | [diff] [blame] | 352 | } |