|
|
@@ -269,14 +269,20 @@ ip_vs_sh_get_port(const struct sk_buff *skb, struct ip_vs_iphdr *iph)
|
|
|
switch (iph->protocol) {
|
|
|
case IPPROTO_TCP:
|
|
|
th = skb_header_pointer(skb, iph->len, sizeof(_tcph), &_tcph);
|
|
|
+ if (unlikely(th == NULL))
|
|
|
+ return 0;
|
|
|
port = th->source;
|
|
|
break;
|
|
|
case IPPROTO_UDP:
|
|
|
uh = skb_header_pointer(skb, iph->len, sizeof(_udph), &_udph);
|
|
|
+ if (unlikely(uh == NULL))
|
|
|
+ return 0;
|
|
|
port = uh->source;
|
|
|
break;
|
|
|
case IPPROTO_SCTP:
|
|
|
sh = skb_header_pointer(skb, iph->len, sizeof(_sctph), &_sctph);
|
|
|
+ if (unlikely(sh == NULL))
|
|
|
+ return 0;
|
|
|
port = sh->source;
|
|
|
break;
|
|
|
default:
|