tcp_iface.hh (11263:8dcc6b40f164) | tcp_iface.hh (11290:1640dd68b0a4) |
---|---|
1/* 2 * Copyright (c) 2015 ARM Limited 3 * All rights reserved 4 * 5 * The license below extends only to copyright in the software and shall 6 * not be construed as granting a license to any other intellectual 7 * property including but not limited to intellectual property relating 8 * to a hardware implementation of the functionality of the software --- 21 unchanged lines hidden (view full) --- 30 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 31 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 32 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 33 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 34 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 35 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 36 * 37 * Authors: Gabor Dozsa | 1/* 2 * Copyright (c) 2015 ARM Limited 3 * All rights reserved 4 * 5 * The license below extends only to copyright in the software and shall 6 * not be construed as granting a license to any other intellectual 7 * property including but not limited to intellectual property relating 8 * to a hardware implementation of the functionality of the software --- 21 unchanged lines hidden (view full) --- 30 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 31 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 32 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 33 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 34 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 35 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 36 * 37 * Authors: Gabor Dozsa |
38 * Mohammad Alian |
|
38 */ 39 40/* @file | 39 */ 40 41/* @file |
41 * TCP stream socket based interface class for multi gem5 runs. | 42 * TCP stream socket based interface class for dist-gem5 runs. |
42 * | 43 * |
43 * For a high level description about multi gem5 see comments in 44 * header file multi_iface.hh. | 44 * For a high level description about dist-gem5 see comments in 45 * header file dist_iface.hh. |
45 * | 46 * |
46 * The TCP subclass of MultiIface uses a separate server process 47 * (see tcp_server.[hh,cc] under directory gem5/util/multi). Each gem5 48 * process connects to the server via a stream socket. The server process | 47 * Each gem5 process connects to the server (another gem5 process which 48 * simulates a switch box) via a stream socket. The server process |
49 * transfers messages and co-ordinates the synchronisation among the gem5 50 * peers. 51 */ 52#ifndef __DEV_NET_TCP_IFACE_HH__ 53#define __DEV_NET_TCP_IFACE_HH__ 54 55 56#include <string> 57 | 49 * transfers messages and co-ordinates the synchronisation among the gem5 50 * peers. 51 */ 52#ifndef __DEV_NET_TCP_IFACE_HH__ 53#define __DEV_NET_TCP_IFACE_HH__ 54 55 56#include <string> 57 |
58#include "dev/net/multi_iface.hh" | 58#include "dev/net/dist_iface.hh" |
59 60class EventManager; 61 | 59 60class EventManager; 61 |
62class TCPIface : public MultiIface | 62class TCPIface : public DistIface |
63{ 64 private: 65 /** 66 * The stream socket to connect to the server. 67 */ 68 int sock; 69 | 63{ 64 private: 65 /** 66 * The stream socket to connect to the server. 67 */ 68 int sock; 69 |
70 std::string serverName; 71 int serverPort; 72 73 bool isSwitch; 74 75 bool listening; 76 static bool anyListening; 77 static int fdStatic; 78 |
|
70 /** | 79 /** |
71 * Registry for all sockets to the server opened by this gem5 process. | 80 * Compute node info and storage for the very first connection from each 81 * node (used by the switch) |
72 */ | 82 */ |
83 struct NodeInfo 84 { 85 unsigned rank; 86 unsigned distIfaceId; 87 unsigned distIfaceNum; 88 }; 89 static std::vector<std::pair<NodeInfo, int> > nodes; 90 /** 91 * Storage for all opened sockets 92 */ |
|
73 static std::vector<int> sockRegistry; 74 75 private: 76 77 /** 78 * Send out a message through a TCP stream socket. 79 * 80 * @param sock TCP stream socket. 81 * @param buf Start address of the message. 82 * @param length Size of the message in bytes. 83 */ 84 void | 93 static std::vector<int> sockRegistry; 94 95 private: 96 97 /** 98 * Send out a message through a TCP stream socket. 99 * 100 * @param sock TCP stream socket. 101 * @param buf Start address of the message. 102 * @param length Size of the message in bytes. 103 */ 104 void |
85 sendTCP(int sock, void *buf, unsigned length); | 105 sendTCP(int sock, const void *buf, unsigned length); |
86 87 /** 88 * Receive the next incoming message through a TCP stream socket. 89 * 90 * @param sock TCP stream socket. 91 * @param buf Start address of buffer to store the message. 92 * @param length Exact size of the expected message in bytes. 93 */ 94 bool recvTCP(int sock, void *buf, unsigned length); | 106 107 /** 108 * Receive the next incoming message through a TCP stream socket. 109 * 110 * @param sock TCP stream socket. 111 * @param buf Start address of buffer to store the message. 112 * @param length Exact size of the expected message in bytes. 113 */ 114 bool recvTCP(int sock, void *buf, unsigned length); |
115 bool listen(int port); 116 void accept(); 117 void connect(); 118 int getfdStatic() const { return fdStatic; } 119 bool islistening() const { return listening; } 120 bool anyislistening() const { return anyListening; } 121 void establishConnection(); |
|
95 | 122 |
96 | |
97 protected: 98 | 123 protected: 124 |
99 virtual void 100 sendRaw(void *buf, unsigned length, 101 const MultiHeaderPkt::AddressType dest_addr=nullptr) override 102 { 103 sendTCP(sock, buf, length); 104 } | 125 void sendPacket(const Header &header, 126 const EthPacketPtr &packet) override; |
105 | 127 |
106 virtual bool recvRaw(void *buf, unsigned length) override 107 { 108 return recvTCP(sock, buf, length); 109 } | 128 void sendCmd(const Header &header) override; |
110 | 129 |
111 virtual void syncRaw(MultiHeaderPkt::MsgType sync_req, 112 Tick sync_tick) override; | 130 bool recvHeader(Header &header) override; |
113 | 131 |
132 void recvPacket(const Header &header, EthPacketPtr &packet) override; 133 134 void initTransport() override; 135 |
|
114 public: 115 /** 116 * The ctor creates and connects the stream socket to the server. 117 * @param server_name The name (or IP address) of the host running the 118 * server process. 119 * @param server_port The port number the server listening for new 120 * connections. | 136 public: 137 /** 138 * The ctor creates and connects the stream socket to the server. 139 * @param server_name The name (or IP address) of the host running the 140 * server process. 141 * @param server_port The port number the server listening for new 142 * connections. |
121 * @param sync_start The tick for the first multi synchronisation. 122 * @param sync_repeat The frequency of multi synchronisation. | 143 * @param sync_start The tick for the first dist synchronisation. 144 * @param sync_repeat The frequency of dist synchronisation. |
123 * @param em The EventManager object associated with the simulated 124 * Ethernet link. 125 */ 126 TCPIface(std::string server_name, unsigned server_port, | 145 * @param em The EventManager object associated with the simulated 146 * Ethernet link. 147 */ 148 TCPIface(std::string server_name, unsigned server_port, |
127 unsigned multi_rank, Tick sync_start, Tick sync_repeat, 128 EventManager *em); | 149 unsigned dist_rank, unsigned dist_size, 150 Tick sync_start, Tick sync_repeat, EventManager *em, 151 bool is_switch, int num_nodes); |
129 130 ~TCPIface() override; 131}; 132 133#endif // __DEV_NET_TCP_IFACE_HH__ | 152 153 ~TCPIface() override; 154}; 155 156#endif // __DEV_NET_TCP_IFACE_HH__ |