--- libelwix/src/pack.c 2014/02/11 13:25:09 1.6.2.1 +++ libelwix/src/pack.c 2015/06/25 00:36:48 1.7.8.1 @@ -3,7 +3,7 @@ * by Michael Pounov * * $Author: misho $ -* $Id: pack.c,v 1.6.2.1 2014/02/11 13:25:09 misho Exp $ +* $Id: pack.c,v 1.7.8.1 2015/06/25 00:36:48 misho Exp $ * ************************************************************************** The ELWIX and AITNET software is distributed under the following @@ -12,7 +12,7 @@ terms: All of the documentation and software included in the ELWIX and AITNET Releases is copyrighted by ELWIX - Sofia/Bulgaria -Copyright 2004 - 2014 +Copyright 2004 - 2015 by Michael Pounov . All rights reserved. Redistribution and use in source and binary forms, with or without @@ -225,7 +225,7 @@ rpack_uint8(rpack_t * __restrict rp, uint8_t * __restr * * @rp = raw buffer * @n = set value if !=NULL - * @be = extract in big-endian + * @be = byte order [-1 little endian, 1 big endian and 0 host order] * return: -1 error or get value */ uint16_t @@ -240,7 +240,16 @@ rpack_uint16(rpack_t * __restrict rp, uint16_t * __res if (!(next = rpack_align_and_reserve(rp, sizeof(uint16_t)))) return (uint16_t) -1; - u = be ? EXTRACT_BE_16(next) : EXTRACT_LE_16(next); + if (be < 0) + u = EXTRACT_LE_16(next); + else if (be > 0) + u = EXTRACT_BE_16(next); + else +#if BYTE_ORDER == BIG_ENDIAN + u = EXTRACT_BE_16(next); +#else + u = EXTRACT_LE_16(next); +#endif if (n) RPACK_SET_16(next, n); @@ -253,7 +262,7 @@ rpack_uint16(rpack_t * __restrict rp, uint16_t * __res * * @rp = raw buffer * @n = set value if !=NULL - * @be = extract in big-endian + * @be = byte order [-1 little endian, 1 big endian and 0 host order] * return: -1 error or get value */ uint32_t @@ -268,7 +277,16 @@ rpack_uint24(rpack_t * __restrict rp, uint32_t * __res if (!(next = rpack_align_and_reserve(rp, sizeof(uint32_t)))) return (uint32_t) -1; - u = be ? EXTRACT_BE_24(next) : EXTRACT_LE_24(next); + if (be < 0) + u = EXTRACT_LE_24(next); + else if (be > 0) + u = EXTRACT_BE_24(next); + else +#if BYTE_ORDER == BIG_ENDIAN + u = EXTRACT_BE_24(next); +#else + u = EXTRACT_LE_24(next); +#endif if (n) RPACK_SET_24(next, n); @@ -281,7 +299,7 @@ rpack_uint24(rpack_t * __restrict rp, uint32_t * __res * * @rp = raw buffer * @n = set value if !=NULL - * @be = extract in big-endian + * @be = byte order [-1 little endian, 1 big endian and 0 host order] * return: -1 error or get value */ uint32_t @@ -296,7 +314,16 @@ rpack_uint32(rpack_t * __restrict rp, uint32_t * __res if (!(next = rpack_align_and_reserve(rp, sizeof(uint32_t)))) return (uint32_t) -1; - u = be ? EXTRACT_BE_32(next) : EXTRACT_LE_32(next); + if (be < 0) + u = EXTRACT_LE_32(next); + else if (be > 0) + u = EXTRACT_BE_32(next); + else +#if BYTE_ORDER == BIG_ENDIAN + u = EXTRACT_BE_32(next); +#else + u = EXTRACT_LE_32(next); +#endif if (n) RPACK_SET_32(next, n); @@ -309,7 +336,7 @@ rpack_uint32(rpack_t * __restrict rp, uint32_t * __res * * @rp = raw buffer * @n = set value if !=NULL - * @be = extract in big-endian + * @be = byte order [-1 little endian, 1 big endian and 0 host order] * return: -1 error or get value */ uint64_t @@ -324,7 +351,16 @@ rpack_uint64(rpack_t * __restrict rp, uint64_t * __res if (!(next = rpack_align_and_reserve(rp, sizeof(uint64_t)))) return (uint64_t) -1; - u = be ? EXTRACT_BE_64(next) : EXTRACT_LE_64(next); + if (be < 0) + u = EXTRACT_LE_64(next); + else if (be > 0) + u = EXTRACT_BE_64(next); + else +#if BYTE_ORDER == BIG_ENDIAN + u = EXTRACT_BE_64(next); +#else + u = EXTRACT_LE_64(next); +#endif if (n) RPACK_SET_64(next, n); @@ -402,7 +438,7 @@ rpack_rdata(rpack_t * __restrict rp, void * __restrict * * @rp = raw buffer * @n = set value if !=NULL - * @be = extract in big-endian + * @be = byte order [-1 little endian, 1 big endian and 0 host order] * return: -1 error or get value */ uint16_t @@ -416,7 +452,16 @@ rpack_ruint16(rpack_t * __restrict rp, uint16_t * __re if (sizeof(uint16_t) + rp->r_next - rp->r_buf > rp->r_len) return (uint16_t) -1; - u = be ? EXTRACT_BE_16(rp->r_next) : EXTRACT_LE_16(rp->r_next); + if (be < 0) + u = EXTRACT_LE_16(rp->r_next); + else if (be > 0) + u = EXTRACT_BE_16(rp->r_next); + else +#if BYTE_ORDER == BIG_ENDIAN + u = EXTRACT_BE_16(rp->r_next); +#else + u = EXTRACT_LE_16(rp->r_next); +#endif if (n) RPACK_SET_16(rp->r_next, n); @@ -429,7 +474,7 @@ rpack_ruint16(rpack_t * __restrict rp, uint16_t * __re * * @rp = raw buffer * @n = set value if !=NULL - * @be = extract in big-endian + * @be = byte order [-1 little endian, 1 big endian and 0 host order] * return: -1 error or get value */ uint32_t @@ -443,7 +488,16 @@ rpack_ruint24(rpack_t * __restrict rp, uint32_t * __re if (sizeof(uint32_t) + rp->r_next - rp->r_buf > rp->r_len) return (uint32_t) -1; - u = be ? EXTRACT_BE_24(rp->r_next) : EXTRACT_LE_24(rp->r_next); + if (be < 0) + u = EXTRACT_LE_24(rp->r_next); + else if (be > 0) + u = EXTRACT_BE_24(rp->r_next); + else +#if BYTE_ORDER == BIG_ENDIAN + u = EXTRACT_BE_24(rp->r_next); +#else + u = EXTRACT_LE_24(rp->r_next); +#endif if (n) RPACK_SET_24(rp->r_next, n); @@ -456,7 +510,7 @@ rpack_ruint24(rpack_t * __restrict rp, uint32_t * __re * * @rp = raw buffer * @n = set value if !=NULL - * @be = extract in big-endian + * @be = byte order [-1 little endian, 1 big endian and 0 host order] * return: -1 error or get value */ uint32_t @@ -470,7 +524,16 @@ rpack_ruint32(rpack_t * __restrict rp, uint32_t * __re if (sizeof(uint32_t) + rp->r_next - rp->r_buf > rp->r_len) return (uint32_t) -1; - u = be ? EXTRACT_BE_32(rp->r_next) : EXTRACT_LE_32(rp->r_next); + if (be < 0) + u = EXTRACT_LE_32(rp->r_next); + else if (be > 0) + u = EXTRACT_BE_32(rp->r_next); + else +#if BYTE_ORDER == BIG_ENDIAN + u = EXTRACT_BE_32(rp->r_next); +#else + u = EXTRACT_LE_32(rp->r_next); +#endif if (n) RPACK_SET_32(rp->r_next, n); @@ -483,7 +546,7 @@ rpack_ruint32(rpack_t * __restrict rp, uint32_t * __re * * @rp = raw buffer * @n = set value if !=NULL - * @be = extract in big-endian + * @be = byte order [-1 little endian, 1 big endian and 0 host order] * return: -1 error or get value */ uint64_t @@ -497,10 +560,67 @@ rpack_ruint64(rpack_t * __restrict rp, uint64_t * __re if (sizeof(uint64_t) + rp->r_next - rp->r_buf > rp->r_len) return (uint64_t) -1; - u = be ? EXTRACT_BE_64(rp->r_next) : EXTRACT_LE_64(rp->r_next); + if (be < 0) + u = EXTRACT_LE_64(rp->r_next); + else if (be > 0) + u = EXTRACT_BE_64(rp->r_next); + else +#if BYTE_ORDER == BIG_ENDIAN + u = EXTRACT_BE_64(rp->r_next); +#else + u = EXTRACT_LE_64(rp->r_next); +#endif if (n) RPACK_SET_64(rp->r_next, n); rp->r_next += sizeof(uint64_t); return u; +} + +/* + * rpack_next() - Get and set current position + * + * @rp = raw packet + * @after_len = move aligned current position after length + * return: NULL error or current position + */ +uint8_t * +rpack_next(rpack_t * __restrict rp, size_t after_len) +{ + uint8_t *cur = NULL, *next = NULL; + + if (!RPACK_SANITY(rp)) + return NULL; + /* No space left */ + if (!(next = rpack_align_and_reserve(rp, after_len))) + return NULL; + + cur = rp->r_next; + + rp->r_next = next + after_len; + return cur; +} + +/* + * rpack_rnext() - Get and set raw current position + * + * @rp = raw packet + * @after_len = !=0 move current position after length + * return: NULL error or raw current position + */ +uint8_t * +rpack_rnext(rpack_t * __restrict rp, size_t after_len) +{ + uint8_t *next = NULL; + + if (!RPACK_SANITY(rp)) + return NULL; + /* No space left */ + if (after_len + rp->r_next - rp->r_buf > rp->r_len) + return NULL; + + next = rp->r_next; + + rp->r_next += after_len; + return next; }