ipv6: sr: restruct ifdefines

There are too many ifdef in IPv6 segment routing code that may cause logic
problems. like commit 160e9d2752 ("ipv6: sr: fix invalid unregister error
path"). To avoid this, the init functions are redefined for both cases. The
code could be more clear after all fidefs are removed.

Suggested-by: Simon Horman <horms@kernel.org>
Suggested-by: David Ahern <dsahern@kernel.org>
Signed-off-by: Hangbin Liu <liuhangbin@gmail.com>
Reviewed-by: Sabrina Dubroca <sd@queasysnail.net>
Reviewed-by: David Ahern <dsahern@kernel.org>
Link: https://lore.kernel.org/r/20240529040908.3472952-1-liuhangbin@gmail.com
Signed-off-by: Jakub Kicinski <kuba@kernel.org>
This commit is contained in:
Hangbin Liu 2024-05-29 12:09:08 +08:00 committed by Jakub Kicinski
parent 57e3c5af2b
commit a79d8fe2ff
3 changed files with 20 additions and 29 deletions

View File

@ -52,10 +52,17 @@ static inline struct seg6_pernet_data *seg6_pernet(struct net *net)
extern int seg6_init(void); extern int seg6_init(void);
extern void seg6_exit(void); extern void seg6_exit(void);
#ifdef CONFIG_IPV6_SEG6_LWTUNNEL
extern int seg6_iptunnel_init(void); extern int seg6_iptunnel_init(void);
extern void seg6_iptunnel_exit(void); extern void seg6_iptunnel_exit(void);
extern int seg6_local_init(void); extern int seg6_local_init(void);
extern void seg6_local_exit(void); extern void seg6_local_exit(void);
#else
static inline int seg6_iptunnel_init(void) { return 0; }
static inline void seg6_iptunnel_exit(void) {}
static inline int seg6_local_init(void) { return 0; }
static inline void seg6_local_exit(void) {}
#endif
extern bool seg6_validate_srh(struct ipv6_sr_hdr *srh, int len, bool reduced); extern bool seg6_validate_srh(struct ipv6_sr_hdr *srh, int len, bool reduced);
extern struct ipv6_sr_hdr *seg6_get_srh(struct sk_buff *skb, int flags); extern struct ipv6_sr_hdr *seg6_get_srh(struct sk_buff *skb, int flags);

View File

@ -49,9 +49,16 @@ extern int seg6_hmac_info_del(struct net *net, u32 key);
extern int seg6_push_hmac(struct net *net, struct in6_addr *saddr, extern int seg6_push_hmac(struct net *net, struct in6_addr *saddr,
struct ipv6_sr_hdr *srh); struct ipv6_sr_hdr *srh);
extern bool seg6_hmac_validate_skb(struct sk_buff *skb); extern bool seg6_hmac_validate_skb(struct sk_buff *skb);
#ifdef CONFIG_IPV6_SEG6_HMAC
extern int seg6_hmac_init(void); extern int seg6_hmac_init(void);
extern void seg6_hmac_exit(void); extern void seg6_hmac_exit(void);
extern int seg6_hmac_net_init(struct net *net); extern int seg6_hmac_net_init(struct net *net);
extern void seg6_hmac_net_exit(struct net *net); extern void seg6_hmac_net_exit(struct net *net);
#else
static inline int seg6_hmac_init(void) { return 0; }
static inline void seg6_hmac_exit(void) {}
static inline int seg6_hmac_net_init(struct net *net) { return 0; }
static inline void seg6_hmac_net_exit(struct net *net) {}
#endif
#endif #endif

View File

@ -21,9 +21,7 @@
#include <net/genetlink.h> #include <net/genetlink.h>
#include <linux/seg6.h> #include <linux/seg6.h>
#include <linux/seg6_genl.h> #include <linux/seg6_genl.h>
#ifdef CONFIG_IPV6_SEG6_HMAC
#include <net/seg6_hmac.h> #include <net/seg6_hmac.h>
#endif
bool seg6_validate_srh(struct ipv6_sr_hdr *srh, int len, bool reduced) bool seg6_validate_srh(struct ipv6_sr_hdr *srh, int len, bool reduced)
{ {
@ -437,13 +435,11 @@ static int __net_init seg6_net_init(struct net *net)
net->ipv6.seg6_data = sdata; net->ipv6.seg6_data = sdata;
#ifdef CONFIG_IPV6_SEG6_HMAC
if (seg6_hmac_net_init(net)) { if (seg6_hmac_net_init(net)) {
kfree(rcu_dereference_raw(sdata->tun_src)); kfree(rcu_dereference_raw(sdata->tun_src));
kfree(sdata); kfree(sdata);
return -ENOMEM; return -ENOMEM;
} }
#endif
return 0; return 0;
} }
@ -452,9 +448,7 @@ static void __net_exit seg6_net_exit(struct net *net)
{ {
struct seg6_pernet_data *sdata = seg6_pernet(net); struct seg6_pernet_data *sdata = seg6_pernet(net);
#ifdef CONFIG_IPV6_SEG6_HMAC
seg6_hmac_net_exit(net); seg6_hmac_net_exit(net);
#endif
kfree(rcu_dereference_raw(sdata->tun_src)); kfree(rcu_dereference_raw(sdata->tun_src));
kfree(sdata); kfree(sdata);
@ -520,41 +514,28 @@ int __init seg6_init(void)
if (err) if (err)
goto out_unregister_pernet; goto out_unregister_pernet;
#ifdef CONFIG_IPV6_SEG6_LWTUNNEL
err = seg6_iptunnel_init(); err = seg6_iptunnel_init();
if (err) if (err)
goto out_unregister_genl; goto out_unregister_genl;
err = seg6_local_init(); err = seg6_local_init();
if (err) {
seg6_iptunnel_exit();
goto out_unregister_genl;
}
#endif
#ifdef CONFIG_IPV6_SEG6_HMAC
err = seg6_hmac_init();
if (err) if (err)
goto out_unregister_iptun; goto out_unregister_iptun;
#endif
err = seg6_hmac_init();
if (err)
goto out_unregister_seg6;
pr_info("Segment Routing with IPv6\n"); pr_info("Segment Routing with IPv6\n");
out: out:
return err; return err;
#ifdef CONFIG_IPV6_SEG6_HMAC out_unregister_seg6:
out_unregister_iptun:
#ifdef CONFIG_IPV6_SEG6_LWTUNNEL
seg6_local_exit(); seg6_local_exit();
out_unregister_iptun:
seg6_iptunnel_exit(); seg6_iptunnel_exit();
#endif
#endif
#ifdef CONFIG_IPV6_SEG6_LWTUNNEL
out_unregister_genl: out_unregister_genl:
#endif
#if IS_ENABLED(CONFIG_IPV6_SEG6_LWTUNNEL) || IS_ENABLED(CONFIG_IPV6_SEG6_HMAC)
genl_unregister_family(&seg6_genl_family); genl_unregister_family(&seg6_genl_family);
#endif
out_unregister_pernet: out_unregister_pernet:
unregister_pernet_subsys(&ip6_segments_ops); unregister_pernet_subsys(&ip6_segments_ops);
goto out; goto out;
@ -562,13 +543,9 @@ out_unregister_pernet:
void seg6_exit(void) void seg6_exit(void)
{ {
#ifdef CONFIG_IPV6_SEG6_HMAC
seg6_hmac_exit(); seg6_hmac_exit();
#endif
#ifdef CONFIG_IPV6_SEG6_LWTUNNEL
seg6_local_exit(); seg6_local_exit();
seg6_iptunnel_exit(); seg6_iptunnel_exit();
#endif
genl_unregister_family(&seg6_genl_family); genl_unregister_family(&seg6_genl_family);
unregister_pernet_subsys(&ip6_segments_ops); unregister_pernet_subsys(&ip6_segments_ops);
} }