net: handle optional VLAN header in checksum computation.

Signed-off-by: Jean-Christophe Dubois <jcd@tribudubois.net>
Signed-off-by: Jason Wang <jasowang@redhat.com>
This commit is contained in:
Jean-Christophe Dubois 2016-05-30 19:25:48 +02:00 committed by Jason Wang
parent 50dbce6538
commit ade6bad111

View File

@ -55,7 +55,7 @@ uint16_t net_checksum_tcpudp(uint16_t length, uint16_t proto,
void net_checksum_calculate(uint8_t *data, int length) void net_checksum_calculate(uint8_t *data, int length)
{ {
int ip_len; int mac_hdr_len, ip_len;
struct ip_header *ip; struct ip_header *ip;
/* /*
@ -64,12 +64,39 @@ void net_checksum_calculate(uint8_t *data, int length)
* struct members (just in case). * struct members (just in case).
*/ */
/* Ensure data has complete L2 & L3 headers. */ /* Ensure we have at least an Eth header */
if (length < (sizeof(struct eth_header) + sizeof(struct ip_header))) { if (length < sizeof(struct eth_header)) {
return; return;
} }
ip = (struct ip_header *)(data + sizeof(struct eth_header)); /* Handle the optionnal VLAN headers */
switch (lduw_be_p(&PKT_GET_ETH_HDR(data)->h_proto)) {
case ETH_P_VLAN:
mac_hdr_len = sizeof(struct eth_header) +
sizeof(struct vlan_header);
break;
case ETH_P_DVLAN:
if (lduw_be_p(&PKT_GET_VLAN_HDR(data)->h_proto) == ETH_P_VLAN) {
mac_hdr_len = sizeof(struct eth_header) +
2 * sizeof(struct vlan_header);
} else {
mac_hdr_len = sizeof(struct eth_header) +
sizeof(struct vlan_header);
}
break;
default:
mac_hdr_len = sizeof(struct eth_header);
break;
}
length -= mac_hdr_len;
/* Now check we have an IP header (with an optionnal VLAN header) */
if (length < sizeof(struct ip_header)) {
return;
}
ip = (struct ip_header *)(data + mac_hdr_len);
if (IP_HEADER_VERSION(ip) != IP_HEADER_VERSION_4) { if (IP_HEADER_VERSION(ip) != IP_HEADER_VERSION_4) {
return; /* not IPv4 */ return; /* not IPv4 */