Branch data     Line data    Source code

       1                 :            : /* SPDX-License-Identifier: GPL-2.0 */
       2                 :            : #ifndef __SOCK_DIAG_H__
       3                 :            : #define __SOCK_DIAG_H__
       4                 :            : 
       5                 :            : #include <linux/netlink.h>
       6                 :            : #include <linux/user_namespace.h>
       7                 :            : #include <net/net_namespace.h>
       8                 :            : #include <net/sock.h>
       9                 :            : #include <uapi/linux/sock_diag.h>
      10                 :            : 
      11                 :            : struct sk_buff;
      12                 :            : struct nlmsghdr;
      13                 :            : struct sock;
      14                 :            : 
      15                 :            : struct sock_diag_handler {
      16                 :            :         __u8 family;
      17                 :            :         int (*dump)(struct sk_buff *skb, struct nlmsghdr *nlh);
      18                 :            :         int (*get_info)(struct sk_buff *skb, struct sock *sk);
      19                 :            :         int (*destroy)(struct sk_buff *skb, struct nlmsghdr *nlh);
      20                 :            : };
      21                 :            : 
      22                 :            : int sock_diag_register(const struct sock_diag_handler *h);
      23                 :            : void sock_diag_unregister(const struct sock_diag_handler *h);
      24                 :            : 
      25                 :            : void sock_diag_register_inet_compat(int (*fn)(struct sk_buff *skb, struct nlmsghdr *nlh));
      26                 :            : void sock_diag_unregister_inet_compat(int (*fn)(struct sk_buff *skb, struct nlmsghdr *nlh));
      27                 :            : 
      28                 :            : u64 sock_gen_cookie(struct sock *sk);
      29                 :            : int sock_diag_check_cookie(struct sock *sk, const __u32 *cookie);
      30                 :            : void sock_diag_save_cookie(struct sock *sk, __u32 *cookie);
      31                 :            : 
      32                 :            : int sock_diag_put_meminfo(struct sock *sk, struct sk_buff *skb, int attr);
      33                 :            : int sock_diag_put_filterinfo(bool may_report_filterinfo, struct sock *sk,
      34                 :            :                              struct sk_buff *skb, int attrtype);
      35                 :            : 
      36                 :            : static inline
      37                 :          3 : enum sknetlink_groups sock_diag_destroy_group(const struct sock *sk)
      38                 :            : {
      39                 :          3 :         switch (sk->sk_family) {
      40                 :            :         case AF_INET:
      41                 :          3 :                 if (sk->sk_type == SOCK_RAW)
      42                 :            :                         return SKNLGRP_NONE;
      43                 :            : 
      44                 :          3 :                 switch (sk->sk_protocol) {
      45                 :            :                 case IPPROTO_TCP:
      46                 :            :                         return SKNLGRP_INET_TCP_DESTROY;
      47                 :            :                 case IPPROTO_UDP:
      48                 :          3 :                         return SKNLGRP_INET_UDP_DESTROY;
      49                 :            :                 default:
      50                 :          0 :                         return SKNLGRP_NONE;
      51                 :            :                 }
      52                 :            :         case AF_INET6:
      53                 :          3 :                 if (sk->sk_type == SOCK_RAW)
      54                 :            :                         return SKNLGRP_NONE;
      55                 :            : 
      56                 :          3 :                 switch (sk->sk_protocol) {
      57                 :            :                 case IPPROTO_TCP:
      58                 :            :                         return SKNLGRP_INET6_TCP_DESTROY;
      59                 :            :                 case IPPROTO_UDP:
      60                 :          3 :                         return SKNLGRP_INET6_UDP_DESTROY;
      61                 :            :                 default:
      62                 :          0 :                         return SKNLGRP_NONE;
      63                 :            :                 }
      64                 :            :         default:
      65                 :            :                 return SKNLGRP_NONE;
      66                 :            :         }
      67                 :            : }
      68                 :            : 
      69                 :            : static inline
      70                 :          3 : bool sock_diag_has_destroy_listeners(const struct sock *sk)
      71                 :            : {
      72                 :            :         const struct net *n = sock_net(sk);
      73                 :          3 :         const enum sknetlink_groups group = sock_diag_destroy_group(sk);
      74                 :            : 
      75                 :          3 :         return group != SKNLGRP_NONE && n->diag_nlsk &&
      76                 :          3 :                 netlink_has_listeners(n->diag_nlsk, group);
      77                 :            : }
      78                 :            : void sock_diag_broadcast_destroy(struct sock *sk);
      79                 :            : 
      80                 :            : int sock_diag_destroy(struct sock *sk, int err);
      81                 :            : #endif