blob: 5c999a1faf8be9e034e90d31db1ad947db1397f0 [file] [log] [blame]
Wentao Shang53df1632014-04-21 12:01:32 -07001/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */
2/**
Junxiao Shi08d07a72014-06-09 23:17:57 -07003 * Copyright (c) 2014, Regents of the University of California,
4 * Arizona Board of Regents,
5 * Colorado State University,
6 * University Pierre & Marie Curie, Sorbonne University,
7 * Washington University in St. Louis,
8 * Beijing Institute of Technology,
9 * The University of Memphis
Wentao Shang53df1632014-04-21 12:01:32 -070010 *
11 * This file is part of NFD (Named Data Networking Forwarding Daemon).
12 * See AUTHORS.md for complete list of NFD authors and contributors.
13 *
14 * NFD is free software: you can redistribute it and/or modify it under the terms
15 * of the GNU General Public License as published by the Free Software Foundation,
16 * either version 3 of the License, or (at your option) any later version.
17 *
18 * NFD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
19 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
20 * PURPOSE. See the GNU General Public License for more details.
21 *
22 * You should have received a copy of the GNU General Public License along with
23 * NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>.
24 **/
25
26#include "websocket-face.hpp"
Wentao Shang98733142014-09-17 12:13:57 -070027#include "core/global-io.hpp"
Wentao Shang53df1632014-04-21 12:01:32 -070028
29namespace nfd {
30
31NFD_LOG_INIT("WebSocketFace");
32
33WebSocketFace::WebSocketFace(const FaceUri& remoteUri, const FaceUri& localUri,
34 websocketpp::connection_hdl hdl,
35 websocket::Server& server)
36 : Face(remoteUri, localUri)
37 , m_handle(hdl)
38 , m_server(server)
39 , m_closed(false)
40{
Wentao Shang98733142014-09-17 12:13:57 -070041 this->setOnDemand(true);
Wentao Shang53df1632014-04-21 12:01:32 -070042}
43
Wentao Shang53df1632014-04-21 12:01:32 -070044void
45WebSocketFace::sendInterest(const Interest& interest)
46{
Wentao Shang3d36c2b2014-07-04 10:09:58 -040047 if (m_closed)
48 return;
49
Wentao Shang53df1632014-04-21 12:01:32 -070050 this->onSendInterest(interest);
51 const Block& payload = interest.wireEncode();
Wentao Shange612e2f2014-08-04 10:24:59 -040052 this->getMutableCounters().getNOutBytes() += payload.size();
Wentao Shang3d36c2b2014-07-04 10:09:58 -040053
54 try {
55 m_server.send(m_handle, payload.wire(), payload.size(),
56 websocketpp::frame::opcode::binary);
57 }
58 catch (const websocketpp::lib::error_code& e) {
59 NFD_LOG_DEBUG("Failed to send Interest because: " << e
60 << "(" << e.message() << ")");
61 }
Wentao Shang53df1632014-04-21 12:01:32 -070062}
63
64void
65WebSocketFace::sendData(const Data& data)
66{
Wentao Shang3d36c2b2014-07-04 10:09:58 -040067 if (m_closed)
68 return;
69
Wentao Shang53df1632014-04-21 12:01:32 -070070 this->onSendData(data);
71 const Block& payload = data.wireEncode();
Wentao Shange612e2f2014-08-04 10:24:59 -040072 this->getMutableCounters().getNOutBytes() += payload.size();
Wentao Shang3d36c2b2014-07-04 10:09:58 -040073
74 try {
75 m_server.send(m_handle, payload.wire(), payload.size(),
76 websocketpp::frame::opcode::binary);
77 }
78 catch (const websocketpp::lib::error_code& e) {
79 NFD_LOG_DEBUG("Failed to send Data because: " << e
80 << "(" << e.message() << ")");
81 }
Wentao Shang53df1632014-04-21 12:01:32 -070082}
83
84void
85WebSocketFace::close()
86{
87 if (m_closed == false)
88 {
89 m_closed = true;
Wentao Shang98733142014-09-17 12:13:57 -070090 scheduler::cancel(m_pingEventId);
Wentao Shang53df1632014-04-21 12:01:32 -070091 websocketpp::lib::error_code ecode;
92 m_server.close(m_handle, websocketpp::close::status::normal, "closed by nfd", ecode);
Alexander Afanasyev251f3c12014-06-02 18:39:58 +030093
Junxiao Shi08d07a72014-06-09 23:17:57 -070094 fail("Face closed");
Wentao Shang53df1632014-04-21 12:01:32 -070095 }
96}
97
98void
99WebSocketFace::handleReceive(const std::string& msg)
100{
101 // Copy message into Face internal buffer
Wentao Shang6990e4c2014-11-05 11:17:35 -0800102 if (msg.size() > MAX_NDN_PACKET_SIZE)
103 {
104 NFD_LOG_WARN("[id:" << this->getId()
105 << "] Received WebSocket message size ["
106 << msg.size() << "] is too big");
107 return;
108 }
Wentao Shang53df1632014-04-21 12:01:32 -0700109
Wentao Shange612e2f2014-08-04 10:24:59 -0400110 this->getMutableCounters().getNInBytes() += msg.size();
111
Wentao Shang53df1632014-04-21 12:01:32 -0700112 // Try to parse message data
113 bool isOk = true;
114 Block element;
115 isOk = Block::fromBuffer(reinterpret_cast<const uint8_t*>(msg.c_str()), msg.size(), element);
116 if (!isOk)
117 {
Wentao Shang6990e4c2014-11-05 11:17:35 -0800118 NFD_LOG_WARN("[id:" << this->getId()
119 << "] Received invalid NDN packet of length ["
120 << msg.size() << "]");
Wentao Shang53df1632014-04-21 12:01:32 -0700121 return;
122 }
123
124 if (!this->decodeAndDispatchInput(element))
125 {
126 NFD_LOG_WARN("[id:" << this->getId()
127 << "] Received unrecognized block of type ["
128 << element.type() << "]");
129 // ignore unknown packet and proceed
130 }
131}
132
133} // namespace nfd