request request::parse(zmq::message_t& msg) {
request req;
std::string result(static_cast<const char *>(msg.data()), msg.size());
std::vector<std::string> results = utils::split(result, " ", 3);
req.sender = results[0];
req.conn_id = results[1];
req.path = results[2];
std::string body;
std::string ign;
req.headers = utils::parse_json(utils::parse_netstring(results[3], body));
req.body = utils::parse_netstring(body, ign);
//check disconnect flag
req.disconnect = false;
for (std::vector<header>::const_iterator it = req.headers.begin();
it != req.headers.end(); ++it) {
if (it->first == "METHOD" && it->second == "JSON" &&
req.body == "{\"type\":\"disconnect\"}") {
req.disconnect = true;
break;
}
}
return req;
}
bool SocketToTF::updateTransformFromSocketTransform(zmq::message_t& message) {
socket_to_tf::TransformStamped transform;
boost::iostreams::basic_array_source<char> source((char*)message.data(), message.size());
boost::iostreams::stream<boost::iostreams::basic_array_source <char> > input_stream(source);
boost::archive::binary_iarchive ia(input_stream);
ia >> transform;
if (boost::math::isfinite(transform.x) && boost::math::isfinite(transform.y) && boost::math::isfinite(transform.z) &&
boost::math::isfinite(transform.qx) && boost::math::isfinite(transform.qy) && boost::math::isfinite(transform.qz) && boost::math::isfinite(transform.qw)) {
transform_stamped_.transform.translation.x = transform.x;
transform_stamped_.transform.translation.y = transform.y;
transform_stamped_.transform.translation.z = transform.z;
transform_stamped_.transform.rotation.x = transform.qx;
transform_stamped_.transform.rotation.y = transform.qy;
transform_stamped_.transform.rotation.z = transform.qz;
transform_stamped_.transform.rotation.w = transform.qw;
transform_stamped_.child_frame_id = transform.source_frame;
transform_stamped_.header.frame_id = transform.target_frame;
transform_stamped_.header.stamp.sec = transform.timestamp_seconds;
transform_stamped_.header.stamp.nsec = transform.timestamp_nanoseconds;
std::stringstream ss_data;
ss_data << "{ " << transform.x << " " << transform.y << " " << transform.z << " " << transform.qx << " " << transform.qy << " " << transform.qz << " " << transform.qw << " " << transform.source_frame << " " << transform.target_frame << " }";
std::string transform_data = ss_data.str();
ROS_INFO_STREAM("Received message with size " << message.size() << ": " << transform_data);
return true;
}
return false;
}
inline T
get_string(zmq::message_t& message, size_t limit)
{
return T(
static_cast<char*>(message.data()), std::min(message.size(), limit)
);
}
/**
* Return memory region as RawMessage. No copying takes place.
*/
ZMQMESSAGE_DLL_PUBLIC
inline
RawMessage
get_raw(zmq::message_t& message)
{
return RawMessage(message.data(), message.size());
}
inline
T
get(zmq::message_t& message,
typename Private::EnableIf<Private::IsStr<T>::value>::type* = 0)
{
return T(static_cast<char*>(message.data()), message.size());
}
ZMQMESSAGE_DLL_PUBLIC
inline
T
get_string(zmq::message_t& message)
{
return T(static_cast<char*>(message.data()), message.size());
}
Msg(const zmq::message_t &message) {
msgpack::unpacked msg;
msgpack::unpack(&msg, (const char *)message.data(), message.size());
msgpack::object obj = msg.get();
obj.convert(&data);
}
static bool unpack(zmq::message_t& message, std::string& value) {
value.assign(
static_cast<const char*>(message.data()),
message.size()
);
return true;
}
/**
* Compare message contents to specified memory region.
* @return like @c memcmp
*/
ZMQMESSAGE_DLL_PUBLIC
inline
int
msgcmp(zmq::message_t& message, const char* str, size_t len)
{
int ret = memcmp(message.data(), str, std::min(message.size(), len));
return ret ? ret : message.size() - len;
}
ZMQMESSAGE_DLL_PUBLIC
inline
void
get(zmq::message_t& message, T& t,
typename Private::EnableIf<Private::IsStr<T>::value>::type* = 0)
{
t = T(static_cast<char*>(message.data()), message.size());
}
bool SocketToTF::updateTransformFromSocketPointTranslation(zmq::message_t& message) {
socket_to_tf::PointTranslation point_translation;
if (use_boost_to_parse_point_translation_message_) {
boost::iostreams::basic_array_source<char> source((char*)message.data(), message.size());
boost::iostreams::stream<boost::iostreams::basic_array_source <char> > input_stream(source);
boost::archive::binary_iarchive ia(input_stream);
ia >> point_translation;
} else {
inline
void
get(zmq::message_t& message, T& t,
typename Private::DisableIf<Private::IsStr<T>::value>::type* = 0,
typename Private::DisableIf<Private::IsRaw<T>::value>::type* = 0)
{
std::stringstream ss;
ss.write(static_cast<char*>(message.data()), message.size());
ss >> t;
}
static inline
void
unpack(/* const */ zmq::message_t& message,
std::string& value)
{
value.assign(
static_cast<const char*>(message.data()),
message.size()
);
}
static bool recv_zmq_message(zmq::socket_t& sock,
zmq::message_t& msg,
T& object,
int flags = ZMQ_NOBLOCK)
{
if (!sock.recv(&msg, flags)) {
return false;
}
memcpy(&object, msg.data(), msg.size());
return true;
}
static inline
void
pack(zmq::message_t& message,
const std::string& value)
{
message.rebuild(value.size());
std::memcpy(
message.data(),
value.data(),
value.size()
);
}
ZMQMESSAGE_DLL_PUBLIC
inline
T
get(zmq::message_t& message,
typename Private::DisableIf<Private::IsStr<T>::value>::type* = 0,
typename Private::DisableIf<Private::IsRaw<T>::value>::type* = 0)
{
std::stringstream ss;
ss.write(static_cast<char*>(message.data()), message.size());
T t = T();
ss >> t;
return t;
}
bool
nutils::recv_zmq_message(zmq::socket_t& sock,
zmq::message_t& msg,
std::string& str,
int flags)
{
if (!sock.recv(&msg, flags)) {
return false;
}
str.clear();
str.append(reinterpret_cast<char*>(msg.data()), msg.size());
return true;
}
Request::Request(const zmq::message_t& message) : m_valid(false)
{
std::string in(static_cast<const char*>(message.data()), message.size());
std::istringstream stream(in);
std::string path;
std::string headers;
// Decode message
stream >> m_sender >> m_connectionId;
stream >> path;
headers = utility::readNetString(stream);
m_body = utility::readNetString(stream);
// Decode path
if(!utility::decodeUrl(path, m_path)) {
// TODO: Throw exception
m_path = path;
}
// Parse http header
Json::Reader reader;
if ( !reader.parse(headers, m_headers) )
{
// TODO: Throw exception
}
// Parse json body
if(m_headers.isMember("METHOD") && m_headers["METHOD"] == "JSON") {
if ( !reader.parse(m_body, m_jsonBody) )
{
// TODO: Throw exception
}
}
// Parse query parameters
std::string query = m_headers.get("QUERY", std::string()).asString();
if(!query.empty()) {
std::string decodedQuery;
if(utility::decodeUrl(query, decodedQuery)) {
utility::parseQuery(decodedQuery, m_query);
}
else {
// TODO: Throw exception
}
}
m_valid = true;
}
ZMQMESSAGE_DLL_PUBLIC
inline
void
get_bin(zmq::message_t& message, T& t)
{
t = *(reinterpret_cast<T*>(message.data()));
}
bool
nutils::recv_zmq_message(zmq::socket_t& sock,
zmq::message_t& msg,
msgpack::object& obj,
int flags)
{
if (!sock.recv(&msg, flags)) {
return false;
}
msgpack::unpacked unpacked;
msgpack::unpack(&unpacked, reinterpret_cast<const char*>(msg.data()), msg.size());
obj = unpacked.get();
return true;
}
/** Unpacks a message from a zeromq message */
static Message unpack(const zmq::message_t &msg) {
auto begin = reinterpret_cast<const capnp::word*>(msg.data()); // NOLINT
auto end = reinterpret_cast<const capnp::word*>(static_cast<const char*>(msg.data()) + msg.size()); // NOLINT
kj::ArrayPtr<const capnp::word> ptr(begin, end);
capnp::FlatArrayMessageReader msg_reader(ptr);
capnqml::Message ret;
auto reader = msg_reader.getRoot<serialize::Message>();
ret.setType(static_cast<Message::Type>(reader.getType()));
ret.setEndpoint(reader.getEndpoint().cStr());
ret.setExpectsAnswer(reader.getExpectsAnswer());
ret.setId(reader.getId());
ret.setOriginator(reader.getOriginator());
ret.setSchema(reader.getSchema().cStr());
auto payload = reader.getPayload();
Message::Payload data(payload.begin(), payload.end());
ret.setData(std::move(data));
return ret;
}
void
blastbeat_t::on_body(const std::string& sid,
zmq::message_t& body)
{
stream_map_t::iterator it(
m_streams.find(sid)
);
if(it == m_streams.end()) {
COCAINE_LOG_WARNING(m_log, "received an unknown session body");
return;
}
try {
it->second->write(
static_cast<const char*>(body.data()),
body.size()
);
} catch(const cocaine::error_t& e) {
COCAINE_LOG_ERROR(m_log, "unable to push a body chunk to a session - %s", e.what());
}
}
/**
* Copy contents of one message part to another
*/
inline void
copy_msg(zmq::message_t& target, zmq::message_t& source)
throw(ZmqErrorType)
{
try
{
target.copy(&source);
}
catch (const zmq::error_t& e)
{
throw_zmq_exception(e);
}
}
/**
* Compare message contents to specified memory region.
* @return like @c memcmp
*/
ZMQMESSAGE_DLL_PUBLIC
inline
int
msgcmp(zmq::message_t& message1, zmq::message_t& message2)
{
int ret = memcmp(message1.data(), message2.data(),
std::min(message1.size(), message2.size()));
return ret ? ret : message1.size() - message2.size();
}
void
init_msg(const void* t, size_t sz, zmq::message_t& msg)
{
try
{
void *data = ::malloc(sz);
if (!data)
{
throw zmq::error_t();
}
::memcpy(data, t, sz);
msg.rebuild(data, sz, &zmqmessage_free, 0);
}
catch (const zmq::error_t& e)
{
throw_zmq_exception(e);
}
}
void server_worker(zmq::message_t& msg, std::string str) {
// std::cout << "Received: " << str << std::endl;
std::string uuid_src("");
std::string uuid_dst("");
std::string data("");
if (!str.empty()) {
Utils::ParseTranserJson(str, uuid_src, uuid_dst, data);
if (!uuid_dst.empty() && !data.empty()) {
ServerManager::Instance().InsertMessage(uuid_src, uuid_dst, data);
}
}
From_Message fm = ServerManager::Instance().GetMessage(uuid_src);
std::string content = Utils::BuildTranserJson(fm.uuid_dst, "", fm.data);
std::cout << "received: " << str << " will send: " << content << std::endl;
//memcpy(msg.data(), content.c_str(), content.length()+1); // segment error
msg.rebuild(content.c_str(), content.length());
}
void
blastbeat_t::on_uwsgi(const std::string& sid,
zmq::message_t& message)
{
std::shared_ptr<blastbeat_stream_t> upstream(
std::make_shared<blastbeat_stream_t>(
*this,
sid
)
);
stream_map_t::iterator it;
try {
std::tie(it, std::ignore) = m_streams.emplace(
sid,
engine().enqueue(api::event_t(m_event), upstream)
);
} catch(const cocaine::error_t& e) {
COCAINE_LOG_ERROR(m_log, "unable to enqueue an event - %s", e.what());
return;
}
std::map<
std::string,
std::string
> env;
const char * ptr = static_cast<const char*>(message.data()),
* const end = ptr + message.size();
// NOTE: Skip the uwsgi header, as we already know the message
// length and we don't need uwsgi modifiers.
ptr += sizeof(uwsgi_header_t);
// Parse the HTTP headers.
while(ptr < end) {
const uint16_t * ksz,
* vsz;
ksz = reinterpret_cast<const uint16_t*>(ptr);
ptr += sizeof(*ksz);
std::string key(ptr, *ksz);
ptr += *ksz;
vsz = reinterpret_cast<const uint16_t*>(ptr);
ptr += sizeof(*vsz);
std::string value(ptr, *vsz);
ptr += *vsz;
env[key] = value;
}
// Serialize the headers.
msgpack::sbuffer buffer;
msgpack::packer<msgpack::sbuffer> packer(buffer);
packer << env;
try {
it->second->write(
buffer.data(),
buffer.size()
);
} catch(const cocaine::error_t& e) {
COCAINE_LOG_ERROR(m_log, "unable to push headers to a session - %s", e.what());
}
}
void readReply(zmq::message_t &recMsg){
// cfile is a c file descriptor (not to be confused with a protobuf FileDescriptor object)
int cfile = open("allProto.desc", O_RDONLY);
FileDescriptorSet fds;
// Parse a FileDescriptorSet object directly from the file
// Has the format of a protobuf Message - subclass FileDescriptorSet, defined in <google/protobuf/descriptor.pb.h>
// https://developers.google.com/protocol-buffers/docs/reference/cpp/google.protobuf.descriptor.pb#FieldOptions_CType.details
fds.ParseFromFileDescriptor(cfile);
// Use FileDescriptorSet method to print to screen
// fds.SerializeToOstream(&cout);
close(cfile);
// A DescriptorPool is required: provides methods to identify and manage message types at run-time
// DescriptorPool can be populated from a SimpleDescriptorDatabase, which can be populated with FileDescriptorProto objects
SimpleDescriptorDatabase sddb;
for ( int i = 0; i < fds.file_size() ; i++ ){
//Iterate over the "file" collection in the FileDescriptorSet
//Populate the sddb
sddb.Add(fds.file(i));
}
// Now construct the DescriptorPool
DescriptorPool dp(&sddb);
// DynamicMessageFactory is constucted from a populated DescriptorPool
// DescriptorPool, Descriptor, FieldDescriptor etc.: see descriptor.h -
// https://developers.google.com/protocol-buffers/docs/reference/cpp/google.protobuf.descriptor
DynamicMessageFactory dmf(&dp);
const Descriptor* desc;
const Descriptor* payload_desc;
desc = dp.FindMessageTypeByName("DescribedMessage");
// Example of dynamically creating a message from a Descriptor, retrieved by name string
Message *msg = dmf.GetPrototype(desc)->New();
msg->ParseFromArray(recMsg.data(),recMsg.size());
// Messages with required fields - Need populated.
// Requires FieldDescriptor objects to access
const FieldDescriptor* nameField = desc->FindFieldByName("full_name");
const FieldDescriptor* dataField = desc->FindFieldByName("message");
// Reflection object provides R/W access to dynamic message fields, using FieldDescriptors
// https://developers.google.com/protocol-buffers/docs/reference/cpp/google.protobuf.message#Message.GetReflection.details
// Good example of Reflection at top of that page
const Reflection *msgRefl = msg->GetReflection();
// Make payload message
payload_desc = dp.FindMessageTypeByName(msgRefl->GetString(*msg, nameField));
Message *payload_msg = dmf.GetPrototype(payload_desc)->New();
payload_msg->ParseFromString(msgRefl->GetString( *msg, dataField));
// Reflection of payload message
const Reflection *main_msgRefl = payload_msg->GetReflection();
// Payload fielddescriptors
const FieldDescriptor* main_debugField = payload_desc->FindFieldByName("debug");
main_msgRefl->SetString (payload_msg,main_debugField,"Read");
std::cout << "Payload Read" << endl;
// Put the payload data back into the envelope
string payload_data;
payload_msg->SerializeToString(&payload_data);
msgRefl->SetString(msg, dataField, payload_data);
// Now that required fields are populated, the dynamic message can be serialized and printed out.
string data;
msg->SerializeToString(&data);
// put data back into message to be replied
memcpy(recMsg.data(), data.c_str(), data.length());
// Useful examples of dynamic protobuf logic here : http://www.programmershare.com/2803644/
// (English not very good)
// 3.4 also describes dynamic compilation Uses google::protobuf::compiler Importer class
// Another link : dynamic stuff :
// http://stackoverflow.com/questions/11996557/how-to-dynamically-build-a-new-protobuf-from-a-set-of-already-defined-descriptor
// https://bitbucket.org/g19fanatic/prototest/src/dfd73a577dcc9bb51cbb3e99319cad352bfc60a8/src/main.cpp?at=master&fileviewer=file-view-default
}
inline std::string msg_to_string(zmq::message_t &message) {
return std::string(reinterpret_cast<char*>(message.data()), message.size());
}
inline T
get_bin(zmq::message_t& message)
{
return *(reinterpret_cast<T*>(message.data()));
}