icmp.c

Go to the documentation of this file.

/*
 * Copyright (C) 2013 Michael Brown <mbrown@fensystems.co.uk>.
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License as
 * published by the Free Software Foundation; either version 2 of the
 * License, or any later version.
 *
 * 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.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 *
 * You can also choose to distribute this program under the terms of
 * the Unmodified Binary Distribution Licence (as given in the file
 * COPYING.UBDL), provided that you have satisfied its requirements.
 */

FILE_LICENCE ( GPL2_OR_LATER_OR_UBDL );
FILE_SECBOOT ( PERMITTED );

#include <string.h>
#include <byteswap.h>
#include <errno.h>
#include <ipxe/iobuf.h>
#include <ipxe/in.h>
#include <ipxe/tcpip.h>
#include <ipxe/ping.h>
#include <ipxe/crc32.h>
#include <ipxe/icmp.h>

/** @file
 *
 * ICMP protocol
 *
 */

/**
 * Identify ICMP echo protocol
 *
 * @v st_family         Address family
 * @ret echo_protocol   ICMP echo protocol, or NULL
 */
static struct icmp_echo_protocol * icmp_echo_protocol ( sa_family_t family ) {
        struct icmp_echo_protocol *echo_protocol;

        for_each_table_entry ( echo_protocol, ICMP_ECHO_PROTOCOLS ) {
                if ( echo_protocol->family == family )
                        return echo_protocol;
        }
        return NULL;
}

/**
 *
 * Determine debugging colour for ICMP debug messages
 *
 * @v st_peer           Peer address
 * @ret col             Debugging colour (for DBGC())
 */
static uint32_t icmpcol ( struct sockaddr_tcpip *st_peer ) {

        return crc32_le ( 0, st_peer, sizeof ( *st_peer ) );
}

/**
 * Transmit ICMP echo packet
 *
 * @v iobuf             I/O buffer
 * @v st_dest           Destination socket address
 * @v echo_protocol     ICMP echo protocol
 * @ret rc              Return status code
 */
static int icmp_tx_echo ( struct io_buffer *iobuf,
                          struct sockaddr_tcpip *st_dest,
                          struct icmp_echo_protocol *echo_protocol ) {
        struct icmp_echo *echo = iobuf->data;
        int rc;

        /* Set ICMP type and (re)calculate checksum */
        echo->icmp.chksum = 0;
        echo->icmp.chksum = tcpip_chksum ( echo, iob_len ( iobuf ) );

        /* Transmit packet */
        if ( ( rc = tcpip_tx ( iobuf, echo_protocol->tcpip_protocol, NULL,
                               st_dest, NULL,
                               ( echo_protocol->net_checksum ?
                                 &echo->icmp.chksum : NULL ) ) ) != 0 )
                return rc;

        return 0;
}

/**
 * Transmit ICMP echo request
 *
 * @v iobuf             I/O buffer
 * @v st_dest           Destination socket address
 * @ret rc              Return status code
 */
int icmp_tx_echo_request ( struct io_buffer *iobuf,
                           struct sockaddr_tcpip *st_dest ) {
        struct icmp_echo *echo = iobuf->data;
        struct icmp_echo_protocol *echo_protocol;
        int rc;

        /* Identify ICMP echo protocol */
        echo_protocol = icmp_echo_protocol ( st_dest->st_family );
        if ( ! echo_protocol ) {
                DBGC ( icmpcol ( st_dest ), "ICMP TX echo request unknown "
                       "address family %d\n", st_dest->st_family );
                free_iob ( iobuf );
                return -ENOTSUP;
        }

        /* Set type */
        echo->icmp.type = echo_protocol->request;

        /* Transmit request */
        DBGC ( icmpcol ( st_dest ), "ICMP TX echo request id %04x seq %04x\n",
               ntohs ( echo->ident ), ntohs ( echo->sequence ) );
        if ( ( rc = icmp_tx_echo ( iobuf, st_dest, echo_protocol ) ) != 0 )
                return rc;

        return 0;
}

/**
 * Transmit ICMP echo reply
 *
 * @v iobuf             I/O buffer
 * @v st_dest           Destination socket address
 * @ret rc              Return status code
 */
static int icmp_tx_echo_reply ( struct io_buffer *iobuf,
                                struct sockaddr_tcpip *st_dest,
                                struct icmp_echo_protocol *echo_protocol ) {
        struct icmp_echo *echo = iobuf->data;
        int rc;

        /* Set type */
        echo->icmp.type = echo_protocol->reply;

        /* Transmit reply */
        DBGC ( icmpcol ( st_dest ), "ICMP TX echo reply id %04x seq %04x\n",
               ntohs ( echo->ident ), ntohs ( echo->sequence ) );
        if ( ( rc = icmp_tx_echo ( iobuf, st_dest, echo_protocol ) ) != 0 )
                return rc;

        return 0;
}

/**
 * Process a received ICMP echo request
 *
 * @v iobuf             I/O buffer
 * @v st_src            Source socket address
 * @v echo_protocol     ICMP echo protocol
 * @ret rc              Return status code
 */
int icmp_rx_echo_request ( struct io_buffer *iobuf,
                           struct sockaddr_tcpip *st_src,
                           struct icmp_echo_protocol *echo_protocol ) {
        struct icmp_echo *echo = iobuf->data;
        int rc;

        /* Sanity check */
        if ( iob_len ( iobuf ) < sizeof ( *echo ) ) {
                DBGC ( icmpcol ( st_src ), "ICMP RX echo request too short at "
                       "%zd bytes (min %zd bytes)\n",
                       iob_len ( iobuf ), sizeof ( *echo ) );
                free_iob ( iobuf );
                return -EINVAL;
        }
        DBGC ( icmpcol ( st_src ), "ICMP RX echo request id %04x seq %04x\n",
               ntohs ( echo->ident ), ntohs ( echo->sequence ) );

        /* Transmit echo reply */
        if ( ( rc = icmp_tx_echo_reply ( iobuf, st_src, echo_protocol ) ) != 0 )
                return rc;

        return 0;
}

/**
 * Process a received ICMP echo request
 *
 * @v iobuf             I/O buffer
 * @v st_src            Source socket address
 * @ret rc              Return status code
 */
int icmp_rx_echo_reply ( struct io_buffer *iobuf,
                         struct sockaddr_tcpip *st_src ) {
        struct icmp_echo *echo = iobuf->data;
        int rc;

        /* Sanity check */
        if ( iob_len ( iobuf ) < sizeof ( *echo ) ) {
                DBGC ( icmpcol ( st_src ), "ICMP RX echo reply too short at "
                       "%zd bytes (min %zd bytes)\n",
                       iob_len ( iobuf ), sizeof ( *echo ) );
                free_iob ( iobuf );
                return -EINVAL;
        }
        DBGC ( icmpcol ( st_src ), "ICMP RX echo reply id %04x seq %04x\n",
               ntohs ( echo->ident ), ntohs ( echo->sequence ) );

        /* Deliver to ping protocol */
        if ( ( rc = ping_rx ( iobuf, st_src ) ) != 0 )
                return rc;

        return 0;
}

/**
 * Receive ping reply (when no ping protocol is present)
 *
 * @v iobuf             I/O buffer
 * @v st_src            Source socket address
 * @ret rc              Return status code
 */
__weak int ping_rx ( struct io_buffer *iobuf,
                     struct sockaddr_tcpip *st_src __unused ) {
        free_iob ( iobuf );
        return 0;
}