source: mainline/uspace/srv/net/checksum.c@ 7715994

/*
 * Copyright (c) 2009 Lukas Mejdrech
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * - Redistributions of source code must retain the above copyright
 *   notice, this list of conditions and the following disclaimer.
 * - Redistributions in binary form must reproduce the above copyright
 *   notice, this list of conditions and the following disclaimer in the
 *   documentation and/or other materials provided with the distribution.
 * - The name of the author may not be used to endorse or promote products
 *   derived from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

/** @addtogroup net
 *  @{
 */

/** @file
 *  General CRC and checksum computation implementation.
 */

#include <sys/types.h>

#include "include/checksum.h"

/** Big-endian encoding CRC divider.
 */
#define CRC_DIVIDER_BE  0x04C11DB7

/** Little-endian encoding CRC divider.
 */
#define CRC_DIVIDER_LE  0xEDB88320

uint16_t compact_checksum(uint32_t sum){
        // shorten to the 16 bits
        while(sum >> 16){
                sum = (sum &0xFFFF) + (sum >> 16);
        }

        return (uint16_t) sum;
}

uint32_t compute_checksum(uint32_t seed, uint8_t * data, size_t length){
        size_t index;

        // sum all the 16 bit fields
        for(index = 0; index + 1 < length; index += 2){
                seed += (data[index] << 8) + data[index + 1];
        }

        // last odd byte with zero padding
        if(index + 1 == length){
                seed += data[index] << 8;
        }

        return seed;
}

uint32_t compute_crc32_be(uint32_t seed, uint8_t * data, size_t length){
        size_t index;

        // process full bytes
        while(length >= 8){
                // add the data
                seed ^= (*data) << 24;
                // for each added bit
                for(index = 0; index < 8; ++ index){
                        // if the first bit is set
                        if(seed &0x80000000){
                                // shift and divide the checksum
                                seed = (seed << 1) ^ ((uint32_t) CRC_DIVIDER_BE);
                        }else{
                                // shift otherwise
                                seed <<= 1;
                        }
                }
                // move to the next byte
                ++ data;
                length -= 8;
        }

        // process the odd bits
        if(length > 0){
                // add the data with zero padding
                seed ^= ((*data) &(0xFF << (8 - length))) << 24;
                // for each added bit
                for(index = 0; index < length; ++ index){
                        // if the first bit is set
                        if(seed &0x80000000){
                                // shift and divide the checksum
                                seed = (seed << 1) ^ ((uint32_t) CRC_DIVIDER_BE);
                        }else{
                                // shift otherwise
                                seed <<= 1;
                        }
                }
        }

        return seed;
}

uint32_t compute_crc32_le(uint32_t seed, uint8_t * data, size_t length){
        size_t index;

        // process full bytes
        while(length >= 8){
                // add the data
                seed ^= (*data);
                // for each added bit
                for(index = 0; index < 8; ++ index){
                        // if the last bit is set
                        if(seed &1){
                                // shift and divide the checksum
                                seed = (seed >> 1) ^ ((uint32_t) CRC_DIVIDER_LE);
                        }else{
                                // shift otherwise
                                seed >>= 1;
                        }
                }
                // move to the next byte
                ++ data;
                length -= 8;
        }

        // process the odd bits
        if(length > 0){
                // add the data with zero padding
                seed ^= (*data) >> (8 - length);
                for(index = 0; index < length; ++ index){
                        // if the last bit is set
                        if(seed &1){
                                // shift and divide the checksum
                                seed = (seed >> 1) ^ ((uint32_t) CRC_DIVIDER_LE);
                        }else{
                                // shift otherwise
                                seed >>= 1;
                        }
                }
        }

        return seed;
}

uint16_t flip_checksum(uint16_t checksum){
        // flip, zero is returned as 0xFFFF (not flipped)
        checksum = ~ checksum;
        return checksum ? checksum : IP_CHECKSUM_ZERO;
}

uint16_t ip_checksum(uint8_t * data, size_t length){
        // compute, compact and flip the data checksum
        return flip_checksum(compact_checksum(compute_checksum(0, data, length)));
}

/** @}
 */