blob: cb1c6e3877a11e554b1b8a660fc586fdc0d295bc [file] [log] [blame]
Jeff Thompson25b4e612013-10-10 16:03:24 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil -*- */
Jeff Thompsonfcf347d2013-07-15 11:30:44 -07002/**
Jeff Thompson7687dc02013-09-13 11:54:07 -07003 * Copyright (C) 2013 Regents of the University of California.
4 * @author: Jeff Thompson <jefft0@remap.ucla.edu>
Jeff Thompsonfcf347d2013-07-15 11:30:44 -07005 * See COPYING for copyright and distribution information.
6 */
7
8#include <stdexcept>
Jeff Thompson53430e02013-10-23 10:12:41 -07009#include <stdlib.h>
Jeff Thompson25b4e612013-10-10 16:03:24 -070010#include <ndn-cpp/node.hpp>
11#include "../c/transport/tcp-transport.h"
12#include "../c/encoding/binary-xml-element-reader.h"
Jeff Thompson5e275b42013-07-16 19:10:11 -070013#include "../c/util/ndn_realloc.h"
Jeff Thompson25b4e612013-10-10 16:03:24 -070014#include <ndn-cpp/transport/tcp-transport.hpp>
Jeff Thompsonfcf347d2013-07-15 11:30:44 -070015
16using namespace std;
17
18namespace ndn {
19
Alexander Afanasyev0b688dc2013-12-18 16:43:37 -080020TcpTransport::TcpTransport(const char *host, unsigned short port/* = 6363*/)
21 : host_(host), port_(port)
22 , isConnected_(false), transport_(new struct ndn_TcpTransport), elementReader_(new struct ndn_BinaryXmlElementReader)
Jeff Thompson25b4e612013-10-10 16:03:24 -070023{
24 ndn_TcpTransport_initialize(transport_.get());
25 elementReader_->partialData.array = 0;
26}
27
Jeff Thompson0050abe2013-09-17 12:50:25 -070028void
Alexander Afanasyev0b688dc2013-12-18 16:43:37 -080029TcpTransport::connect(ElementListener& elementListener)
Jeff Thompsonfcf347d2013-07-15 11:30:44 -070030{
31 ndn_Error error;
Alexander Afanasyev0b688dc2013-12-18 16:43:37 -080032 if ((error = ndn_TcpTransport_connect(transport_.get(), (char *)host_.c_str(), port_)))
Jeff Thompson4affbf52013-10-18 14:36:46 -070033 throw runtime_error(ndn_getErrorString(error));
Jeff Thompsonb002f902013-07-16 18:07:18 -070034
35 // TODO: This belongs in the socket listener.
Jeff Thompson97223af2013-09-24 17:01:27 -070036 const size_t initialLength = 1000;
Jeff Thompson10e34382013-08-22 13:34:46 -070037 // Automatically cast elementReader_ to (struct ndn_ElementListener *)
Jeff Thompsond1427fb2013-08-29 17:20:32 -070038 ndn_BinaryXmlElementReader_initialize
Jeff Thompson25b4e612013-10-10 16:03:24 -070039 (elementReader_.get(), &elementListener, (uint8_t *)malloc(initialLength), initialLength, ndn_realloc);
Jeff Thompson0cb7aee2013-07-16 16:18:06 -070040
Jeff Thompsona4056972013-08-22 11:52:21 -070041 isConnected_ = true;
Jeff Thompsonfcf347d2013-07-15 11:30:44 -070042}
43
Jeff Thompson0050abe2013-09-17 12:50:25 -070044void
Jeff Thompson97223af2013-09-24 17:01:27 -070045TcpTransport::send(const uint8_t *data, size_t dataLength)
Jeff Thompsonfcf347d2013-07-15 11:30:44 -070046{
47 ndn_Error error;
Jeff Thompson25b4e612013-10-10 16:03:24 -070048 if ((error = ndn_TcpTransport_send(transport_.get(), (uint8_t *)data, dataLength)))
Jeff Thompson4affbf52013-10-18 14:36:46 -070049 throw runtime_error(ndn_getErrorString(error));
Jeff Thompsonfcf347d2013-07-15 11:30:44 -070050}
51
Jeff Thompson0050abe2013-09-17 12:50:25 -070052void
53TcpTransport::processEvents()
Jeff Thompsonfcf347d2013-07-15 11:30:44 -070054{
Jeff Thompson432c8be2013-08-09 16:16:08 -070055 int receiveIsReady;
56 ndn_Error error;
Jeff Thompson25b4e612013-10-10 16:03:24 -070057 if ((error = ndn_TcpTransport_receiveIsReady(transport_.get(), &receiveIsReady)))
Jeff Thompson4affbf52013-10-18 14:36:46 -070058 throw runtime_error(ndn_getErrorString(error));
Jeff Thompson432c8be2013-08-09 16:16:08 -070059 if (!receiveIsReady)
60 return;
Jeff Thompson7ed97c72013-07-16 17:56:41 -070061
Jeff Thompson10ad12a2013-09-24 16:19:11 -070062 uint8_t buffer[8000];
Jeff Thompson97223af2013-09-24 17:01:27 -070063 size_t nBytes;
Jeff Thompson25b4e612013-10-10 16:03:24 -070064 if ((error = ndn_TcpTransport_receive(transport_.get(), buffer, sizeof(buffer), &nBytes)))
Jeff Thompson4affbf52013-10-18 14:36:46 -070065 throw runtime_error(ndn_getErrorString(error));
Jeff Thompson432c8be2013-08-09 16:16:08 -070066
Jeff Thompson25b4e612013-10-10 16:03:24 -070067 ndn_BinaryXmlElementReader_onReceivedData(elementReader_.get(), buffer, nBytes);
Jeff Thompsonfcf347d2013-07-15 11:30:44 -070068}
69
Jeff Thompson0050abe2013-09-17 12:50:25 -070070bool
71TcpTransport::getIsConnected()
Jeff Thompsona4056972013-08-22 11:52:21 -070072{
73 return isConnected_;
74}
75
Jeff Thompson0050abe2013-09-17 12:50:25 -070076void
77TcpTransport::close()
Jeff Thompson57963882013-08-05 16:01:25 -070078{
79 ndn_Error error;
Jeff Thompson25b4e612013-10-10 16:03:24 -070080 if ((error = ndn_TcpTransport_close(transport_.get())))
Jeff Thompson4affbf52013-10-18 14:36:46 -070081 throw runtime_error(ndn_getErrorString(error));
Jeff Thompson57963882013-08-05 16:01:25 -070082}
83
Jeff Thompsona00f4eb2013-08-12 12:36:48 -070084TcpTransport::~TcpTransport()
85{
Jeff Thompson25b4e612013-10-10 16:03:24 -070086 if (elementReader_->partialData.array)
Jeff Thompsona00f4eb2013-08-12 12:36:48 -070087 // Free the memory allocated in connect.
Jeff Thompson25b4e612013-10-10 16:03:24 -070088 free(elementReader_->partialData.array);
Jeff Thompsona00f4eb2013-08-12 12:36:48 -070089}
90
Jeff Thompsonfcf347d2013-07-15 11:30:44 -070091}