| /* |
| * Copyright (c) 2014-2015, The Linux Foundation. All rights reserved. |
| * |
| * This program is free software; you can redistribute it and/or modify |
| * it under the terms of the GNU General Public License version 2 and |
| * only version 2 as published by the Free Software Foundation. |
| * |
| * This program is distributed in the hope that it will be useful, |
| * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
| * GNU General Public License for more details. |
| * |
| * |
| * Default SOCKEV client implementation |
| * |
| */ |
| |
| #include <linux/module.h> |
| #include <linux/kernel.h> |
| #include <linux/export.h> |
| #include <linux/netlink.h> |
| #include <linux/sockev.h> |
| #include <net/sock.h> |
| |
| static int registration_status; |
| static struct sock *socknlmsgsk; |
| |
| static void sockev_skmsg_recv(struct sk_buff *skb) |
| { |
| pr_debug("%s(): Got unsolicited request\n", __func__); |
| } |
| |
| static struct netlink_kernel_cfg nlcfg = { |
| .input = sockev_skmsg_recv |
| }; |
| |
| static void _sockev_event(unsigned long event, __u8 *evstr, int buflen) |
| { |
| switch (event) { |
| case SOCKEV_SOCKET: |
| strlcpy(evstr, "SOCKEV_SOCKET", buflen); |
| break; |
| case SOCKEV_BIND: |
| strlcpy(evstr, "SOCKEV_BIND", buflen); |
| break; |
| case SOCKEV_LISTEN: |
| strlcpy(evstr, "SOCKEV_LISTEN", buflen); |
| break; |
| case SOCKEV_ACCEPT: |
| strlcpy(evstr, "SOCKEV_ACCEPT", buflen); |
| break; |
| case SOCKEV_CONNECT: |
| strlcpy(evstr, "SOCKEV_CONNECT", buflen); |
| break; |
| case SOCKEV_SHUTDOWN: |
| strlcpy(evstr, "SOCKEV_SHUTDOWN", buflen); |
| break; |
| default: |
| strlcpy(evstr, "UNKOWN", buflen); |
| } |
| } |
| |
| static int sockev_client_cb(struct notifier_block *nb, |
| unsigned long event, void *data) |
| { |
| |
| struct sk_buff *skb; |
| struct nlmsghdr *nlh; |
| struct sknlsockevmsg *smsg; |
| struct socket *sock; |
| |
| sock = (struct socket *)data; |
| if (socknlmsgsk == 0) |
| goto done; |
| if ((socknlmsgsk == NULL) || (sock == NULL) || (sock->sk == NULL)) |
| goto done; |
| |
| if (sock->sk->sk_family != AF_INET && sock->sk->sk_family != AF_INET6) |
| goto done; |
| |
| skb = nlmsg_new(sizeof(struct sknlsockevmsg), GFP_KERNEL); |
| if (skb == NULL) |
| goto done; |
| |
| nlh = nlmsg_put(skb, 0, 0, event, sizeof(struct sknlsockevmsg), 0); |
| if (nlh == NULL) { |
| kfree_skb(skb); |
| goto done; |
| } |
| |
| NETLINK_CB(skb).dst_group = SKNLGRP_SOCKEV; |
| |
| smsg = nlmsg_data(nlh); |
| smsg->pid = current->pid; |
| _sockev_event(event, smsg->event, sizeof(smsg->event)); |
| smsg->skfamily = sock->sk->sk_family; |
| smsg->skstate = sock->sk->sk_state; |
| smsg->skprotocol = sock->sk->sk_protocol; |
| smsg->sktype = sock->sk->sk_type; |
| smsg->skflags = sock->sk->sk_flags; |
| |
| nlmsg_notify(socknlmsgsk, skb, 0, SKNLGRP_SOCKEV, 0, GFP_KERNEL); |
| done: |
| return 0; |
| } |
| |
| static struct notifier_block sockev_notifier_client = { |
| .notifier_call = sockev_client_cb, |
| .next = 0, |
| .priority = 0 |
| }; |
| |
| /* ***************** Startup/Shutdown *************************************** */ |
| |
| static int __init sockev_client_init(void) |
| { |
| int rc; |
| registration_status = 1; |
| rc = sockev_register_notify(&sockev_notifier_client); |
| if (rc != 0) { |
| registration_status = 0; |
| pr_err("%s(): Failed to register cb (%d)\n", __func__, rc); |
| } |
| socknlmsgsk = netlink_kernel_create(&init_net, NETLINK_SOCKEV, &nlcfg); |
| if (!socknlmsgsk) { |
| pr_err("%s(): Failed to initialize netlink socket\n", __func__); |
| if (registration_status) |
| sockev_unregister_notify(&sockev_notifier_client); |
| registration_status = 0; |
| } |
| |
| return rc; |
| } |
| static void __exit sockev_client_exit(void) |
| { |
| if (registration_status) |
| sockev_unregister_notify(&sockev_notifier_client); |
| } |
| module_init(sockev_client_init) |
| module_exit(sockev_client_exit) |
| MODULE_LICENSE("GPL v2"); |
| |