blob: e628ff06aff8148b80f6321e062455770a9d25f5 [file] [log] [blame]
/**
* Copyright © 2018 Intel Corporation
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#pragma once
#include <algorithm>
#include <boost/asio/spawn.hpp>
#include <cstdint>
#include <ipmid/message/types.hpp>
#include <memory>
#include <phosphor-logging/log.hpp>
#include <tuple>
#include <utility>
#include <vector>
namespace ipmi
{
struct Context
{
using ptr = std::shared_ptr<Context>;
Context() = default;
Context(NetFn netFn, Cmd cmd, int channel, int userId, Privilege priv,
boost::asio::yield_context* yield = nullptr) :
netFn(netFn),
cmd(cmd), channel(channel), userId(userId), priv(priv), yield(yield)
{
}
// normal IPMI context (what call is this, from whence it came...)
NetFn netFn = 0;
Cmd cmd = 0;
int channel = 0;
int userId = 0;
Privilege priv = Privilege::None;
// if non-null, use this to do blocking asynchronous asio calls
boost::asio::yield_context* yield = nullptr;
};
namespace message
{
namespace details
{
template <typename A>
struct UnpackSingle;
template <typename T>
using UnpackSingle_t = UnpackSingle<utility::TypeIdDowncast_t<T>>;
template <typename A>
struct PackSingle;
template <typename T>
using PackSingle_t = PackSingle<utility::TypeIdDowncast_t<T>>;
// size to hold 64 bits plus one (possibly-)partial byte
static constexpr size_t bitStreamSize = ((sizeof(uint64_t) + 1) * CHAR_BIT);
} // namespace details
/**
* @brief a payload class that provides a mechanism to pack and unpack data
*
* When a new request is being executed, the Payload class is responsible for
* attempting to unpack all the required arguments from the incoming blob. For
* variable-length functions, it is possible to have function signature have a
* Payload object, which will then allow the remaining data to be extracted as
* needed.
*
* When creating a response, the parameters returned from the callback use a
* newly created payload object to pack all the parameters into a buffer that is
* then returned to the requester.
*
* These interfaces make calls into the message/pack.hpp and message/unpack.hpp
* functions.
*/
struct Payload
{
Payload() = default;
Payload(const Payload&) = default;
Payload& operator=(const Payload&) = default;
Payload(Payload&&) = default;
Payload& operator=(Payload&&) = default;
explicit Payload(std::vector<uint8_t>&& data) :
raw(std::move(data)), unpackCheck(false)
{
}
~Payload()
{
using namespace phosphor::logging;
if (trailingOk && !unpackCheck && !fullyUnpacked())
{
log<level::ERR>("Failed to check request for full unpack");
}
}
/******************************************************************
* raw vector access
*****************************************************************/
/**
* @brief return the size of the underlying raw buffer
*/
size_t size() const
{
return raw.size();
}
/**
* @brief resize the underlying raw buffer to a new size
*
* @param sz - new size for the buffer
*/
void resize(size_t sz)
{
raw.resize(sz);
}
/**
* @brief return a pointer to the underlying raw buffer
*/
uint8_t* data()
{
return raw.data();
}
/**
* @brief return a const pointer to the underlying raw buffer
*/
const uint8_t* data() const
{
return raw.data();
}
/******************************************************************
* Response operations
*****************************************************************/
/**
* @brief append a series of bytes to the buffer
*
* @tparam T - the type pointer to return; must be compatible to a byte
*
* @param begin - a pointer to the beginning of the series
* @param end - a pointer to the end of the series
*/
template <typename T>
void append(T* begin, T* end)
{
static_assert(
std::is_same_v<utility::TypeIdDowncast_t<T>, int8_t> ||
std::is_same_v<utility::TypeIdDowncast_t<T>, uint8_t> ||
std::is_same_v<utility::TypeIdDowncast_t<T>, char>,
"begin and end must be signed or unsigned byte pointers");
// this interface only allows full-byte access; pack in partial bytes
drain();
raw.insert(raw.end(), reinterpret_cast<const uint8_t*>(begin),
reinterpret_cast<const uint8_t*>(end));
}
/**
* @brief append a series of bits to the buffer
*
* Only the lowest @count order of bits will be appended, with the most
* significant of those bits getting appended first.
*
* @param count - number of bits to append
* @param bits - a byte with count significant bits to append
*/
void appendBits(size_t count, uint8_t bits)
{
// drain whole bytes out
drain(true);
// add in the new bits as the higher-order bits, filling LSBit first
fixed_uint_t<details::bitStreamSize> tmp = bits;
tmp <<= bitCount;
bitStream |= tmp;
bitCount += count;
// drain any whole bytes we have appended
drain(true);
}
/**
* @brief empty out the bucket and pack it as bytes LSB-first
*
* @param wholeBytesOnly - if true, only the whole bytes will be drained
*/
void drain(bool wholeBytesOnly = false)
{
while (bitCount > 0)
{
uint8_t retVal;
if (bitCount < CHAR_BIT)
{
if (wholeBytesOnly)
{
break;
}
}
size_t bitsOut = std::min(static_cast<size_t>(CHAR_BIT), bitCount);
retVal = static_cast<uint8_t>(bitStream);
raw.push_back(retVal);
bitStream >>= bitsOut;
bitCount -= bitsOut;
}
}
// base empty pack
int pack()
{
return 0;
}
/**
* @brief pack arbitrary values (of any supported type) into the buffer
*
* @tparam Arg - the type of the first argument
* @tparam Args - the type of the optional remaining arguments
*
* @param arg - the first argument to pack
* @param args... - the optional remaining arguments to pack
*
* @return int - non-zero on pack errors
*/
template <typename Arg, typename... Args>
int pack(Arg&& arg, Args&&... args)
{
int packRet =
details::PackSingle_t<Arg>::op(*this, std::forward<Arg>(arg));
if (packRet)
{
return packRet;
}
packRet = pack(std::forward<Args>(args)...);
drain();
return packRet;
}
/**
* @brief pack a tuple of values (of any supported type) into the buffer
*
* This will pack the elements of the tuple as if each one was passed in
* individually, as if passed into the above variadic function.
*
* @tparam Types - the implicitly declared list of the tuple element types
*
* @param t - the tuple of values to pack
*
* @return int - non-zero on pack errors
*/
template <typename... Types>
int pack(std::tuple<Types...>& t)
{
return std::apply([this](Types&... args) { return pack(args...); }, t);
}
/******************************************************************
* Request operations
*****************************************************************/
/**
* @brief pop a series of bytes from the raw buffer
*
* @tparam T - the type pointer to return; must be compatible to a byte
*
* @param count - the number of bytes to return
*
* @return - a tuple of pointers (begin,begin+count)
*/
template <typename T>
auto pop(size_t count)
{
static_assert(
std::is_same_v<utility::TypeIdDowncast_t<T>, int8_t> ||
std::is_same_v<utility::TypeIdDowncast_t<T>, uint8_t> ||
std::is_same_v<utility::TypeIdDowncast_t<T>, char>,
"T* must be signed or unsigned byte pointers");
// this interface only allows full-byte access; skip partial bits
if (bitCount)
{
// WARN on unused bits?
discardBits();
}
if (count <= (raw.size() - rawIndex))
{
auto range = std::make_tuple(
reinterpret_cast<T*>(raw.data() + rawIndex),
reinterpret_cast<T*>(raw.data() + rawIndex + count));
rawIndex += count;
return range;
}
unpackError = true;
return std::make_tuple(reinterpret_cast<T*>(NULL),
reinterpret_cast<T*>(NULL));
}
/**
* @brief fill bit stream with at least count bits for consumption
*
* @param count - number of bit needed
*
* @return - unpackError
*/
bool fillBits(size_t count)
{
// add more bits to the top end of the bitstream
// so we consume bits least-significant first
if (count > (details::bitStreamSize - CHAR_BIT))
{
unpackError = true;
return unpackError;
}
while (bitCount < count)
{
if (rawIndex < raw.size())
{
fixed_uint_t<details::bitStreamSize> tmp = raw[rawIndex++];
tmp <<= bitCount;
bitStream |= tmp;
bitCount += CHAR_BIT;
}
else
{
// raw has run out of bytes to pop
unpackError = true;
return unpackError;
}
}
return false;
}
/**
* @brief consume count bits from bitstream (must call fillBits first)
*
* @param count - number of bit needed
*
* @return - count bits from stream
*/
uint8_t popBits(size_t count)
{
if (bitCount < count)
{
unpackError = true;
return 0;
}
// consume bits low-order bits first
auto bits = bitStream.convert_to<uint8_t>();
bits &= ((1 << count) - 1);
bitStream >>= count;
bitCount -= count;
return bits;
}
/**
* @brief discard all partial bits
*/
void discardBits()
{
bitStream = 0;
bitCount = 0;
}
/**
* @brief fully reset the unpack stream
*/
void reset()
{
discardBits();
rawIndex = 0;
unpackError = false;
}
/**
* @brief check to see if the stream has been fully unpacked
*
* @return bool - true if the stream has been unpacked and has no errors
*/
bool fullyUnpacked()
{
unpackCheck = true;
return raw.size() == rawIndex && bitCount == 0 && !unpackError;
}
// base empty unpack
int unpack()
{
return 0;
}
/**
* @brief unpack arbitrary values (of any supported type) from the buffer
*
* @tparam Arg - the type of the first argument
* @tparam Args - the type of the optional remaining arguments
*
* @param arg - the first argument to unpack
* @param args... - the optional remaining arguments to unpack
*
* @return int - non-zero for unpack error
*/
template <typename Arg, typename... Args>
int unpack(Arg&& arg, Args&&... args)
{
int unpackRet =
details::UnpackSingle_t<Arg>::op(*this, std::forward<Arg>(arg));
if (unpackRet)
{
unpackError = true;
return unpackRet;
}
return unpack(std::forward<Args>(args)...);
}
/**
* @brief unpack a tuple of values (of any supported type) from the buffer
*
* This will unpack the elements of the tuple as if each one was passed in
* individually, as if passed into the above variadic function.
*
* @tparam Types - the implicitly declared list of the tuple element types
*
* @param t - the tuple of values to unpack
*
* @return int - non-zero on unpack error
*/
template <typename... Types>
int unpack(std::tuple<Types...>& t)
{
// roll back checkpoint so that unpacking a tuple is atomic
size_t priorBitCount = bitCount;
size_t priorIndex = rawIndex;
fixed_uint_t<details::bitStreamSize> priorBits = bitStream;
int ret =
std::apply([this](Types&... args) { return unpack(args...); }, t);
if (ret)
{
bitCount = priorBitCount;
bitStream = priorBits;
rawIndex = priorIndex;
}
return ret;
}
// partial bytes in the form of bits
fixed_uint_t<details::bitStreamSize> bitStream;
size_t bitCount = 0;
std::vector<uint8_t> raw;
size_t rawIndex = 0;
bool trailingOk = false;
bool unpackCheck = true;
bool unpackError = false;
};
/**
* @brief high-level interface to an IPMI response
*
* Make it easy to just pack in the response args from the callback into a
* buffer that goes back to the requester.
*/
struct Response
{
/* Define all of the basic class operations:
* Not allowed:
* - Default constructor to avoid nullptrs.
* Allowed:
* - Copy operations.
* - Move operations.
* - Destructor.
*/
Response() = delete;
Response(const Response&) = default;
Response& operator=(const Response&) = default;
Response(Response&&) = default;
Response& operator=(Response&&) = default;
~Response() = default;
using ptr = std::shared_ptr<Response>;
explicit Response(Context::ptr& context) :
payload(), ctx(context), cc(ccSuccess)
{
}
/**
* @brief pack arbitrary values (of any supported type) into the payload
*
* @tparam Args - the type of the optional arguments
*
* @param args... - the optional arguments to pack
*
* @return int - non-zero on pack errors
*/
template <typename... Args>
int pack(Args&&... args)
{
return payload.pack(std::forward<Args>(args)...);
}
/**
* @brief pack a tuple of values (of any supported type) into the payload
*
* This will pack the elements of the tuple as if each one was passed in
* individually, as if passed into the above variadic function.
*
* @tparam Types - the implicitly declared list of the tuple element types
*
* @param t - the tuple of values to pack
*
* @return int - non-zero on pack errors
*/
template <typename... Types>
int pack(std::tuple<Types...>& t)
{
return payload.pack(t);
}
Payload payload;
Context::ptr ctx;
Cc cc;
};
/**
* @brief high-level interface to an IPMI request
*
* Make it easy to unpack the buffer into the request args for the callback.
*/
struct Request
{
/* Define all of the basic class operations:
* Not allowed:
* - Default constructor to avoid nullptrs.
* Allowed:
* - Copy operations.
* - Move operations.
* - Destructor.
*/
Request() = delete;
Request(const Request&) = default;
Request& operator=(const Request&) = default;
Request(Request&&) = default;
Request& operator=(Request&&) = default;
~Request() = default;
using ptr = std::shared_ptr<Request>;
explicit Request(Context::ptr context, std::vector<uint8_t>&& d) :
payload(std::forward<std::vector<uint8_t>>(d)), ctx(context)
{
}
/**
* @brief unpack arbitrary values (of any supported type) from the payload
*
* @tparam Args - the type of the optional arguments
*
* @param args... - the optional arguments to unpack
*
* @return int - non-zero for unpack error
*/
template <typename... Args>
int unpack(Args&&... args)
{
int unpackRet = payload.unpack(std::forward<Args>(args)...);
if (unpackRet == ipmi::ccSuccess)
{
if (!payload.trailingOk)
{
if (!payload.fullyUnpacked())
{
// not all bits were consumed by requested parameters
return ipmi::ccReqDataLenInvalid;
}
payload.unpackCheck = false;
}
}
return unpackRet;
}
/**
* @brief unpack a tuple of values (of any supported type) from the payload
*
* This will unpack the elements of the tuple as if each one was passed in
* individually, as if passed into the above variadic function.
*
* @tparam Types - the implicitly declared list of the tuple element types
*
* @param t - the tuple of values to unpack
*
* @return int - non-zero on unpack error
*/
template <typename... Types>
int unpack(std::tuple<Types...>& t)
{
return std::apply([this](Types&... args) { return unpack(args...); },
t);
}
/** @brief Create a response message that corresponds to this request
*
* @return A shared_ptr to the response message created
*/
Response::ptr makeResponse()
{
return std::make_shared<Response>(ctx);
}
Payload payload;
Context::ptr ctx;
};
} // namespace message
} // namespace ipmi
// include packing and unpacking of types
#include <ipmid/message/pack.hpp>
#include <ipmid/message/unpack.hpp>