/* $Id: tf_bytes.c,v 1.1.1.1 2005/11/23 10:25:07 steveb Exp $ */

/*

  Copyright (C) 2004 Peter Urbanec <toppy at urbanec.net>

  This file is part of puppy.

  puppy 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
  (at your option) any later version.

  puppy 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 puppy; if not, write to the Free Software
  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

*/

#define _XOPEN_SOURCE 600
#include <unistd.h>

#include <stdio.h>
#include "tf_bytes.h"

__u16 get_u16(const void *addr)
{
    const __u8 *b = addr;

    return ((b[0] << 8) & 0xff00) | ((b[1] << 0) & 0x00ff);
}

void put_u16(void *addr, __u16 val)
{
    __u8 *b = addr;

    b[0] = (val >> 8) & 0xFF;
    b[1] = (val & 0xFF);
}

__u32 get_u24(const void *addr)
{
    const __u8 *b = addr;

    return (b[0] << 16) | (b[1] << 8) | (b[2]);
}

__u32 get_u32(const void *addr)
{
    const __u8 *b = addr;

    return (b[0] << 24) | (b[1] << 16) | (b[2] << 8) | (b[3]);
}

/* Retrieve a 32-bit integer from the raw buffer (prior to byteswapping) */
__u32 get_u32_raw(const void *addr)
{
    const __u8 *b = addr;

    return (b[1] << 24) | (b[0] << 16) | (b[3] << 8) | (b[2]);
}

__u16 get_u16_raw(const void *addr)
{
    const __u8 *b = addr;

    return ((b[1] << 8) & 0xff00) | ((b[0] << 0) & 0x00ff);
}


void put_u32(void *addr, __u32 val)
{
    __u8 *b = addr;

    b[0] = (val >> 24) & 0xFF;
    b[1] = (val >> 16) & 0xFF;
    b[2] = (val >> 8) & 0xFF;
    b[3] = (val & 0xFF);
}

__u64 get_u64(const void *addr)
{
    const __u8 *b = addr;
    __u64 r = b[0];

    r = (r << 8) | b[1];
    r = (r << 8) | b[2];
    r = (r << 8) | b[3];
    r = (r << 8) | b[4];
    r = (r << 8) | b[5];
    r = (r << 8) | b[6];
    r = (r << 8) | b[7];
    return r;
}

void put_u64(void *addr, __u64 val)
{
    __u8 *b = addr;

    b[0] = (val >> 56) & 0xFF;
    b[1] = (val >> 48) & 0xFF;
    b[2] = (val >> 40) & 0xFF;
    b[3] = (val >> 32) & 0xFF;
    b[4] = (val >> 24) & 0xFF;
    b[5] = (val >> 16) & 0xFF;
    b[6] = (val >> 8) & 0xFF;
    b[7] = (val & 0xFF);
}

void byte_swap(void *packet, size_t count)
{
#ifdef NO_SWAB
	/* On the offchance that swab() isn't available
	 * or doesn't work in-place, we do it manually
	 */
    int i;
	__u8 *d = (__u8 *)packet;

    for (i = 0; i < count; i += 2) {
        __u8 t = d[i];

        d[i] = d[i + 1];
        d[i + 1] = t;
    }
#else
	swab(packet, packet, count);
#endif
}