Subversion Repositories an-fpga-implementation-of-low-latency-noc-based-mpsoc

/*
 * message.cpp
 *
 *  Created on: Dec 13, 2009
 *      Author: sam
 */
 
#include "messages.h"
#include <cassert>
#include <cstdlib>
 
using namespace std;
 
SocketStream& operator<<(SocketStream& os, StreamMessage& msg)
{
#ifdef NS_DEBUG_EXTRA
    std::cout << "<MessageSend> Sending message: " << msg.type << ", size: " << msg.size << std::endl;
#endif
 
    // cork the connection
    int flag = 1;
    setsockopt (os.so, SOL_TCP, TCP_CORK, &flag, sizeof (flag));
 
    os.put(&(msg.size), sizeof(int));
    os.put(&msg, msg.size);
 
    // uncork the connection
    flag = 0;
    setsockopt (os.so, SOL_TCP, TCP_CORK, &flag, sizeof (flag));
 
    // os.flush();
 
    return os;
}
 
SocketStream& operator>>(SocketStream& is, StreamMessage*& msg)
{
#ifdef NS_DEBUG_EXTRA
    std::cout << "<MessageRecv> Waiting for message" << std::endl;
#endif
 
    int msgSize = -1;
    int gotBytes = is.get(&msgSize, sizeof(int));
 
    if (gotBytes != sizeof(int))
        return is;
 
    assert(msgSize > 0);
 
    msg = (StreamMessage*) malloc(msgSize);
    is.get(msg, msgSize);
 
 
#ifdef NS_DEBUG_EXTRA
    std::cout << "<MessageRecv> Got message: " << msg->type << std::endl;
#endif
 
    return is;
}
 
SocketStream& operator>>(SocketStream& is, StreamMessage& msg)
{
#ifdef NS_DEBUG_EXTRA
    std::cout << "<MessageRecvSync> Waiting for message" << std::endl;
#endif
 
    int msgSize = -1;
    int gotBytes = is.get(&msgSize, sizeof(int));
 
    if (gotBytes != sizeof(int))
        return is;
 
    std::cout << "Debug: " << msgSize << msg.size << " " << msg.id  << std::endl;
    assert(msgSize == msg.size);
    is.get(&msg, msgSize);
 
#ifdef NS_DEBUG_EXTRA
    std::cout << "<MessageRecvSync> Got message: " << msg.type << std::endl;
#endif
 
    return is;
}
 
void StreamMessage::destroy(StreamMessage* msg)
{
    assert (msg != NULL);
    free(msg);
}