security/ccsecurity/network.c

/*
 * security/ccsecurity/network.c
 *
 * Copyright (C) 2005-2009  NTT DATA CORPORATION
 *
 * Version: 1.7.0-pre   2009/08/24
 *
 * This file is applicable to both 2.4.30 and 2.6.11 and later.
 * See README.ccs for ChangeLog.
 *
 */

#include <linux/net.h>
#include <linux/inet.h>
#include <linux/in.h>
#include <linux/in6.h>
#include <net/ip.h>
#include <net/ipv6.h>
#include <net/udp.h>
#include "internal.h"

/**
 * ccs_audit_network_log - Audit network log.
 *
 * @r:          Pointer to "struct ccs_request_info".
 * @operation:  The name of operation.
 * @address:    An IPv4 or IPv6 address.
 * @port:       Port number.
 * @is_granted: True if this is a granted log.
 *
 * Returns 0 on success, negative value otherwise.
 */
static int ccs_audit_network_log(struct ccs_request_info *r,
                                 const char *operation, const char *address,
                                 const u16 port, const bool is_granted)
{
        if (!is_granted)
                ccs_warn_log(r, "%s %s %u", operation, address, port);
        return ccs_write_audit_log(is_granted, r, CCS_KEYWORD_ALLOW_NETWORK
                                   "%s %s %u\n", operation, address, port);
}

/**
 * ccs_parse_ip_address - Parse an IP address.
 *
 * @address: String to parse.
 * @min:     Pointer to store min address.
 * @max:     Pointer to store max address.
 *
 * Returns 2 if @address is an IPv6, 1 if @address is an IPv4, 0 otherwise.
 */
int ccs_parse_ip_address(char *address, u16 *min, u16 *max)
{
        int count = sscanf(address, "%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx"
                           "-%hx:%hx:%hx:%hx:%hx:%hx:%hx:%hx",
                           &min[0], &min[1], &min[2], &min[3],
                           &min[4], &min[5], &min[6], &min[7],
                           &max[0], &max[1], &max[2], &max[3],
                           &max[4], &max[5], &max[6], &max[7]);
        if (count == 8 || count == 16) {
                u8 i;
                if (count == 8)
                        memmove(max, min, sizeof(u16) * 8);
                for (i = 0; i < 8; i++) {
                        min[i] = htons(min[i]);
                        max[i] = htons(max[i]);
                }
                return 2;
        }
        count = sscanf(address, "%hu.%hu.%hu.%hu-%hu.%hu.%hu.%hu",
                       &min[0], &min[1], &min[2], &min[3],
                       &max[0], &max[1], &max[2], &max[3]);
        if (count == 4 || count == 8) {
                u32 ip = htonl((((u8) min[0]) << 24) + (((u8) min[1]) << 16)
                               + (((u8) min[2]) << 8) + (u8) min[3]);
                memmove(min, &ip, sizeof(ip));
                if (count == 8)
                        ip = htonl((((u8) max[0]) << 24) + (((u8) max[1]) << 16)
                                   + (((u8) max[2]) << 8) + (u8) max[3]);
                memmove(max, &ip, sizeof(ip));
                return 1;
        }
        return 0;
}

#if !defined(NIP6)
#define NIP6(addr)                                                      \
        ntohs((addr).s6_addr16[0]), ntohs((addr).s6_addr16[1]),         \
                ntohs((addr).s6_addr16[2]), ntohs((addr).s6_addr16[3]), \
                ntohs((addr).s6_addr16[4]), ntohs((addr).s6_addr16[5]), \
                ntohs((addr).s6_addr16[6]), ntohs((addr).s6_addr16[7])
#endif

/**
 * ccs_print_ipv6 - Print an IPv6 address.
 *
 * @buffer:     Buffer to write to.
 * @buffer_len: Size of @buffer.
 * @ip:         Pointer to "struct in6_addr".
 *
 * Returns nothing.
 */
void ccs_print_ipv6(char *buffer, const int buffer_len,
                    const struct in6_addr *ip)
{
        memset(buffer, 0, buffer_len);
        snprintf(buffer, buffer_len - 1, "%x:%x:%x:%x:%x:%x:%x:%x", NIP6(*ip));
}

/**
 * ccs_net2keyword - Convert network operation index to network operation name.
 *
 * @operation: Type of operation.
 *
 * Returns the name of operation.
 */
const char *ccs_net2keyword(const u8 operation)
{
        const char *keyword = "unknown";
        switch (operation) {
        case CCS_NETWORK_UDP_BIND:
                keyword = "UDP bind";
                break;
        case CCS_NETWORK_UDP_CONNECT:
                keyword = "UDP connect";
                break;
        case CCS_NETWORK_TCP_BIND:
                keyword = "TCP bind";
                break;
        case CCS_NETWORK_TCP_LISTEN:
                keyword = "TCP listen";
                break;
        case CCS_NETWORK_TCP_CONNECT:
                keyword = "TCP connect";
                break;
        case CCS_NETWORK_TCP_ACCEPT:
                keyword = "TCP accept";
                break;
        case CCS_NETWORK_RAW_BIND:
                keyword = "RAW bind";
                break;
        case CCS_NETWORK_RAW_CONNECT:
                keyword = "RAW connect";
                break;
        }
        return keyword;
}

/**
 * ccs_network_entry2 - Check permission for network operation.
 *
 * @is_ipv6:   True if @address is an IPv6 address.
 * @operation: Type of operation.
 * @address:   An IPv4 or IPv6 address.
 * @port:      Port number.
 *
 * Returns 0 on success, negative value otherwise.
 *
 * Caller holds ccs_read_lock().
 */
static int ccs_network_entry2(const bool is_ipv6, const u8 operation,
                              const u32 *address, const u16 port)
{
        struct ccs_request_info r;
        struct ccs_acl_info *ptr;
        const char *keyword = ccs_net2keyword(operation);
        const u16 perm = 1 << operation;
        /* using host byte order to allow u32 comparison than memcmp().*/
        const u32 ip = ntohl(*address);
        int error;
        char buf[64];
        if (ccs_init_request_info(&r, NULL,
                                  CCS_MAC_NETWORK_UDP_BIND + operation)
            == CCS_MAC_MODE_DISABLED)
                return 0;
        memset(buf, 0, sizeof(buf));
        if (is_ipv6)
                ccs_print_ipv6(buf, sizeof(buf), (const struct in6_addr *)
                               address);
        else
                snprintf(buf, sizeof(buf) - 1, "%u.%u.%u.%u", HIPQUAD(ip));
        do {
                error = -EPERM;
                list_for_each_entry_rcu(ptr, &r.domain->acl_info_list, list) {
                        struct ccs_ip_network_acl *acl;
                        if (ptr->is_deleted ||
                            ptr->type != CCS_TYPE_IP_NETWORK_ACL)
                                continue;
                        acl = container_of(ptr, struct ccs_ip_network_acl,
                                           head);
                        if (!(acl->perm & perm))
                                continue;
                        if (!ccs_compare_number_union(port, &acl->port) ||
                            !ccs_condition(&r, ptr))
                                continue;
                        switch (acl->address_type) {
                        case CCS_IP_ADDRESS_TYPE_ADDRESS_GROUP:
                                if (!ccs_address_matches_group(is_ipv6,
                                                               address,
                                                               acl->address.
                                                               group))
                                        continue;
                                break;
                        case CCS_IP_ADDRESS_TYPE_IPv4:
                                if (is_ipv6 || ip < acl->address.ipv4.min ||
                                    acl->address.ipv4.max < ip)
                                        continue;
                                break;
                        default:
                                if (!is_ipv6 ||
                                    memcmp(acl->address.ipv6.min, address, 16)
                                    > 0 ||
                                    memcmp(address, acl->address.ipv6.max, 16)
                                    > 0)
                                        continue;
                                break;
                        }
                        r.cond = ptr->cond;
                        error = 0;
                        break;
                }
                ccs_audit_network_log(&r, keyword, buf, port, !error);
                if (!error)
                        break;
                error = ccs_supervisor(&r, CCS_KEYWORD_ALLOW_NETWORK
                                       "%s %s %u\n", keyword, buf, port);
        } while (error == 1);
        if (r.mode != CCS_MAC_MODE_ENFORCING)
                error = 0;
        return error;
}

/**
 * ccs_network_entry - Check permission for network operation.
 *
 * @is_ipv6:   True if @address is an IPv6 address.
 * @operation: Type of operation.
 * @address:   An IPv4 or IPv6 address.
 * @port:      Port number.
 *
 * Returns 0 on success, negative value otherwise.
 */
static int ccs_network_entry(const bool is_ipv6, const u8 operation,
                             const u32 *address, const u16 port)
{
        const int idx = ccs_read_lock();
        const int error = ccs_network_entry2(is_ipv6, operation,
                                             address, port);
        ccs_read_unlock(idx);
        return error;
}

/**
 * ccs_write_network_policy - Write "struct ccs_ip_network_acl" list.
 *
 * @data:      String to parse.
 * @domain:    Pointer to "struct ccs_domain_info".
 * @condition: Pointer to "struct ccs_condition". May be NULL.
 * @is_delete: True if it is a delete request.
 *
 * Returns 0 on success, negative value otherwise.
 */
int ccs_write_network_policy(char *data, struct ccs_domain_info *domain,
                             struct ccs_condition *condition,
                             const bool is_delete)
{
        struct ccs_ip_network_acl *entry = NULL;
        struct ccs_acl_info *ptr;
        struct ccs_ip_network_acl e = {
                .head.type = CCS_TYPE_IP_NETWORK_ACL,
                .head.cond = condition,
        };
        u16 min_address[8];
        u16 max_address[8];
        int error = is_delete ? -ENOENT : -ENOMEM;
        u8 sock_type;
        char *w[4];
        if (!ccs_tokenize(data, w, sizeof(w)) || !w[3][0])
                return -EINVAL;
        if (!strcmp(w[0], "TCP"))
                sock_type = SOCK_STREAM;
        else if (!strcmp(w[0], "UDP"))
                sock_type = SOCK_DGRAM;
        else if (!strcmp(w[0], "RAW"))
                sock_type = SOCK_RAW;
        else
                return -EINVAL;
        if (!strcmp(w[1], "bind"))
                switch (sock_type) {
                case SOCK_STREAM:
                        e.perm = 1 << CCS_NETWORK_TCP_BIND;
                        break;
                case SOCK_DGRAM:
                        e.perm = 1 << CCS_NETWORK_UDP_BIND;
                        break;
                default:
                        e.perm = 1 << CCS_NETWORK_RAW_BIND;
                        break;
                }
        else if (!strcmp(w[1], "connect"))
                switch (sock_type) {
                case SOCK_STREAM:
                        e.perm = 1 << CCS_NETWORK_TCP_CONNECT;
                        break;
                case SOCK_DGRAM:
                        e.perm = 1 << CCS_NETWORK_UDP_CONNECT;
                        break;
                default:
                        e.perm = 1 << CCS_NETWORK_RAW_CONNECT;
                        break;
                }
        else if (sock_type == SOCK_STREAM && !strcmp(w[1], "listen"))
                e.perm = 1 << CCS_NETWORK_TCP_LISTEN;
        else if (sock_type == SOCK_STREAM && !strcmp(w[1], "accept"))
                e.perm = 1 << CCS_NETWORK_TCP_ACCEPT;
        else
                return -EINVAL;
        switch (ccs_parse_ip_address(w[2], min_address, max_address)) {
        case 2:
                e.address_type = CCS_IP_ADDRESS_TYPE_IPv6;
                e.address.ipv6.min = ccs_get_ipv6_address((struct in6_addr *)
                                                          min_address);
                e.address.ipv6.max = ccs_get_ipv6_address((struct in6_addr *)
                                                          max_address);
                if (!e.address.ipv6.min || !e.address.ipv6.max)
                        goto out;
                break;
        case 1:
                e.address_type = CCS_IP_ADDRESS_TYPE_IPv4;
                /* use host byte order to allow u32 comparison.*/
                e.address.ipv4.min = ntohl(*(u32 *) min_address);
                e.address.ipv4.max = ntohl(*(u32 *) max_address);
                break;
        default:
                if (w[2][0] != '@')
                        return -EINVAL;
                e.address_type = CCS_IP_ADDRESS_TYPE_ADDRESS_GROUP;
                e.address.group = ccs_get_address_group(w[2] + 1);
                if (!e.address.group)
                        return -ENOMEM;
                break;
        }
        if (!ccs_parse_number_union(w[3], &e.port))
                goto out;
        if (!is_delete)
                entry = kmalloc(sizeof(e), GFP_KERNEL);
        mutex_lock(&ccs_policy_lock);
        list_for_each_entry_rcu(ptr, &domain->acl_info_list, list) {
                struct ccs_ip_network_acl *acl =
                        container_of(ptr, struct ccs_ip_network_acl,
                                     head);
                if (ptr->type != CCS_TYPE_IP_NETWORK_ACL ||
                    ptr->cond != condition ||
                    ccs_memcmp(acl, &e, offsetof(typeof(e), address_type),
                               sizeof(e)))
                        continue;
                if (is_delete) {
                        acl->perm &= ~e.perm;
                        if (!acl->perm)
                                ptr->is_deleted = true;
                } else {
                        if (ptr->is_deleted)
                                acl->perm = 0;
                        acl->perm |= e.perm;
                        ptr->is_deleted = false;
                }
                error = 0;
                break;
        }
        if (!is_delete && error && ccs_commit_ok(entry, &e, sizeof(e))) {
                ccs_add_domain_acl(domain, &entry->head);
                entry = NULL;
                error = 0;
        }
        mutex_unlock(&ccs_policy_lock);
 out:
        if (w[2][0] == '@')
                ccs_put_address_group(e.address.group);
        else if (e.address_type == CCS_IP_ADDRESS_TYPE_IPv6) {
                ccs_put_ipv6_address(e.address.ipv6.min);
                ccs_put_ipv6_address(e.address.ipv6.max);
        }
        ccs_put_number_union(&e.port);
        kfree(entry);
        return error;
}

/**
 * ccs_network_listen_acl - Check permission for listen() operation.
 *
 * @is_ipv6: True if @address is an IPv6 address.
 * @address: An IPv4 or IPv6 address.
 * @port:    Port number.
 *
 * Returns 0 on success, negative value otherwise.
 */
static inline int ccs_network_listen_acl(const bool is_ipv6,
                                         const u8 *address,
                                         const u16 port)
{
        return ccs_network_entry(is_ipv6, CCS_NETWORK_TCP_LISTEN,
                                 (const u32 *) address, ntohs(port));
}

/**
 * ccs_network_connect_acl - Check permission for connect() operation.
 *
 * @is_ipv6:   True if @address is an IPv6 address.
 * @sock_type: Type of socket. (TCP or UDP or RAW)
 * @address:   An IPv4 or IPv6 address.
 * @port:      Port number.
 *
 * Returns 0 on success, negative value otherwise.
 */
static inline int ccs_network_connect_acl(const bool is_ipv6,
                                          const int sock_type,
                                          const u8 *address,
                                          const u16 port)
{
        u8 operation;
        switch (sock_type) {
        case SOCK_STREAM:
                operation = CCS_NETWORK_TCP_CONNECT;
                break;
        case SOCK_DGRAM:
                operation = CCS_NETWORK_UDP_CONNECT;
                break;
        default:
                operation = CCS_NETWORK_RAW_CONNECT;
        }
        return ccs_network_entry(is_ipv6, operation,
                                 (const u32 *) address, ntohs(port));
}

/**
 * ccs_network_bind_acl - Check permission for bind() operation.
 *
 * @is_ipv6:   True if @address is an IPv6 address.
 * @sock_type: Type of socket. (TCP or UDP or RAW)
 * @address:   An IPv4 or IPv6 address.
 * @port:      Port number.
 *
 * Returns 0 on success, negative value otherwise.
 */
static int ccs_network_bind_acl(const bool is_ipv6, const int sock_type,
                                const u8 *address, const u16 port)
{
        u8 operation;
        switch (sock_type) {
        case SOCK_STREAM:
                operation = CCS_NETWORK_TCP_BIND;
                break;
        case SOCK_DGRAM:
                operation = CCS_NETWORK_UDP_BIND;
                break;
        default:
                operation = CCS_NETWORK_RAW_BIND;
        }
        return ccs_network_entry(is_ipv6, operation,
                                 (const u32 *) address, ntohs(port));
}

/**
 * ccs_network_accept_acl - Check permission for accept() operation.
 *
 * @is_ipv6: True if @address is an IPv6 address.
 * @address: An IPv4 or IPv6 address.
 * @port:    Port number.
 *
 * Returns 0 on success, negative value otherwise.
 */
static inline int ccs_network_accept_acl(const bool is_ipv6,
                                         const u8 *address,
                                         const u16 port)
{
        int retval;
        current->ccs_flags |= CCS_DONT_SLEEP_ON_ENFORCE_ERROR;
        retval = ccs_network_entry(is_ipv6, CCS_NETWORK_TCP_ACCEPT,
                                   (const u32 *) address, ntohs(port));
        current->ccs_flags &= ~CCS_DONT_SLEEP_ON_ENFORCE_ERROR;
        return retval;
}

/**
 * ccs_network_sendmsg_acl - Check permission for sendmsg() operation.
 *
 * @is_ipv6:   True if @address is an IPv6 address.
 * @sock_type: Type of socket. (UDP or RAW)
 * @address:   An IPv4 or IPv6 address.
 * @port:      Port number.
 *
 * Returns 0 on success, negative value otherwise.
 */
static inline int ccs_network_sendmsg_acl(const bool is_ipv6,
                                          const int sock_type,
                                          const u8 *address,
                                          const u16 port)
{
        u8 operation;
        if (sock_type == SOCK_DGRAM)
                operation = CCS_NETWORK_UDP_CONNECT;
        else
                operation = CCS_NETWORK_RAW_CONNECT;
        return ccs_network_entry(is_ipv6, operation,
                                 (const u32 *) address, ntohs(port));
}

/**
 * ccs_network_recvmsg_acl - Check permission for recvmsg() operation.
 *
 * @is_ipv6:   True if @address is an IPv6 address.
 * @sock_type: Type of socket. (UDP or RAW)
 * @address:   An IPv4 or IPv6 address.
 * @port:      Port number.
 *
 * Returns 0 on success, negative value otherwise.
 */
static inline int ccs_network_recvmsg_acl(const bool is_ipv6,
                                          const int sock_type,
                                          const u8 *address,
                                          const u16 port)
{
        int retval;
        const u8 operation
                = (sock_type == SOCK_DGRAM) ?
                CCS_NETWORK_UDP_CONNECT : CCS_NETWORK_RAW_CONNECT;
        current->ccs_flags |= CCS_DONT_SLEEP_ON_ENFORCE_ERROR;
        retval = ccs_network_entry(is_ipv6, operation,
                                   (const u32 *) address, ntohs(port));
        current->ccs_flags &= ~CCS_DONT_SLEEP_ON_ENFORCE_ERROR;
        return retval;
}

#define MAX_SOCK_ADDR 128 /* net/socket.c */

/* Check permission for creating a socket. */
int ccs_socket_create_permission(int family, int type, int protocol)
{
        int error = 0;
        /* Nothing to do if I am a kernel service. */
        if (segment_eq(get_fs(), KERNEL_DS))
                return 0;
        if (family == PF_PACKET && !ccs_capable(CCS_USE_PACKET_SOCKET))
                return -EPERM;
        if (family == PF_ROUTE && !ccs_capable(CCS_USE_ROUTE_SOCKET))
                return -EPERM;
        if (family != PF_INET && family != PF_INET6)
                return 0;
        switch (type) {
        case SOCK_STREAM:
                if (!ccs_capable(CCS_INET_STREAM_SOCKET_CREATE))
                        error = -EPERM;
                break;
        case SOCK_DGRAM:
                if (!ccs_capable(CCS_USE_INET_DGRAM_SOCKET))
                        error = -EPERM;
                break;
        case SOCK_RAW:
                if (!ccs_capable(CCS_USE_INET_RAW_SOCKET))
                        error = -EPERM;
                break;
        }
        return error;
}

/* Check permission for listening a TCP socket. */
int ccs_socket_listen_permission(struct socket *sock)
{
        int error = 0;
        char addr[MAX_SOCK_ADDR];
        int addr_len;
        /* Nothing to do if I am a kernel service. */
        if (segment_eq(get_fs(), KERNEL_DS))
                return 0;
        if (sock->type != SOCK_STREAM)
                return 0;
        switch (sock->sk->sk_family) {
        case PF_INET:
        case PF_INET6:
                break;
        default:
                return 0;
        }
        if (!ccs_capable(CCS_INET_STREAM_SOCKET_LISTEN))
                return -EPERM;
        if (sock->ops->getname(sock, (struct sockaddr *) addr, &addr_len, 0))
                return -EPERM;
        switch (((struct sockaddr *) addr)->sa_family) {
                struct sockaddr_in6 *addr6;
                struct sockaddr_in *addr4;
        case AF_INET6:
                addr6 = (struct sockaddr_in6 *) addr;
                error = ccs_network_listen_acl(true,
                                               addr6->sin6_addr.s6_addr,
                                               addr6->sin6_port);
                break;
        case AF_INET:
                addr4 = (struct sockaddr_in *) addr;
                error = ccs_network_listen_acl(false,
                                               (u8 *) &addr4->sin_addr,
                                               addr4->sin_port);
                break;
        }
        return error;
}

/* Check permission for setting the remote IP address/port pair of a socket. */
int ccs_socket_connect_permission(struct socket *sock, struct sockaddr *addr,
                                  int addr_len)
{
        int error = 0;
        const unsigned int type = sock->type;
        /* Nothing to do if I am a kernel service. */
        if (segment_eq(get_fs(), KERNEL_DS))
                return 0;
        switch (type) {
        case SOCK_STREAM:
        case SOCK_DGRAM:
        case SOCK_RAW:
                break;
        default:
                return 0;
        }
        switch (addr->sa_family) {
                struct sockaddr_in6 *addr6;
                struct sockaddr_in *addr4;
                u16 port;
        case AF_INET6:
                if (addr_len < SIN6_LEN_RFC2133)
                        break;
                addr6 = (struct sockaddr_in6 *) addr;
                if (type != SOCK_RAW)
                        port = addr6->sin6_port;
                else
                        port = htons(sock->sk->sk_protocol);
                error = ccs_network_connect_acl(true, type,
                                                addr6->sin6_addr.s6_addr,
                                                port);
                break;
        case AF_INET:
                if (addr_len < sizeof(struct sockaddr_in))
                        break;
                addr4 = (struct sockaddr_in *) addr;
                if (type != SOCK_RAW)
                        port = addr4->sin_port;
                else
                        port = htons(sock->sk->sk_protocol);
                error = ccs_network_connect_acl(false, type,
                                                (u8 *) &addr4->sin_addr,
                                                port);
                break;
        }
        if (type != SOCK_STREAM)
                return error;
        switch (sock->sk->sk_family) {
        case PF_INET:
        case PF_INET6:
                if (!ccs_capable(CCS_INET_STREAM_SOCKET_CONNECT))
                        error = -EPERM;
                break;
        }
        return error;
}

/* Check permission for setting the local IP address/port pair of a socket. */
int ccs_socket_bind_permission(struct socket *sock, struct sockaddr *addr,
                               int addr_len)
{
        int error = 0;
        const unsigned int type = sock->type;
        /* Nothing to do if I am a kernel service. */
        if (segment_eq(get_fs(), KERNEL_DS))
                return 0;
        switch (type) {
        case SOCK_STREAM:
        case SOCK_DGRAM:
        case SOCK_RAW:
                break;
        default:
                return 0;
        }
        switch (addr->sa_family) {
                struct sockaddr_in6 *addr6;
                struct sockaddr_in *addr4;
                u16 port;
        case AF_INET6:
                if (addr_len < SIN6_LEN_RFC2133)
                        break;
                addr6 = (struct sockaddr_in6 *) addr;
                if (type != SOCK_RAW)
                        port = addr6->sin6_port;
                else
                        port = htons(sock->sk->sk_protocol);
                error = ccs_network_bind_acl(true, type,
                                             addr6->sin6_addr.s6_addr,
                                             port);
                break;
        case AF_INET:
                if (addr_len < sizeof(struct sockaddr_in))
                        break;
                addr4 = (struct sockaddr_in *) addr;
                if (type != SOCK_RAW)
                        port = addr4->sin_port;
                else
                        port = htons(sock->sk->sk_protocol);
                error = ccs_network_bind_acl(false, type,
                                             (u8 *) &addr4->sin_addr,
                                             port);
                break;
        }
        return error;
}

/*
 * Check permission for accepting a TCP socket.
 *
 * Currently, the LSM hook for this purpose is not provided.
 */
int ccs_socket_accept_permission(struct socket *sock, struct sockaddr *addr)
{
        int error = 0;
        int addr_len;
        /* Nothing to do if I am a kernel service. */
        if (segment_eq(get_fs(), KERNEL_DS))
                return 0;
        switch (sock->sk->sk_family) {
        case PF_INET:
        case PF_INET6:
                break;
        default:
                return 0;
        }
        error = sock->ops->getname(sock, addr, &addr_len, 2);
        if (error)
                return error;
        switch (addr->sa_family) {
                struct sockaddr_in6 *addr6;
                struct sockaddr_in *addr4;
        case AF_INET6:
                addr6 = (struct sockaddr_in6 *) addr;
                error = ccs_network_accept_acl(true,
                                               addr6->sin6_addr.s6_addr,
                                               addr6->sin6_port);
                break;
        case AF_INET:
                addr4 = (struct sockaddr_in *) addr;
                error = ccs_network_accept_acl(false,
                                               (u8 *) &addr4->sin_addr,
                                               addr4->sin_port);
                break;
        }
        return error;
}

/* Check permission for sending a datagram via a UDP or RAW socket. */
int ccs_socket_sendmsg_permission(struct socket *sock, struct sockaddr *addr,
                                  int addr_len)
{
        int error = 0;
        const int type = sock->type;
        /* Nothing to do if I am a kernel service. */
        if (segment_eq(get_fs(), KERNEL_DS))
                return 0;
        if (!addr || (type != SOCK_DGRAM && type != SOCK_RAW))
                return 0;
        switch (addr->sa_family) {
                struct sockaddr_in6 *addr6;
                struct sockaddr_in *addr4;
                u16 port;
        case AF_INET6:
                if (addr_len < SIN6_LEN_RFC2133)
                        break;
                addr6 = (struct sockaddr_in6 *) addr;
                if (type == SOCK_DGRAM)
                        port = addr6->sin6_port;
                else
                        port = htons(sock->sk->sk_protocol);
                error = ccs_network_sendmsg_acl(true, type,
                                                addr6->sin6_addr.s6_addr,
                                                port);
                break;
        case AF_INET:
                if (addr_len < sizeof(struct sockaddr_in))
                        break;
                addr4 = (struct sockaddr_in *) addr;
                if (type == SOCK_DGRAM)
                        port = addr4->sin_port;
                else
                        port = htons(sock->sk->sk_protocol);
                error = ccs_network_sendmsg_acl(false, type,
                                                (u8 *) &addr4->sin_addr,
                                                port);
                break;
        }
        return error;
}

#if LINUX_VERSION_CODE < KERNEL_VERSION(2, 6, 22)
#if !defined(RHEL_MAJOR) || RHEL_MAJOR != 5

static inline struct iphdr *ip_hdr(const struct sk_buff *skb)
{
        return skb->nh.iph;
}

static inline struct udphdr *udp_hdr(const struct sk_buff *skb)
{
        return skb->h.uh;
}

static inline struct ipv6hdr *ipv6_hdr(const struct sk_buff *skb)
{
        return skb->nh.ipv6h;
}

#endif
#endif

#if LINUX_VERSION_CODE < KERNEL_VERSION(2, 6, 12)
static void skb_kill_datagram(struct sock *sk, struct sk_buff *skb,
                              unsigned int flags)
{
        /* Clear queue. */
        if (flags & MSG_PEEK) {
                int clear = 0;
                spin_lock_irq(&sk->sk_receive_queue.lock);
                if (skb == skb_peek(&sk->sk_receive_queue)) {
                        __skb_unlink(skb, &sk->sk_receive_queue);
                        clear = 1;
                }
                spin_unlock_irq(&sk->sk_receive_queue.lock);
                if (clear)
                        kfree_skb(skb);
        }
        skb_free_datagram(sk, skb);
}
#elif LINUX_VERSION_CODE < KERNEL_VERSION(2, 6, 16)
static void skb_kill_datagram(struct sock *sk, struct sk_buff *skb,
                              unsigned int flags)
{
        /* Clear queue. */
        if (flags & MSG_PEEK) {
                int clear = 0;
                spin_lock_bh(&sk->sk_receive_queue.lock);
                if (skb == skb_peek(&sk->sk_receive_queue)) {
                        __skb_unlink(skb, &sk->sk_receive_queue);
                        clear = 1;
                }
                spin_unlock_bh(&sk->sk_receive_queue.lock);
                if (clear)
                        kfree_skb(skb);
        }
        skb_free_datagram(sk, skb);
}
#endif

/*
 * Check permission for receiving a datagram via a UDP or RAW socket.
 *
 * Currently, the LSM hook for this purpose is not provided.
 */
int ccs_socket_recvmsg_permission(struct sock *sk, struct sk_buff *skb,
                                  const unsigned int flags)
{
        int error = 0;
        const unsigned int type = sk->sk_type;
        if (type != SOCK_DGRAM && type != SOCK_RAW)
                return 0;
        /* Nothing to do if I am a kernel service. */
        if (segment_eq(get_fs(), KERNEL_DS))
                return 0;

        switch (sk->sk_family) {
                struct in6_addr sin6;
                struct in_addr sin4;
                u16 port;
        case PF_INET6:
                if (type == SOCK_DGRAM) { /* UDP IPv6 */
                        if (skb->protocol == htons(ETH_P_IP)) {
                                ipv6_addr_set(&sin6, 0, 0, htonl(0xffff),
                                              ip_hdr(skb)->saddr);
                        } else {
                                ipv6_addr_copy(&sin6, &ipv6_hdr(skb)->saddr);
                        }
                        port = udp_hdr(skb)->source;
                } else { /* RAW IPv6 */
                        ipv6_addr_copy(&sin6, &ipv6_hdr(skb)->saddr);
                        port = htons(sk->sk_protocol);
                }
                error = ccs_network_recvmsg_acl(true, type,
                                                (u8 *) &sin6, port);
                break;
        case PF_INET:
                if (type == SOCK_DGRAM) { /* UDP IPv4 */
                        sin4.s_addr = ip_hdr(skb)->saddr;
                        port = udp_hdr(skb)->source;
                } else { /* RAW IPv4 */
                        sin4.s_addr = ip_hdr(skb)->saddr;
                        port = htons(sk->sk_protocol);
                }
                error = ccs_network_recvmsg_acl(false, type,
                                                (u8 *) &sin4, port);
                break;
        }
        if (!error)
                return 0;
        /*
         * Remove from queue if MSG_PEEK is used so that
         * the head message from unwanted source in receive queue will not
         * prevent the caller from picking up next message from wanted source
         * when the caller is using MSG_PEEK flag for picking up.
         */
#if LINUX_VERSION_CODE >= KERNEL_VERSION(2, 6, 25)
        if (type == SOCK_DGRAM)
                lock_sock(sk);
#endif
        skb_kill_datagram(sk, skb, flags);
#if LINUX_VERSION_CODE >= KERNEL_VERSION(2, 6, 25)
        if (type == SOCK_DGRAM)
                release_sock(sk);
#endif
        /* Hope less harmful than -EPERM. */
        return -ENOMEM;
}
EXPORT_SYMBOL(ccs_socket_recvmsg_permission);