// SPDX-License-Identifier: GPL-2.0-or-later
/* Utility routines
 *
 * Copyright (C) 2015 Red Hat, Inc. All Rights Reserved.
 * Written by David Howells (dhowells@redhat.com)
 */

#include <linux/ip.h>
#include <linux/ipv6.h>
#include <linux/udp.h>
#include "ar-internal.h"

/*
 * Fill out a peer address from a socket buffer containing a packet.
 */
int rxrpc_extract_addr_from_skb(struct sockaddr_rxrpc *srx, struct sk_buff *skb)
{
        memset(srx, 0, sizeof(*srx));

        switch (ntohs(skb->protocol)) {
        case ETH_P_IP:
                srx->transport_type = SOCK_DGRAM;
                srx->transport_len = sizeof(srx->transport.sin);
                srx->transport.sin.sin_family = AF_INET;
                srx->transport.sin.sin_port = udp_hdr(skb)->source;
                srx->transport.sin.sin_addr.s_addr = ip_hdr(skb)->saddr;
                return 0;

#ifdef CONFIG_AF_RXRPC_IPV6
        case ETH_P_IPV6:
                srx->transport_type = SOCK_DGRAM;
                srx->transport_len = sizeof(srx->transport.sin6);
                srx->transport.sin6.sin6_family = AF_INET6;
                srx->transport.sin6.sin6_port = udp_hdr(skb)->source;
                srx->transport.sin6.sin6_addr = ipv6_hdr(skb)->saddr;
                return 0;
#endif

        default:
                pr_warn_ratelimited("AF_RXRPC: Unknown eth protocol %u\n",
                                    ntohs(skb->protocol));
                return -EAFNOSUPPORT;
        }
}