inet.hpp

Go to the documentation of this file.

/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 *
 * This file is part of zmqpp.
 * Copyright (c) 2011-2015 Contributors as noted in the AUTHORS file.
 */

#ifndef ZMQPP_INET_HPP_
#define ZMQPP_INET_HPP_

#include <cstdint>

// We get htons and htonl from here
#ifdef _WIN32
#include <WinSock2.h>
#else
#include <netinet/in.h>
#endif

#include "compatibility.hpp"

namespace zmqpp
{

ZMQPP_COMPARABLE_ENUM order {
    big_endian,   
    little_endian 
};

inline uint64_t swap_if_needed(uint64_t const value_to_check)
{
    static order host_order = (htonl(42) == 42) ? order::big_endian : order::little_endian;

    if (order::big_endian == host_order)
    {
        return value_to_check;
    }

    union {
        uint64_t integer;
        uint8_t bytes[8];
    } value { value_to_check };

    std::swap(value.bytes[0], value.bytes[7]);
    std::swap(value.bytes[1], value.bytes[6]);
    std::swap(value.bytes[2], value.bytes[5]);
    std::swap(value.bytes[3], value.bytes[4]);

    return value.integer;
}

#ifndef htonll
inline uint64_t htonll(uint64_t const hostlonglong)
{
    return zmqpp::swap_if_needed(hostlonglong);
}
#endif

#ifndef ntohll
inline uint64_t ntohll(uint64_t const networklonglong)
{
    return zmqpp::swap_if_needed(networklonglong);
}
#endif

inline float htonf(float value)
{
    assert(sizeof(float) == sizeof(uint32_t));

    uint32_t temp;
    memcpy(&temp, &value, sizeof(uint32_t));
    temp = htonl( temp );
    memcpy(&value, &temp, sizeof(uint32_t));

    return value;
}

inline float ntohf(float value)
{
    assert(sizeof(float) == sizeof(uint32_t));

    uint32_t temp;
    memcpy(&temp, &value, sizeof(uint32_t));
    temp = ntohl( temp );
    memcpy(&value, &temp, sizeof(uint32_t));

    return value;
}

inline double htond(double value)
{
    assert(sizeof(double) == sizeof(uint64_t));

    uint64_t temp;
    memcpy(&temp, &value, sizeof(uint64_t));
    temp = htonll(temp);
    memcpy(&value, &temp, sizeof(uint64_t));

    return value;
}

inline double ntohd(double value)
{
    assert(sizeof(double) == sizeof(uint64_t));

    uint64_t temp;
    memcpy(&temp, &value, sizeof(uint64_t));
    temp = ntohll(temp);
    memcpy(&value, &temp, sizeof(uint64_t));

    return value;
}

}

#endif /* INET_HPP_ */