Jeff Thompson | 25b4e61 | 2013-10-10 16:03:24 -0700 | [diff] [blame] | 1 | /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil -*- */ |
Jeff Thompson | fcf347d | 2013-07-15 11:30:44 -0700 | [diff] [blame] | 2 | /** |
Jeff Thompson | 7687dc0 | 2013-09-13 11:54:07 -0700 | [diff] [blame] | 3 | * Copyright (C) 2013 Regents of the University of California. |
| 4 | * @author: Jeff Thompson <jefft0@remap.ucla.edu> |
Jeff Thompson | fcf347d | 2013-07-15 11:30:44 -0700 | [diff] [blame] | 5 | * See COPYING for copyright and distribution information. |
| 6 | */ |
| 7 | |
| 8 | #ifndef NDN_TCPTRANSPORT_HPP |
Jeff Thompson | 2d27e2f | 2013-08-09 12:55:00 -0700 | [diff] [blame] | 9 | #define NDN_TCPTRANSPORT_HPP |
Jeff Thompson | fcf347d | 2013-07-15 11:30:44 -0700 | [diff] [blame] | 10 | |
Jeff Thompson | 10e3438 | 2013-08-22 13:34:46 -0700 | [diff] [blame] | 11 | #include <string> |
Jeff Thompson | 25b4e61 | 2013-10-10 16:03:24 -0700 | [diff] [blame] | 12 | #include "../common.hpp" |
Jeff Thompson | 5341219 | 2013-08-06 13:35:50 -0700 | [diff] [blame] | 13 | #include "transport.hpp" |
Jeff Thompson | fcf347d | 2013-07-15 11:30:44 -0700 | [diff] [blame] | 14 | |
Jeff Thompson | 25b4e61 | 2013-10-10 16:03:24 -0700 | [diff] [blame] | 15 | struct ndn_TcpTransport; |
| 16 | struct ndn_BinaryXmlElementReader; |
| 17 | |
Jeff Thompson | fcf347d | 2013-07-15 11:30:44 -0700 | [diff] [blame] | 18 | namespace ndn { |
| 19 | |
| 20 | class TcpTransport : public Transport { |
| 21 | public: |
Jeff Thompson | 10e3438 | 2013-08-22 13:34:46 -0700 | [diff] [blame] | 22 | /** |
Jeff Thompson | c967108 | 2013-08-22 13:54:05 -0700 | [diff] [blame] | 23 | * A TcpTransport::ConnectionInfo extends Transport::ConnectionInfo to hold the host and port info for the TCP connection. |
Jeff Thompson | 10e3438 | 2013-08-22 13:34:46 -0700 | [diff] [blame] | 24 | */ |
| 25 | class ConnectionInfo : public Transport::ConnectionInfo { |
| 26 | public: |
| 27 | /** |
| 28 | * Create a ConnectionInfo with the given host and port. |
| 29 | * @param host The host for the connection. |
Jeff Thompson | 39d241a | 2013-10-07 16:50:15 -0700 | [diff] [blame] | 30 | * @param port The port number for the connection. If omitted, use 6363. |
Jeff Thompson | 10e3438 | 2013-08-22 13:34:46 -0700 | [diff] [blame] | 31 | */ |
Jeff Thompson | 39d241a | 2013-10-07 16:50:15 -0700 | [diff] [blame] | 32 | ConnectionInfo(const char *host, unsigned short port = 6363) |
Jeff Thompson | 10e3438 | 2013-08-22 13:34:46 -0700 | [diff] [blame] | 33 | : host_(host), port_(port) |
| 34 | { |
| 35 | } |
Jeff Thompson | b2c01a3 | 2013-08-23 19:42:18 -0700 | [diff] [blame] | 36 | |
| 37 | /** |
Jeff Thompson | 10e3438 | 2013-08-22 13:34:46 -0700 | [diff] [blame] | 38 | * Get the host given to the constructor. |
| 39 | * @return A string reference for the host. |
| 40 | */ |
Jeff Thompson | 0050abe | 2013-09-17 12:50:25 -0700 | [diff] [blame] | 41 | const std::string& |
| 42 | getHost() const { return host_; } |
Jeff Thompson | 10e3438 | 2013-08-22 13:34:46 -0700 | [diff] [blame] | 43 | |
| 44 | /** |
| 45 | * Get the port given to the constructor. |
| 46 | * @return The port number. |
| 47 | */ |
Jeff Thompson | 0050abe | 2013-09-17 12:50:25 -0700 | [diff] [blame] | 48 | unsigned short |
| 49 | getPort() const { return port_; } |
Jeff Thompson | 10e3438 | 2013-08-22 13:34:46 -0700 | [diff] [blame] | 50 | |
Jeff Thompson | 0050abe | 2013-09-17 12:50:25 -0700 | [diff] [blame] | 51 | virtual |
| 52 | ~ConnectionInfo(); |
Jeff Thompson | 10e3438 | 2013-08-22 13:34:46 -0700 | [diff] [blame] | 53 | |
| 54 | private: |
| 55 | std::string host_; |
| 56 | unsigned short port_; |
| 57 | }; |
| 58 | |
Jeff Thompson | 25b4e61 | 2013-10-10 16:03:24 -0700 | [diff] [blame] | 59 | TcpTransport(); |
Jeff Thompson | fcf347d | 2013-07-15 11:30:44 -0700 | [diff] [blame] | 60 | |
Jeff Thompson | 2ed62fb | 2013-07-16 18:10:30 -0700 | [diff] [blame] | 61 | /** |
Jeff Thompson | 10e3438 | 2013-08-22 13:34:46 -0700 | [diff] [blame] | 62 | * Connect according to the info in ConnectionInfo, and processEvents() will use elementListener. |
| 63 | * @param connectionInfo A reference to a TcpTransport::ConnectionInfo. |
| 64 | * @param elementListener Not a shared_ptr because we assume that it will remain valid during the life of this object. |
Jeff Thompson | 2ed62fb | 2013-07-16 18:10:30 -0700 | [diff] [blame] | 65 | */ |
Jeff Thompson | 1656e6a | 2013-08-29 18:01:48 -0700 | [diff] [blame] | 66 | virtual void connect(const Transport::ConnectionInfo& connectionInfo, ElementListener& elementListener); |
Jeff Thompson | fcf347d | 2013-07-15 11:30:44 -0700 | [diff] [blame] | 67 | |
Jeff Thompson | 432c8be | 2013-08-09 16:16:08 -0700 | [diff] [blame] | 68 | /** |
| 69 | * Set data to the host |
| 70 | * @param data A pointer to the buffer of data to send. |
| 71 | * @param dataLength The number of bytes in data. |
| 72 | */ |
Jeff Thompson | 97223af | 2013-09-24 17:01:27 -0700 | [diff] [blame] | 73 | virtual void send(const uint8_t *data, size_t dataLength); |
Jeff Thompson | fcf347d | 2013-07-15 11:30:44 -0700 | [diff] [blame] | 74 | |
Jeff Thompson | 432c8be | 2013-08-09 16:16:08 -0700 | [diff] [blame] | 75 | /** |
Jeff Thompson | 10e3438 | 2013-08-22 13:34:46 -0700 | [diff] [blame] | 76 | * Process any data to receive. For each element received, call elementListener.onReceivedElement. |
Jeff Thompson | c7e0744 | 2013-08-19 15:25:43 -0700 | [diff] [blame] | 77 | * This is non-blocking and will return immediately if there is no data to receive. |
| 78 | * You should normally not call this directly since it is called by Face.processEvents. |
Jeff Thompson | 432c8be | 2013-08-09 16:16:08 -0700 | [diff] [blame] | 79 | * @throw This may throw an exception for reading data or in the callback for processing the data. If you |
| 80 | * call this from an main event loop, you may want to catch and log/disregard all exceptions. |
| 81 | */ |
| 82 | virtual void processEvents(); |
Jeff Thompson | 5796388 | 2013-08-05 16:01:25 -0700 | [diff] [blame] | 83 | |
Jeff Thompson | a405697 | 2013-08-22 11:52:21 -0700 | [diff] [blame] | 84 | virtual bool getIsConnected(); |
| 85 | |
Jeff Thompson | 432c8be | 2013-08-09 16:16:08 -0700 | [diff] [blame] | 86 | /** |
| 87 | * Close the connection to the host. |
| 88 | */ |
Jeff Thompson | 5796388 | 2013-08-05 16:01:25 -0700 | [diff] [blame] | 89 | virtual void close(); |
Jeff Thompson | fcf347d | 2013-07-15 11:30:44 -0700 | [diff] [blame] | 90 | |
Jeff Thompson | a00f4eb | 2013-08-12 12:36:48 -0700 | [diff] [blame] | 91 | ~TcpTransport(); |
| 92 | |
Jeff Thompson | fcf347d | 2013-07-15 11:30:44 -0700 | [diff] [blame] | 93 | private: |
Jeff Thompson | 25b4e61 | 2013-10-10 16:03:24 -0700 | [diff] [blame] | 94 | ptr_lib::shared_ptr<struct ndn_TcpTransport> transport_; |
Jeff Thompson | a405697 | 2013-08-22 11:52:21 -0700 | [diff] [blame] | 95 | bool isConnected_; |
Jeff Thompson | b002f90 | 2013-07-16 18:07:18 -0700 | [diff] [blame] | 96 | // TODO: This belongs in the socket listener. |
Jeff Thompson | 25b4e61 | 2013-10-10 16:03:24 -0700 | [diff] [blame] | 97 | ptr_lib::shared_ptr<struct ndn_BinaryXmlElementReader> elementReader_; |
Jeff Thompson | fcf347d | 2013-07-15 11:30:44 -0700 | [diff] [blame] | 98 | }; |
| 99 | |
| 100 | } |
| 101 | |
| 102 | #endif |