| /* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil; -*- */ |
| /* |
| * Copyright (c) 2014-2017, Regents of the University of California, |
| * Arizona Board of Regents, |
| * Colorado State University, |
| * University Pierre & Marie Curie, Sorbonne University, |
| * Washington University in St. Louis, |
| * Beijing Institute of Technology, |
| * The University of Memphis. |
| * |
| * This file is part of NFD (Named Data Networking Forwarding Daemon). |
| * See AUTHORS.md for complete list of NFD authors and contributors. |
| * |
| * NFD is free software: you can redistribute it and/or modify it under the terms |
| * of the GNU General Public License as published by the Free Software Foundation, |
| * either version 3 of the License, or (at your option) any later version. |
| * |
| * NFD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; |
| * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR |
| * PURPOSE. See the GNU General Public License for more details. |
| * |
| * You should have received a copy of the GNU General Public License along with |
| * NFD, e.g., in COPYING.md file. If not, see <http://www.gnu.org/licenses/>. |
| */ |
| |
| #include "face-module.hpp" |
| #include "find-face.hpp" |
| #include "format-helpers.hpp" |
| |
| namespace nfd { |
| namespace tools { |
| namespace nfdc { |
| |
| void |
| FaceModule::registerCommands(CommandParser& parser) |
| { |
| CommandDefinition defFaceList("face", "list"); |
| defFaceList |
| .setTitle("print face list") |
| .addArg("remote", ArgValueType::FACE_URI, Required::NO, Positional::YES) |
| .addArg("local", ArgValueType::FACE_URI, Required::NO, Positional::NO) |
| .addArg("scheme", ArgValueType::STRING, Required::NO, Positional::NO, "scheme"); |
| parser.addCommand(defFaceList, &FaceModule::list); |
| |
| CommandDefinition defFaceShow("face", "show"); |
| defFaceShow |
| .setTitle("show face information") |
| .addArg("id", ArgValueType::UNSIGNED, Required::YES, Positional::YES); |
| parser.addCommand(defFaceShow, &FaceModule::show); |
| |
| CommandDefinition defFaceCreate("face", "create"); |
| defFaceCreate |
| .setTitle("create a face") |
| .addArg("remote", ArgValueType::FACE_URI, Required::YES, Positional::YES) |
| .addArg("persistency", ArgValueType::FACE_PERSISTENCY, Required::NO, Positional::YES) |
| .addArg("local", ArgValueType::FACE_URI, Required::NO, Positional::NO) |
| .addArg("reliability", ArgValueType::BOOLEAN, Required::NO, Positional::NO); |
| parser.addCommand(defFaceCreate, &FaceModule::create); |
| |
| CommandDefinition defFaceDestroy("face", "destroy"); |
| defFaceDestroy |
| .setTitle("destroy a face") |
| .addArg("face", ArgValueType::FACE_ID_OR_URI, Required::YES, Positional::YES); |
| parser.addCommand(defFaceDestroy, &FaceModule::destroy); |
| } |
| |
| void |
| FaceModule::list(ExecuteContext& ctx) |
| { |
| auto remoteUri = ctx.args.getOptional<FaceUri>("remote"); |
| auto localUri = ctx.args.getOptional<FaceUri>("local"); |
| auto uriScheme = ctx.args.getOptional<std::string>("scheme"); |
| |
| FaceQueryFilter filter; |
| if (remoteUri) { |
| filter.setRemoteUri(remoteUri->toString()); |
| } |
| if (localUri) { |
| filter.setLocalUri(localUri->toString()); |
| } |
| if (uriScheme) { |
| filter.setUriScheme(*uriScheme); |
| } |
| |
| FindFace findFace(ctx); |
| FindFace::Code res = findFace.execute(filter, true); |
| |
| ctx.exitCode = static_cast<int>(res); |
| switch (res) { |
| case FindFace::Code::OK: |
| for (const FaceStatus& item : findFace.getResults()) { |
| formatItemText(ctx.out, item, false); |
| ctx.out << '\n'; |
| } |
| break; |
| case FindFace::Code::ERROR: |
| case FindFace::Code::NOT_FOUND: |
| case FindFace::Code::CANONIZE_ERROR: |
| ctx.err << findFace.getErrorReason() << '\n'; |
| break; |
| default: |
| BOOST_ASSERT_MSG(false, "unexpected FindFace result"); |
| break; |
| } |
| } |
| |
| void |
| FaceModule::show(ExecuteContext& ctx) |
| { |
| uint64_t faceId = ctx.args.get<uint64_t>("id"); |
| |
| FindFace findFace(ctx); |
| FindFace::Code res = findFace.execute(faceId); |
| |
| ctx.exitCode = static_cast<int>(res); |
| switch (res) { |
| case FindFace::Code::OK: |
| formatItemText(ctx.out, findFace.getFaceStatus(), true); |
| break; |
| case FindFace::Code::ERROR: |
| case FindFace::Code::NOT_FOUND: |
| ctx.err << findFace.getErrorReason() << '\n'; |
| break; |
| default: |
| BOOST_ASSERT_MSG(false, "unexpected FindFace result"); |
| break; |
| } |
| } |
| |
| /** \brief order persistency in NONE < ON_DEMAND < PERSISTENCY < PERMANENT |
| */ |
| static bool |
| persistencyLessThan(FacePersistency x, FacePersistency y) |
| { |
| switch (x) { |
| case FacePersistency::FACE_PERSISTENCY_NONE: |
| return y != FacePersistency::FACE_PERSISTENCY_NONE; |
| case FacePersistency::FACE_PERSISTENCY_ON_DEMAND: |
| return y == FacePersistency::FACE_PERSISTENCY_PERSISTENT || |
| y == FacePersistency::FACE_PERSISTENCY_PERMANENT; |
| case FacePersistency::FACE_PERSISTENCY_PERSISTENT: |
| return y == FacePersistency::FACE_PERSISTENCY_PERMANENT; |
| case FacePersistency::FACE_PERSISTENCY_PERMANENT: |
| return false; |
| } |
| return static_cast<int>(x) < static_cast<int>(y); |
| } |
| |
| void |
| FaceModule::create(ExecuteContext& ctx) |
| { |
| auto remoteUri = ctx.args.get<FaceUri>("remote"); |
| auto localUri = ctx.args.getOptional<FaceUri>("local"); |
| auto persistency = ctx.args.get<FacePersistency>("persistency", FacePersistency::FACE_PERSISTENCY_PERSISTENT); |
| auto lpReliability = ctx.args.getTribool("reliability"); |
| |
| FaceUri canonicalRemote; |
| ndn::optional<FaceUri> canonicalLocal; |
| |
| auto handleCanonizeError = [&] (const FaceUri& faceUri, const std::string& error) { |
| ctx.exitCode = 4; |
| ctx.err << "Error when canonizing '" << faceUri << "': " << error << '\n'; |
| }; |
| |
| auto printPositiveResult = [&] (const std::string& actionSummary, const ControlParameters& resp) { |
| text::ItemAttributes ia; |
| ctx.out << actionSummary << ' ' |
| << ia("id") << resp.getFaceId() |
| << ia("local") << resp.getLocalUri() |
| << ia("remote") << resp.getUri() |
| << ia("persistency") << resp.getFacePersistency() |
| << ia("reliability") << (resp.getFlagBit(ndn::nfd::BIT_LP_RELIABILITY_ENABLED) ? "on" : "off") |
| << '\n'; |
| }; |
| |
| auto updateFace = [&printPositiveResult] (ControlParameters respParams, ControlParameters resp) { |
| // faces/update response does not have FaceUris, copy from faces/create response |
| resp.setLocalUri(respParams.getLocalUri()) |
| .setUri(respParams.getUri()); |
| printPositiveResult("face-updated", resp); |
| }; |
| |
| auto handle409 = [&] (const ControlResponse& resp) { |
| ControlParameters respParams(resp.getBody()); |
| if (respParams.getUri() != canonicalRemote.toString()) { |
| // we are conflicting with a different face, which is a general error |
| return false; |
| } |
| |
| if (persistencyLessThan(respParams.getFacePersistency(), persistency)) { |
| // need to upgrade persistency |
| ControlParameters params; |
| params.setFaceId(respParams.getFaceId()).setFacePersistency(persistency); |
| if (!boost::logic::indeterminate(lpReliability)) { |
| params.setFlagBit(ndn::nfd::BIT_LP_RELIABILITY_ENABLED, lpReliability); |
| } |
| ctx.controller.start<ndn::nfd::FaceUpdateCommand>( |
| params, |
| bind(updateFace, respParams, _1), |
| ctx.makeCommandFailureHandler("upgrading face persistency"), |
| ctx.makeCommandOptions()); |
| } |
| else if (lpReliability && !respParams.getFlagBit(ndn::nfd::BIT_LP_RELIABILITY_ENABLED)) { |
| // enable reliability |
| ControlParameters params; |
| params.setFaceId(respParams.getFaceId()).setFlagBit(ndn::nfd::BIT_LP_RELIABILITY_ENABLED, true); |
| ctx.controller.start<ndn::nfd::FaceUpdateCommand>( |
| params, |
| bind(updateFace, respParams, _1), |
| ctx.makeCommandFailureHandler("enabling reliability"), |
| ctx.makeCommandOptions()); |
| } |
| else if (!lpReliability && respParams.getFlagBit(ndn::nfd::BIT_LP_RELIABILITY_ENABLED)) { |
| // disable reliability |
| ControlParameters params; |
| params.setFaceId(respParams.getFaceId()).setFlagBit(ndn::nfd::BIT_LP_RELIABILITY_ENABLED, false); |
| ctx.controller.start<ndn::nfd::FaceUpdateCommand>( |
| params, |
| bind(updateFace, respParams, _1), |
| ctx.makeCommandFailureHandler("disabling reliability"), |
| ctx.makeCommandOptions()); |
| } |
| else { |
| // don't do anything |
| printPositiveResult("face-exists", respParams); |
| } |
| return true; |
| }; |
| |
| auto doCreateFace = [&] { |
| ControlParameters params; |
| params.setUri(canonicalRemote.toString()); |
| if (canonicalLocal) { |
| params.setLocalUri(canonicalLocal->toString()); |
| } |
| params.setFacePersistency(persistency); |
| if (!boost::logic::indeterminate(lpReliability)) { |
| params.setFlagBit(ndn::nfd::BIT_LP_RELIABILITY_ENABLED, lpReliability); |
| } |
| |
| ctx.controller.start<ndn::nfd::FaceCreateCommand>( |
| params, |
| bind(printPositiveResult, "face-created", _1), |
| [&] (const ControlResponse& resp) { |
| if (resp.getCode() == 409 && handle409(resp)) { |
| return; |
| } |
| ctx.makeCommandFailureHandler("creating face")(resp); // invoke general error handler |
| }, |
| ctx.makeCommandOptions()); |
| }; |
| |
| remoteUri.canonize( |
| [&] (const FaceUri& canonicalUri) { |
| canonicalRemote = canonicalUri; |
| if (localUri) { |
| localUri->canonize( |
| [&] (const FaceUri& canonicalUri) { |
| canonicalLocal = canonicalUri; |
| doCreateFace(); |
| }, |
| bind(handleCanonizeError, *localUri, _1), |
| ctx.face.getIoService(), ctx.getTimeout()); |
| } |
| else { |
| doCreateFace(); |
| } |
| }, |
| bind(handleCanonizeError, remoteUri, _1), |
| ctx.face.getIoService(), ctx.getTimeout()); |
| |
| ctx.face.processEvents(); |
| } |
| |
| void |
| FaceModule::destroy(ExecuteContext& ctx) |
| { |
| const boost::any& faceIdOrUri = ctx.args.at("face"); |
| |
| FindFace findFace(ctx); |
| FindFace::Code res = findFace.execute(faceIdOrUri); |
| |
| ctx.exitCode = static_cast<int>(res); |
| switch (res) { |
| case FindFace::Code::OK: |
| break; |
| case FindFace::Code::ERROR: |
| case FindFace::Code::CANONIZE_ERROR: |
| case FindFace::Code::NOT_FOUND: |
| ctx.err << findFace.getErrorReason() << '\n'; |
| return; |
| case FindFace::Code::AMBIGUOUS: |
| ctx.err << "Multiple faces match specified remote FaceUri. Re-run the command with a FaceId:"; |
| findFace.printDisambiguation(ctx.err, FindFace::DisambiguationStyle::LOCAL_URI); |
| ctx.err << '\n'; |
| return; |
| default: |
| BOOST_ASSERT_MSG(false, "unexpected FindFace result"); |
| return; |
| } |
| |
| const FaceStatus& face = findFace.getFaceStatus(); |
| |
| ctx.controller.start<ndn::nfd::FaceDestroyCommand>( |
| ControlParameters().setFaceId(face.getFaceId()), |
| [&] (const ControlParameters& resp) { |
| ctx.out << "face-destroyed "; |
| text::ItemAttributes ia; |
| ctx.out << ia("id") << face.getFaceId() |
| << ia("local") << face.getLocalUri() |
| << ia("remote") << face.getRemoteUri() |
| << ia("persistency") << face.getFacePersistency() |
| << ia("reliability") << (resp.getFlagBit(ndn::nfd::BIT_LP_RELIABILITY_ENABLED) ? "on" : "off") |
| << '\n'; |
| }, |
| ctx.makeCommandFailureHandler("destroying face"), |
| ctx.makeCommandOptions()); |
| |
| ctx.face.processEvents(); |
| } |
| |
| void |
| FaceModule::fetchStatus(Controller& controller, |
| const function<void()>& onSuccess, |
| const Controller::DatasetFailCallback& onFailure, |
| const CommandOptions& options) |
| { |
| controller.fetch<ndn::nfd::FaceDataset>( |
| [this, onSuccess] (const std::vector<FaceStatus>& result) { |
| m_status = result; |
| onSuccess(); |
| }, |
| onFailure, options); |
| } |
| |
| void |
| FaceModule::formatStatusXml(std::ostream& os) const |
| { |
| os << "<faces>"; |
| for (const FaceStatus& item : m_status) { |
| this->formatItemXml(os, item); |
| } |
| os << "</faces>"; |
| } |
| |
| void |
| FaceModule::formatItemXml(std::ostream& os, const FaceStatus& item) const |
| { |
| os << "<face>"; |
| |
| os << "<faceId>" << item.getFaceId() << "</faceId>"; |
| os << "<remoteUri>" << xml::Text{item.getRemoteUri()} << "</remoteUri>"; |
| os << "<localUri>" << xml::Text{item.getLocalUri()} << "</localUri>"; |
| |
| if (item.hasExpirationPeriod()) { |
| os << "<expirationPeriod>" << xml::formatDuration(item.getExpirationPeriod()) |
| << "</expirationPeriod>"; |
| } |
| os << "<faceScope>" << item.getFaceScope() << "</faceScope>"; |
| os << "<facePersistency>" << item.getFacePersistency() << "</facePersistency>"; |
| os << "<linkType>" << item.getLinkType() << "</linkType>"; |
| |
| if (item.getFlags() == 0) { |
| os << "<flags/>"; |
| } |
| else { |
| os << "<flags>"; |
| if (item.getFlagBit(ndn::nfd::BIT_LOCAL_FIELDS_ENABLED)) { |
| os << "<localFieldsEnabled/>"; |
| } |
| if (item.getFlagBit(ndn::nfd::BIT_LP_RELIABILITY_ENABLED)) { |
| os << "<lpReliabilityEnabled/>"; |
| } |
| os << "</flags>"; |
| } |
| |
| os << "<packetCounters>"; |
| os << "<incomingPackets>" |
| << "<nInterests>" << item.getNInInterests() << "</nInterests>" |
| << "<nData>" << item.getNInData() << "</nData>" |
| << "<nNacks>" << item.getNInNacks() << "</nNacks>" |
| << "</incomingPackets>"; |
| os << "<outgoingPackets>" |
| << "<nInterests>" << item.getNOutInterests() << "</nInterests>" |
| << "<nData>" << item.getNOutData() << "</nData>" |
| << "<nNacks>" << item.getNOutNacks() << "</nNacks>" |
| << "</outgoingPackets>"; |
| os << "</packetCounters>"; |
| |
| os << "<byteCounters>"; |
| os << "<incomingBytes>" << item.getNInBytes() << "</incomingBytes>"; |
| os << "<outgoingBytes>" << item.getNOutBytes() << "</outgoingBytes>"; |
| os << "</byteCounters>"; |
| |
| os << "</face>"; |
| } |
| |
| void |
| FaceModule::formatStatusText(std::ostream& os) const |
| { |
| os << "Faces:\n"; |
| for (const FaceStatus& item : m_status) { |
| os << " "; |
| formatItemText(os, item, false); |
| os << '\n'; |
| } |
| } |
| |
| void |
| FaceModule::formatItemText(std::ostream& os, const FaceStatus& item, bool wantMultiLine) |
| { |
| text::ItemAttributes ia(wantMultiLine, 8); |
| |
| os << ia("faceid") << item.getFaceId(); |
| os << ia("remote") << item.getRemoteUri(); |
| os << ia("local") << item.getLocalUri(); |
| |
| if (item.hasExpirationPeriod()) { |
| os << ia("expires") << text::formatDuration(item.getExpirationPeriod()); |
| } |
| |
| os << ia("counters") |
| << "{in={" |
| << item.getNInInterests() << "i " |
| << item.getNInData() << "d " |
| << item.getNInNacks() << "n " |
| << item.getNInBytes() << "B} " |
| << "out={" |
| << item.getNOutInterests() << "i " |
| << item.getNOutData() << "d " |
| << item.getNOutNacks() << "n " |
| << item.getNOutBytes() << "B}}"; |
| |
| os << ia("flags") << '{'; |
| text::Separator flagSep("", " "); |
| os << flagSep << item.getFaceScope(); |
| os << flagSep << item.getFacePersistency(); |
| os << flagSep << item.getLinkType(); |
| if (item.getFlagBit(ndn::nfd::BIT_LOCAL_FIELDS_ENABLED)) { |
| os << flagSep << "local-fields"; |
| } |
| if (item.getFlagBit(ndn::nfd::BIT_LP_RELIABILITY_ENABLED)) { |
| os << flagSep << "lp-reliability"; |
| } |
| os << '}'; |
| |
| os << ia.end(); |
| } |
| |
| } // namespace nfdc |
| } // namespace tools |
| } // namespace nfd |