networkmessage_rw.cc

分布式仿真 开放源码

NetworkMessage::readFederateName(MessageBody *Body)
{
    Body->readString(federateName, MAX_FEDERATE_NAME_LENGTH);
}


// -----------
// -- Write --
// -----------

void
NetworkMessage::write(Socket *socket)
    throw (NetworkError,
           NetworkSignal)
{

#ifdef USE_HEADER_AND_BODY
    // 1- Call WriteHeader
    bool result = writeHeader(socket);

    // 2- If WriteHeader returned RTI_TRUE, call WriteBody.
    if (result)
        writeBody(socket);
#else
    socket->send(this, sizeof(NetworkMessage));
#endif

}


// ---------------
// -- WriteBody --
// ---------------

void
NetworkMessage::writeBody(Socket *socket)
{
    MessageBody Body ;
    unsigned short i ;

    // 0- Copy the Header at the beginning of the Body, in order to
    // make a single Socket->Emettre call while sending both.
    // WARNING: As the Body size is not known yet, we will have to
    // change it in the copy also!
    Body.writeBlock((char *) &Header, sizeof(HeaderStruct));

    // 1- Prepare Body Structure according to Message type
    if ((Header.type == UPDATE_ATTRIBUTE_VALUES)
        || (Header.type == REFLECT_ATTRIBUTE_VALUES)) {
        Body.writeLongInt(object);
        Body.writeString(label);
        Body.writeBlock((char *) handleArray,
                        handleArraySize * sizeof(AttributeHandle));

        for (i = 0 ; i < handleArraySize ; i ++) {
            Body.writeString(ValueArray[i]);
        }
    }
    else
        switch(Header.type) {

            // -- O_I Variable Part With date(Body Not Empty) --

          case SEND_INTERACTION:
          case RECEIVE_INTERACTION:
            Body.writeString(label);
            Body.writeBlock((char *) handleArray,
                            handleArraySize * sizeof(AttributeHandle));
            for (i = 0 ; i < handleArraySize ; i ++) {
                Body.writeString(ValueArray[i]);
            }
            break ;

            // -- No Variable Part --

          case CREATE_FEDERATION_EXECUTION:
          case DESTROY_FEDERATION_EXECUTION:
            Body.writeString(federationName);
            break ;

          case REGISTER_FEDERATION_SYNCHRONIZATION_POINT:
          case ANNOUNCE_SYNCHRONIZATION_POINT:
            Body.writeString(label);
            Body.writeString(tag);
            break ;

          case SYNCHRONIZATION_POINT_ACHIEVED:
          case SYNCHRONIZATION_POINT_REGISTRATION_SUCCEEDED:
          case FEDERATION_SYNCHRONIZED:
          case REQUEST_FEDERATION_SAVE:
          case INITIATE_FEDERATE_SAVE:
          case REQUEST_FEDERATION_RESTORE:
          case REQUEST_FEDERATION_RESTORE_SUCCEEDED:
          case INITIATE_FEDERATE_RESTORE:
            Body.writeString(label);
            break ;

          case REQUEST_FEDERATION_RESTORE_FAILED:
            Body.writeString(label);
            Body.writeString(tag);
            break ;

          case DELETE_OBJECT:
          case REMOVE_OBJECT:
            Body.writeLongInt(object);
            Body.writeString(label);
            break ;

          case IS_ATTRIBUTE_OWNED_BY_FEDERATE:
          case INFORM_ATTRIBUTE_OWNERSHIP:
          case ATTRIBUTE_IS_NOT_OWNED:
          case QUERY_ATTRIBUTE_OWNERSHIP:
            Body.writeLongInt(object);
            Body.writeShortInt(handleArray[0]);
            Body.writeString(label);
            break ;

          case NEGOTIATED_ATTRIBUTE_OWNERSHIP_DIVESTITURE:
          case REQUEST_ATTRIBUTE_OWNERSHIP_ASSUMPTION:
          case ATTRIBUTE_OWNERSHIP_ACQUISITION:
          case REQUEST_ATTRIBUTE_OWNERSHIP_RELEASE:
            Body.writeLongInt(object);
            Body.writeShortInt(handleArraySize);
            for (i = 0 ; i < handleArraySize ; i ++)
                Body.writeShortInt(handleArray[i]);
            Body.writeString(label);
            break ;


          case ATTRIBUTE_OWNERSHIP_ACQUISITION_IF_AVAILABLE:
          case ATTRIBUTE_OWNERSHIP_ACQUISITION_NOTIFICATION:
          case ATTRIBUTE_OWNERSHIP_UNAVAILABLE:
          case UNCONDITIONAL_ATTRIBUTE_OWNERSHIP_DIVESTITURE:
          case ATTRIBUTE_OWNERSHIP_DIVESTITURE_NOTIFICATION:
          case CANCEL_NEGOTIATED_ATTRIBUTE_OWNERSHIP_DIVESTITURE:
          case ATTRIBUTE_OWNERSHIP_RELEASE_RESPONSE:
          case CANCEL_ATTRIBUTE_OWNERSHIP_ACQUISITION:
          case CONFIRM_ATTRIBUTE_OWNERSHIP_ACQUISITION_CANCELLATION:
            Body.writeLongInt(object);
            Body.writeShortInt(handleArraySize);
            for (i = 0 ; i < handleArraySize ; i ++)
                Body.writeShortInt(handleArray[i]);
            break ;

          case DDM_CREATE_REGION:
            Body.writeLongInt(space);
            Body.writeLongInt(nbExtents);
            Body.writeLongInt(region);
            break ;

            // -- Join Variable Part --

          case JOIN_FEDERATION_EXECUTION:
            Body.writeString(federationName);
            Body.writeString(federateName);
            break ;

            // -- O_I Variable Part(Body not empty) --

          case PUBLISH_OBJECT_CLASS:
          case SUBSCRIBE_OBJECT_CLASS:
            for (i = 0 ; i < handleArraySize ; i ++)
                Body.writeShortInt(handleArray[i]);
            break ;

          case REGISTER_OBJECT:
          case DISCOVER_OBJECT:
            Body.writeLongInt(object);
            Body.writeString(label);
            break ;

	  case DDM_MODIFY_REGION:
	    writeExtents(Body);
	    break ;

	  case DDM_ASSOCIATE_REGION:
	    Body.writeLongInt(object);
	    Body.writeLongInt(region);
	    Body.writeShortInt(boolean);
            Body.writeShortInt(handleArraySize);
	    for (i = 0 ; i < handleArraySize ; i ++)
                Body.writeShortInt(handleArray[i]);
	    break ;

	  case DDM_SUBSCRIBE_ATTRIBUTES:
	    Body.writeLongInt(objectClass);
	    Body.writeLongInt(region);
	    Body.writeShortInt(boolean);
            Body.writeShortInt(handleArraySize);
	    for (i = 0 ; i < handleArraySize ; i ++)
                Body.writeShortInt(handleArray[i]);
	    break ;

	  case DDM_UNASSOCIATE_REGION:
	    Body.writeLongInt(object);
	    Body.writeLongInt(region);
	    break ;

	  case DDM_UNSUBSCRIBE_ATTRIBUTES:
	    Body.writeLongInt(objectClass);
	    Body.writeLongInt(region);
	    break ;
	    
	  case DDM_SUBSCRIBE_INTERACTION:
	  case DDM_UNSUBSCRIBE_INTERACTION:
	    Body.writeLongInt(interactionClass);
	    Body.writeLongInt(region);
	    Body.writeShortInt(boolean);
	    break ;
	    
            // -- Default Handler --
          default:
            D.Out(pdExcept, "Unknown type %d in WriteBody.", Header.type);
            throw RTIinternalError("Unknown/Unimplemented type for Header.");
        }

    // 2- Set Header.bodySize

    // Body Size does not include the copy of the Header!
    Header.bodySize = Body.getLength() - sizeof(HeaderStruct);


    // Put the real Body Size in the copy of the Header.
    ((HeaderStruct *) Body.getBuffer())->bodySize = Header.bodySize ;

    // 3- Write Header to socket, then write Body to socket.
    // socket->send((void *) &Header, sizeof(HeaderStruct));
    socket->send((void *) Body.getBuffer(), Body.getLength());
}


// -----------------
// -- WriteHeader --
// -----------------

bool
NetworkMessage::writeHeader(Socket *socket)
{
    // 1- Clear Header
    // memset((void *) &Header, '\0', sizeof(HeaderStruct));

    // 2- Fill Header(Static Part)
    Header.type = type ;
    Header.exception = exception ;
    Header.federate = federate ;
    Header.federation = federation ;

    // 3- Fill Header(Variable Part)[Sorted by Variable part type]

    // Note: Header.bodySize is not set to the actual Body size, but
    // to zero to indicate there is no Body, or 1 if a Body is needed.

    if (type == MESSAGE_NULL) {
        Header.bodySize = 0 ;
        Header.VP.time.date = date ;
    }
    else if ((type == UPDATE_ATTRIBUTE_VALUES)
             || (type == REFLECT_ATTRIBUTE_VALUES)) {
        // Body contains ObjectHandle, handleArray, ValueArray, label
        Header.bodySize = 1 ;
        Header.VP.O_I.handle = objectClass ;
        Header.VP.O_I.size = handleArraySize ;
        Header.VP.O_I.date = date ;
    }
    else
        switch(type) {

          case SEND_INTERACTION:
          case RECEIVE_INTERACTION:
            // Body contains handleArray, ValueArray, label.
            Header.bodySize = 1 ;
            Header.VP.O_I.handle = interactionClass ;
            Header.VP.O_I.size = handleArraySize ;
            Header.VP.O_I.date = date ;
            break ;

          case REQUEST_FEDERATION_SAVE:
            Header.bodySize = 1 ;
            Header.VP.O_I.date = date ;
            break ;

          case INITIATE_FEDERATE_SAVE:
          case REQUEST_FEDERATION_RESTORE:
          case REQUEST_FEDERATION_RESTORE_SUCCEEDED:
          case REQUEST_FEDERATION_RESTORE_FAILED:
            Header.bodySize = 1 ;
            break ;

            // -- No Variable Part, No Body --

          case CLOSE_CONNEXION:
          case RESIGN_FEDERATION_EXECUTION:
          case FEDERATE_SAVE_BEGUN:
          case FEDERATE_SAVE_COMPLETE:
          case FEDERATE_SAVE_NOT_COMPLETE:
          case FEDERATION_SAVED:
          case FEDERATION_NOT_SAVED:
          case FEDERATE_RESTORE_COMPLETE:
          case FEDERATE_RESTORE_NOT_COMPLETE:
          case FEDERATION_RESTORE_BEGUN:
          case FEDERATION_RESTORED:
          case FEDERATION_NOT_RESTORED:
            Header.bodySize = 0 ;
            break ;

            // -- No Variable Part, Body not empty --

          case CREATE_FEDERATION_EXECUTION:
          case DESTROY_FEDERATION_EXECUTION:
            // Body Contains federationName.
          case INFORM_ATTRIBUTE_OWNERSHIP:
          case ATTRIBUTE_IS_NOT_OWNED:
          case IS_ATTRIBUTE_OWNED_BY_FEDERATE:
          case QUERY_ATTRIBUTE_OWNERSHIP:
            // Body Contains ObjectHandle and label
          case ATTRIBUTE_OWNERSHIP_ACQUISITION_IF_AVAILABLE:
          case ATTRIBUTE_OWNERSHIP_ACQUISITION_NOTIFICATION:
          case ATTRIBUTE_OWNERSHIP_UNAVAILABLE:
            // Body Contains ObjectHandle and handleArray
          case NEGOTIATED_ATTRIBUTE_OWNERSHIP_DIVESTITURE:
          case REQUEST_ATTRIBUTE_OWNERSHIP_ASSUMPTION:
          case UNCONDITIONAL_ATTRIBUTE_OWNERSHIP_DIVESTITURE:
          case ATTRIBUTE_OWNERSHIP_ACQUISITION:
          case REQUEST_ATTRIBUTE_OWNERSHIP_RELEASE:
          case ATTRIBUTE_OWNERSHIP_DIVESTITURE_NOTIFICATION:
          case CANCEL_NEGOTIATED_ATTRIBUTE_OWNERSHIP_DIVESTITURE:
          case ATTRIBUTE_OWNERSHIP_RELEASE_RESPONSE:
          case CANCEL_ATTRIBUTE_OWNERSHIP_ACQUISITION:
          case CONFIRM_ATTRIBUTE_OWNERSHIP_ACQUISITION_CANCELLATION:
          case DDM_CREATE_REGION:
          case INITIATE_FEDERATE_RESTORE:
	  case DDM_ASSOCIATE_REGION:
	  case DDM_UNASSOCIATE_REGION:
	  case DDM_SUBSCRIBE_ATTRIBUTES:
	  case DDM_UNSUBSCRIBE_ATTRIBUTES:
	  case DDM_SUBSCRIBE_INTERACTION:
	  case DDM_UNSUBSCRIBE_INTERACTION:	
            Header.bodySize = 1 ;
            break ;

          case REGISTER_FEDERATION_SYNCHRONIZATION_POINT:
          case SYNCHRONIZATION_POINT_ACHIEVED:
          case SYNCHRONIZATION_POINT_REGISTRATION_SUCCEEDED:
          case FEDERATION_SYNCHRONIZED:
          case ANNOUNCE_SYNCHRONIZATION_POINT:
            // Body Contains Label(should be non-empty)
            // BUG: S'il fait moins de 16 octet, il passe dans le header.
            Header.bodySize = 1 ;
            break ;

          case DELETE_OBJECT:
          case REMOVE_OBJECT:
            // Body Contains ObjectHandle, and label
            Header.bodySize = 1 ;
            break ;

            // -- time Variable Part(No Body)[Continued] --

          case SET_TIME_REGULATING:
            Header.bodySize = 0 ;
            Header.VP.time.date = date ;
            Header.VP.time.R_or_C = regulator ;
            break ;

          case SET_TIME_CONSTRAINED:
            Header.bodySize = 0 ;
            Header.VP.time.date = date ;
            Header.VP.time.R_or_C = constrained ;
            break ;

            // -- T_O Variable Part --

          case CHANGE_ATTRIBUTE_TRANSPORT_TYPE:
          case CHANGE_ATTRIBUTE_ORDER_TYPE:
          case CHANGE_INTERACTION_TRANSPORT_TYPE:
          case CHANGE_INTERACTION_ORDER_TYPE:
            throw RTIinternalError("Write Message not implemented for T/O.");
            break ;

            // -- Join Variable Part --

          case JOIN_FEDERATION_EXECUTION:
            // Body contains federationName and federateName
            Header.bodySize = 1 ;
            Header.VP.Join.NbReg = numberOfRegulators ;
            Header.VP.Join.AdrMC = multicastAddress ;
            Header.VP.Join.Addr = bestEffortAddress ;
            Header.VP.Join.peer = bestEffortPeer ;
            break ;

            // -- O_I Variable Part(No Body) --

          case UNPUBLISH_OBJECT_CLASS:
          case UNSUBSCRIBE_OBJECT_CLASS:
            Header.bodySize = 0 ;
            Header.VP.O_I.handle = objectClass ;
            break ;

          case PUBLISH_INTERACTION_CLASS:
          case UNPUBLISH_INTERACTION_CLASS:
          case SUBSCRIBE_INTERACTION_CLASS:
          case UNSUBSCRIBE_INTERACTION_CLASS:
          case TURN_INTERACTIONS_ON:
          case TURN_INTERACTIONS_OFF:
            Header.bodySize = 0 ;
            Header.VP.O_I.handle = interactionClass ;
            break ;

            // DDM variable part, no body
          case DDM_DELETE_REGION:
            Header.bodySize = 0 ;
            Header.VP.ddm.region = region ;
            break ;

            // -- O_I Variable Part(Body not empty) --

          case PUBLISH_OBJECT_CLASS:
          case SUBSCRIBE_OBJECT_CLASS:
            // Body contains handleArray[handleArraySize](if not empty)

            if (handleArraySize > 0)
                Header.bodySize = 1 ;
            else
                Header.bodySize = 0 ;

            Header.VP.O_I.handle = objectClass ;
            Header.VP.O_I.size = handleArraySize ;
            break ;

          case REGISTER_OBJECT:
          case DISCOVER_OBJECT:
            // Body Contains ObjectHandle and label
            Header.bodySize = 1 ;
            Header.VP.O_I.handle = objectClass ;
            break ;

	  case DDM_MODIFY_REGION:
	    Header.bodySize = 1 ;
	    Header.VP.ddm.region = region ;
	    break ;

            // -- Default Handler --

          default:
            D.Out(pdExcept, "Unknown type %d in WriteHeader.", Header.type);
            throw RTIinternalError("Unknown/Unimplemented type for Header.");
        }

    // 4- If Header.bodySize = 0, send message and return RTI_FALSE,
    // Else send nothing(will be done by WriteBody), and return RTI_TRUE.

    if (Header.bodySize == 0)
        socket->send((void *) &Header, sizeof(HeaderStruct));

    return Header.bodySize != 0 ;
}

} // namespace certi

// $Id: NetworkMessage_RW.cc,v 3.15 2004/01/09 16:13:49 breholee Exp $