diff options
author | Ashlee Young <ashlee@onosfw.com> | 2015-09-09 22:21:41 -0700 |
---|---|---|
committer | Ashlee Young <ashlee@onosfw.com> | 2015-09-09 22:21:41 -0700 |
commit | 8879b125d26e8db1a5633de5a9c692eb2d1c4f83 (patch) | |
tree | c7259d85a991b83dfa85ab2e339360669fc1f58e /framework/src/suricata/src/util-byte.h | |
parent | 13d05bc8458758ee39cb829098241e89616717ee (diff) |
suricata checkin based on commit id a4bce14770beee46a537eda3c3f6e8e8565d5d0a
Change-Id: I9a214fa0ee95e58fc640e50bd604dac7f42db48f
Diffstat (limited to 'framework/src/suricata/src/util-byte.h')
-rw-r--r-- | framework/src/suricata/src/util-byte.h | 292 |
1 files changed, 292 insertions, 0 deletions
diff --git a/framework/src/suricata/src/util-byte.h b/framework/src/suricata/src/util-byte.h new file mode 100644 index 00000000..82c16a4d --- /dev/null +++ b/framework/src/suricata/src/util-byte.h @@ -0,0 +1,292 @@ +/* Copyright (C) 2007-2010 Open Information Security Foundation + * + * You can copy, redistribute or modify this Program under the terms of + * the GNU General Public License version 2 as published by the Free + * Software Foundation. + * + * 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 + * version 2 along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA + * 02110-1301, USA. + */ + +/** + * \file + * + * \author Brian Rectanus <brectanu@gmail.com> + */ + +#ifndef __UTIL_BYTE_H__ +#define __UTIL_BYTE_H__ + +#include <stdint.h> + +#define BYTE_BIG_ENDIAN 0 +#define BYTE_LITTLE_ENDIAN 1 + +/** Wrappers for OS dependent byte swapping functions */ +#ifdef OS_FREEBSD +#include <sys/endian.h> +#define SCByteSwap16(x) bswap16(x) +#define SCByteSwap32(x) bswap32(x) +#define SCByteSwap64(x) bswap64(x) +#elif defined __OpenBSD__ +#include <sys/types.h> +#define SCByteSwap16(x) swap16(x) +#define SCByteSwap32(x) swap32(x) +#define SCByteSwap64(x) swap64(x) +#elif OS_DARWIN +#include <libkern/OSByteOrder.h> +#define SCByteSwap16(x) OSSwapInt16(x) +#define SCByteSwap32(x) OSSwapInt32(x) +#define SCByteSwap64(x) OSSwapInt64(x) +#elif defined(__WIN32) || defined(_WIN32) +/* Quick & dirty solution, nothing seems to exist for this in Win32 API */ +#define SCByteSwap16(x) \ + ((((x) & 0xff00) >> 8) \ + | (((x) & 0x00ff) << 8)) +#define SCByteSwap32(x) \ + ((((x) & 0xff000000) >> 24) \ + | (((x) & 0x00ff0000) >> 8) \ + | (((x) & 0x0000ff00) << 8) \ + | (((x) & 0x000000ff) << 24)) +#define SCByteSwap64(x) \ + ((((x) & 0xff00000000000000ull) >> 56) \ + | (((x) & 0x00ff000000000000ull) >> 40) \ + | (((x) & 0x0000ff0000000000ull) >> 24) \ + | (((x) & 0x000000ff00000000ull) >> 8) \ + | (((x) & 0x00000000ff000000ull) << 8) \ + | (((x) & 0x0000000000ff0000ull) << 24) \ + | (((x) & 0x000000000000ff00ull) << 40) \ + | (((x) & 0x00000000000000ffull) << 56)) +#else +#include <byteswap.h> +#define SCByteSwap16(x) bswap_16(x) +#define SCByteSwap32(x) bswap_32(x) +#define SCByteSwap64(x) bswap_64(x) +#endif /* OS_FREEBSD */ + +/** \brief Turn byte array into string. + * + * All non-printables are copied over, except for '\0', which is + * turned into literal \0 in the string. + * + * \param bytes byte array + * \param nbytes number of bytes + * \return string nul-terminated string or NULL on error + */ +char *BytesToString(const uint8_t *bytes, size_t nbytes); + +/** + * Extract bytes from a byte string and convert to a unint64_t. + * + * \param res Stores result + * \param e Endianness (BYTE_BIG_ENDIAN or BYTE_LITTLE_ENDIAN) + * \param len Number of bytes to extract (8 max) + * \param bytes Data to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractUint64(uint64_t *res, int e, uint16_t len, const uint8_t *bytes); + +/** + * Extract bytes from a byte string and convert to a unint32_t. + * + * \param res Stores result + * \param e Endianness (BYTE_BIG_ENDIAN or BYTE_LITTLE_ENDIAN) + * \param len Number of bytes to extract (8 max) + * \param bytes Data to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractUint32(uint32_t *res, int e, uint16_t len, const uint8_t *bytes); + +/** + * Extract bytes from a byte string and convert to a unint16_t. + * + * \param res Stores result + * \param e Endianness (BYTE_BIG_ENDIAN or BYTE_LITTLE_ENDIAN) + * \param len Number of bytes to extract (8 max) + * \param bytes Data to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractUint16(uint16_t *res, int e, uint16_t len, const uint8_t *bytes); + +/** + * Extract unsigned integer value from a string. + * + * \param res Stores result + * \param base Base of the number to extract + * \param len Number of bytes to extract (23 max or 0 for unbounded) + * \param str String to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractString(uint64_t *res, int base, uint16_t len, const char *str); + +/** + * Extract unsigned integer value from a string as uint64_t. + * + * \param res Stores result + * \param base Base of the number to extract + * \param len Number of bytes to extract (23 max or 0 for unbounded) + * \param len Number of bytes to extract (23 max) + * \param str String to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractStringUint64(uint64_t *res, int base, uint16_t len, const char *str); + +/** + * Extract unsigned integer value from a string as uint32_t. + * + * \param res Stores result + * \param base Base of the number to extract + * \param len Number of bytes to extract (23 max or 0 for unbounded) + * \param str String to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractStringUint32(uint32_t *res, int base, uint16_t len, const char *str); + +/** + * Extract unsigned integer value from a string as uint16_t. + * + * \param res Stores result + * \param base Base of the number to extract + * \param len Number of bytes to extract (23 max or 0 for unbounded) + * \param str String to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractStringUint16(uint16_t *res, int base, uint16_t len, const char *str); + +/** + * Extract unsigned integer value from a string as uint8_t. + * + * \param res Stores result + * \param base Base of the number to extract + * \param len Number of bytes to extract (23 max or 0 for unbounded) + * \param str String to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractStringUint8(uint8_t *res, int base, uint16_t len, const char *str); + +/** + * Extract signed integer value from a string. + * + * \param res Stores result + * \param base Base of the number to extract + * \param len Number of bytes to extract (23 max or 0 for unbounded) + * \param str String to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractStringSigned(int64_t *res, int base, uint16_t len, const char *str); + +/** + * Extract signed integer value from a string as uint64_t. + * + * \param res Stores result + * \param base Base of the number to extract + * \param len Number of bytes to extract (23 max or 0 for unbounded) + * \param str String to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractStringInt64(int64_t *res, int base, uint16_t len, const char *str); + +/** + * Extract signed integer value from a string as uint32_t. + * + * \param res Stores result + * \param base Base of the number to extract + * \param len Number of bytes to extract (23 max or 0 for unbounded) + * \param str String to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractStringInt32(int32_t *res, int base, uint16_t len, const char *str); + +/** + * Extract signed integer value from a string as uint16_t. + * + * \param res Stores result + * \param base Base of the number to extract + * \param len Number of bytes to extract (23 max or 0 for unbounded) + * \param str String to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractStringInt16(int16_t *res, int base, uint16_t len, const char *str); + +/** + * Extract signed integer value from a string as uint8_t. + * + * \param res Stores result + * \param base Base of the number to extract + * \param len Number of bytes to extract (23 max or 0 for unbounded) + * \param str String to extract from + * + * \return n Number of bytes extracted on success + * \return -1 On error + */ +int ByteExtractStringInt8(int8_t *res, int base, uint16_t len, const char *str); + +#ifdef UNITTESTS +void ByteRegisterTests(void); +#endif /* UNITTESTS */ + +/** ------ Inline functions ----- */ +static inline int ByteExtract(uint64_t *res, int e, uint16_t len, const uint8_t *bytes) +{ + uint64_t b = 0; + int i; + + if ((e != BYTE_BIG_ENDIAN) && (e != BYTE_LITTLE_ENDIAN)) { + /** \todo Need standard return values */ + return -1; + } + + *res = 0; + + /* Go through each byte and merge it into the result in the correct order */ + /** \todo Probably a more efficient way to do this. */ + for (i = 0; i < len; i++) { + + if (e == BYTE_LITTLE_ENDIAN) { + b = bytes[i]; + } + else { + b = bytes[len - i - 1]; + } + + *res |= (b << ((i & 7) << 3)); + + } + + return len; +} + + +#endif /* __UTIL_BYTE_H__ */ + |