00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024
00025
00026
00027
00028
00029
00030
00031
00032
00033
00034
00035
00036
00037
00038
00039 #ifndef ARIBA_OMNET_MODULE_H__
00040 #define ARIBA_OMNET_MODULE_H__
00041
00042 #include "ariba/utility/types.h"
00043 #include "ariba/utility/messages.h"
00044 #include "ariba/utility/types/Address.h"
00045 #include "ariba/utility/misc/Helper.h"
00046 #include "ariba/utility/logging/Logging.h"
00047 #include "ariba/utility/serialization/DataStream.hpp"
00048 #include "ariba/communication/modules/transport/TransportProtocol.h"
00049 #include "ariba/communication/modules/transport/TransportLocator.h"
00050 #include "ariba/communication/modules/network/ip/IPv4Locator.h"
00051 #include "AribaOmnetMessage_m.h"
00052 #include <omnetpp.h>
00053 #include <INETDefs.h>
00054 #include <TCPSocket.h>
00055 #include <map>
00056 #include <iostream>
00057 #include <sstream>
00058 #include <vector>
00059
00060 using std::vector;
00061 using std::ostringstream;
00062 using std::cout;
00063 using std::pair;
00064 using std::make_pair;
00065 using std::map;
00066 using std::multimap;
00067 using ariba::utility::seqnum_t;
00068 using ariba::utility::Message;
00069 using ariba::utility::Address;
00070 using ariba::utility::serialization::data_serialize;
00071 using ariba::communication::IPv4Locator;
00072
00073 namespace ariba {
00074 namespace communication {
00075
00076 class INET_API AribaOmnetModule : public cSimpleModule, public TransportProtocol, TCPSocket::CallbackInterface {
00077
00078 use_logging_h( AribaOmnetModule );
00079
00080 public:
00081 AribaOmnetModule();
00082 virtual ~AribaOmnetModule();
00083
00084 void setServerPort(uint16_t _port);
00085
00089 virtual void start();
00090 virtual void stop();
00091 virtual TransportLocator::prot_t getId();
00092 virtual const vector<TransportLocator*> getLocators();
00093 virtual seqnum_t sendMessage( const Message* message );
00094
00098 virtual void socketDataArrived( int connId, void* socket, cMessage* msg, bool urgent );
00099 virtual void socketFailure( int connId, void* socket, int code );
00100 virtual void socketClosed( int connId, void* socket );
00101 virtual void socketPeerClosed( int connId, void* socket );
00102 virtual void socketEstablished( int connId, void* socket );
00103 virtual void socketStatusArrived( int connId, void* socket, TCPStatusInfo *status);
00104
00105 protected:
00109 virtual void initialize(int stage);
00110 virtual int numInitStages () const ;
00111
00116 virtual void handleMessage( cMessage* msg );
00117
00122 virtual void finish();
00123
00124 private:
00128 TCPSocket serverSocket;
00129 typedef multimap<string, TCPSocket*> SocketMap;
00130 SocketMap sockets;
00131 uint16_t serverPort;
00132
00137 typedef map<TCPSocket*, AribaOmnetMessage*> PendingSendQueue;
00138 PendingSendQueue pendingSends;
00139
00140 };
00141
00142
00143
00144
00145
00146 Define_Module( AribaOmnetModule );
00147
00148 }}
00149
00150 #endif // ARIBA_OMNET_MODULE_H__