Skip to content

L
linux
Commit 969e8e25 authored by David S. Miller's avatar David S. Miller
Browse files
Options

ipvs: Stop using NLA_PUT*(). 

These macros contain a hidden goto, and are thus extremely error
prone and make code hard to audit.
Signed-off-by: default avatarDavid S. Miller <davem@davemloft.net>
Acked-by: default avatarSimon Horman <horms@verge.net.au>
parent bae65be8
Show whitespace changes
Inline Side-by-side
Showing with 58 additions and 50 deletions
net/netfilter/ipvs/ip_vs_ctl.c
View file @ 969e8e25
...@@ -2816,17 +2816,17 @@ static int ip_vs_genl_fill_stats(struct sk_buff *skb, int container_type, ...@@ -2816,17 +2816,17 @@ static int ip_vs_genl_fill_stats(struct sk_buff *skb, int container_type,
ip_vs_copy_stats(&ustats, stats); ip_vs_copy_stats(&ustats, stats);
NLA_PUT_U32(skb, IPVS_STATS_ATTR_CONNS, ustats.conns); if (nla_put_u32(skb, IPVS_STATS_ATTR_CONNS, ustats.conns) ||
NLA_PUT_U32(skb, IPVS_STATS_ATTR_INPKTS, ustats.inpkts); nla_put_u32(skb, IPVS_STATS_ATTR_INPKTS, ustats.inpkts) ||
NLA_PUT_U32(skb, IPVS_STATS_ATTR_OUTPKTS, ustats.outpkts); nla_put_u32(skb, IPVS_STATS_ATTR_OUTPKTS, ustats.outpkts) ||
NLA_PUT_U64(skb, IPVS_STATS_ATTR_INBYTES, ustats.inbytes); nla_put_u64(skb, IPVS_STATS_ATTR_INBYTES, ustats.inbytes) ||
NLA_PUT_U64(skb, IPVS_STATS_ATTR_OUTBYTES, ustats.outbytes); nla_put_u64(skb, IPVS_STATS_ATTR_OUTBYTES, ustats.outbytes) ||
NLA_PUT_U32(skb, IPVS_STATS_ATTR_CPS, ustats.cps); nla_put_u32(skb, IPVS_STATS_ATTR_CPS, ustats.cps) ||
NLA_PUT_U32(skb, IPVS_STATS_ATTR_INPPS, ustats.inpps); nla_put_u32(skb, IPVS_STATS_ATTR_INPPS, ustats.inpps) ||
NLA_PUT_U32(skb, IPVS_STATS_ATTR_OUTPPS, ustats.outpps); nla_put_u32(skb, IPVS_STATS_ATTR_OUTPPS, ustats.outpps) ||
NLA_PUT_U32(skb, IPVS_STATS_ATTR_INBPS, ustats.inbps); nla_put_u32(skb, IPVS_STATS_ATTR_INBPS, ustats.inbps) ||
NLA_PUT_U32(skb, IPVS_STATS_ATTR_OUTBPS, ustats.outbps); nla_put_u32(skb, IPVS_STATS_ATTR_OUTBPS, ustats.outbps))
goto nla_put_failure;
nla_nest_end(skb, nl_stats); nla_nest_end(skb, nl_stats);
return 0; return 0;
...@@ -2847,23 +2847,25 @@ static int ip_vs_genl_fill_service(struct sk_buff *skb, ...@@ -2847,23 +2847,25 @@ static int ip_vs_genl_fill_service(struct sk_buff *skb,
if (!nl_service) if (!nl_service)
return -EMSGSIZE; return -EMSGSIZE;
NLA_PUT_U16(skb, IPVS_SVC_ATTR_AF, svc->af); if (nla_put_u16(skb, IPVS_SVC_ATTR_AF, svc->af))
goto nla_put_failure;
if (svc->fwmark) { if (svc->fwmark) {
NLA_PUT_U32(skb, IPVS_SVC_ATTR_FWMARK, svc->fwmark); if (nla_put_u32(skb, IPVS_SVC_ATTR_FWMARK, svc->fwmark))
goto nla_put_failure;
} else { } else {
NLA_PUT_U16(skb, IPVS_SVC_ATTR_PROTOCOL, svc->protocol); if (nla_put_u16(skb, IPVS_SVC_ATTR_PROTOCOL, svc->protocol) ||
NLA_PUT(skb, IPVS_SVC_ATTR_ADDR, sizeof(svc->addr), &svc->addr); nla_put(skb, IPVS_SVC_ATTR_ADDR, sizeof(svc->addr), &svc->addr) ||
NLA_PUT_U16(skb, IPVS_SVC_ATTR_PORT, svc->port); nla_put_u16(skb, IPVS_SVC_ATTR_PORT, svc->port))
goto nla_put_failure;
} }
NLA_PUT_STRING(skb, IPVS_SVC_ATTR_SCHED_NAME, svc->scheduler->name); if (nla_put_string(skb, IPVS_SVC_ATTR_SCHED_NAME, svc->scheduler->name) ||
if (svc->pe) (svc->pe &&
NLA_PUT_STRING(skb, IPVS_SVC_ATTR_PE_NAME, svc->pe->name); nla_put_string(skb, IPVS_SVC_ATTR_PE_NAME, svc->pe->name)) ||
NLA_PUT(skb, IPVS_SVC_ATTR_FLAGS, sizeof(flags), &flags); nla_put(skb, IPVS_SVC_ATTR_FLAGS, sizeof(flags), &flags) ||
NLA_PUT_U32(skb, IPVS_SVC_ATTR_TIMEOUT, svc->timeout / HZ); nla_put_u32(skb, IPVS_SVC_ATTR_TIMEOUT, svc->timeout / HZ) ||
NLA_PUT_U32(skb, IPVS_SVC_ATTR_NETMASK, svc->netmask); nla_put_u32(skb, IPVS_SVC_ATTR_NETMASK, svc->netmask))
goto nla_put_failure;
if (ip_vs_genl_fill_stats(skb, IPVS_SVC_ATTR_STATS, &svc->stats)) if (ip_vs_genl_fill_stats(skb, IPVS_SVC_ATTR_STATS, &svc->stats))
goto nla_put_failure; goto nla_put_failure;
...@@ -3038,21 +3040,22 @@ static int ip_vs_genl_fill_dest(struct sk_buff *skb, struct ip_vs_dest *dest) ...@@ -3038,21 +3040,22 @@ static int ip_vs_genl_fill_dest(struct sk_buff *skb, struct ip_vs_dest *dest)
if (!nl_dest) if (!nl_dest)
return -EMSGSIZE; return -EMSGSIZE;
NLA_PUT(skb, IPVS_DEST_ATTR_ADDR, sizeof(dest->addr), &dest->addr); if (nla_put(skb, IPVS_DEST_ATTR_ADDR, sizeof(dest->addr), &dest->addr) ||
NLA_PUT_U16(skb, IPVS_DEST_ATTR_PORT, dest->port); nla_put_u16(skb, IPVS_DEST_ATTR_PORT, dest->port) ||
nla_put_u32(skb, IPVS_DEST_ATTR_FWD_METHOD,
NLA_PUT_U32(skb, IPVS_DEST_ATTR_FWD_METHOD, (atomic_read(&dest->conn_flags) &
atomic_read(&dest->conn_flags) & IP_VS_CONN_F_FWD_MASK); IP_VS_CONN_F_FWD_MASK)) ||
NLA_PUT_U32(skb, IPVS_DEST_ATTR_WEIGHT, atomic_read(&dest->weight)); nla_put_u32(skb, IPVS_DEST_ATTR_WEIGHT,
NLA_PUT_U32(skb, IPVS_DEST_ATTR_U_THRESH, dest->u_threshold); atomic_read(&dest->weight)) ||
NLA_PUT_U32(skb, IPVS_DEST_ATTR_L_THRESH, dest->l_threshold); nla_put_u32(skb, IPVS_DEST_ATTR_U_THRESH, dest->u_threshold) ||
NLA_PUT_U32(skb, IPVS_DEST_ATTR_ACTIVE_CONNS, nla_put_u32(skb, IPVS_DEST_ATTR_L_THRESH, dest->l_threshold) ||
atomic_read(&dest->activeconns)); nla_put_u32(skb, IPVS_DEST_ATTR_ACTIVE_CONNS,
NLA_PUT_U32(skb, IPVS_DEST_ATTR_INACT_CONNS, atomic_read(&dest->activeconns)) ||
atomic_read(&dest->inactconns)); nla_put_u32(skb, IPVS_DEST_ATTR_INACT_CONNS,
NLA_PUT_U32(skb, IPVS_DEST_ATTR_PERSIST_CONNS, atomic_read(&dest->inactconns)) ||
atomic_read(&dest->persistconns)); nla_put_u32(skb, IPVS_DEST_ATTR_PERSIST_CONNS,
atomic_read(&dest->persistconns)))
goto nla_put_failure;
if (ip_vs_genl_fill_stats(skb, IPVS_DEST_ATTR_STATS, &dest->stats)) if (ip_vs_genl_fill_stats(skb, IPVS_DEST_ATTR_STATS, &dest->stats))
goto nla_put_failure; goto nla_put_failure;
...@@ -3181,10 +3184,10 @@ static int ip_vs_genl_fill_daemon(struct sk_buff *skb, __be32 state, ...@@ -3181,10 +3184,10 @@ static int ip_vs_genl_fill_daemon(struct sk_buff *skb, __be32 state,
if (!nl_daemon) if (!nl_daemon)
return -EMSGSIZE; return -EMSGSIZE;
NLA_PUT_U32(skb, IPVS_DAEMON_ATTR_STATE, state); if (nla_put_u32(skb, IPVS_DAEMON_ATTR_STATE, state) ||
NLA_PUT_STRING(skb, IPVS_DAEMON_ATTR_MCAST_IFN, mcast_ifn); nla_put_string(skb, IPVS_DAEMON_ATTR_MCAST_IFN, mcast_ifn) ||
NLA_PUT_U32(skb, IPVS_DAEMON_ATTR_SYNC_ID, syncid); nla_put_u32(skb, IPVS_DAEMON_ATTR_SYNC_ID, syncid))
goto nla_put_failure;
nla_nest_end(skb, nl_daemon); nla_nest_end(skb, nl_daemon);
return 0; return 0;
...@@ -3473,21 +3476,26 @@ static int ip_vs_genl_get_cmd(struct sk_buff *skb, struct genl_info *info) ...@@ -3473,21 +3476,26 @@ static int ip_vs_genl_get_cmd(struct sk_buff *skb, struct genl_info *info)
__ip_vs_get_timeouts(net, &t); __ip_vs_get_timeouts(net, &t);
#ifdef CONFIG_IP_VS_PROTO_TCP #ifdef CONFIG_IP_VS_PROTO_TCP
NLA_PUT_U32(msg, IPVS_CMD_ATTR_TIMEOUT_TCP, t.tcp_timeout); if (nla_put_u32(msg, IPVS_CMD_ATTR_TIMEOUT_TCP,
NLA_PUT_U32(msg, IPVS_CMD_ATTR_TIMEOUT_TCP_FIN, t.tcp_timeout) ||
t.tcp_fin_timeout); nla_put_u32(msg, IPVS_CMD_ATTR_TIMEOUT_TCP_FIN,
t.tcp_fin_timeout))
goto nla_put_failure;
#endif #endif
#ifdef CONFIG_IP_VS_PROTO_UDP #ifdef CONFIG_IP_VS_PROTO_UDP
NLA_PUT_U32(msg, IPVS_CMD_ATTR_TIMEOUT_UDP, t.udp_timeout); if (nla_put_u32(msg, IPVS_CMD_ATTR_TIMEOUT_UDP, t.udp_timeout))
goto nla_put_failure;
#endif #endif
break; break;
} }
case IPVS_CMD_GET_INFO: case IPVS_CMD_GET_INFO:
NLA_PUT_U32(msg, IPVS_INFO_ATTR_VERSION, IP_VS_VERSION_CODE); if (nla_put_u32(msg, IPVS_INFO_ATTR_VERSION,
NLA_PUT_U32(msg, IPVS_INFO_ATTR_CONN_TAB_SIZE, IP_VS_VERSION_CODE) ||
ip_vs_conn_tab_size); nla_put_u32(msg, IPVS_INFO_ATTR_CONN_TAB_SIZE,
ip_vs_conn_tab_size))
goto nla_put_failure;
break; break;
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment
GitLab Nexedi Edition | About GitLab | About Nexedi | 沪ICP备2021021310号-2 | 沪ICP备2021021310号-7