blob: 9d2f90c7858e00f1a311bf860224b435a697d75d [file] [log] [blame]
/*
*
* Copyright (c) 2020 Project CHIP Authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include <cstdio>
#include <memory>
#include <arpa/inet.h>
#include <strings.h>
#include <TracingCommandLineArgument.h>
#include <lib/dnssd/Advertiser.h>
#include <lib/support/CHIPArgParser.hpp>
#include <lib/support/CHIPMem.h>
#include <lib/support/Span.h>
#include <platform/CHIPDeviceLayer.h>
using namespace chip;
namespace {
enum class AdvertisingMode
{
kCommissionableNode,
kOperational,
kOperationalMultiAdmin,
kCommissioner,
};
struct Options
{
bool enableIpV4 = false;
AdvertisingMode advertisingMode = AdvertisingMode::kCommissionableNode;
// commissionable node / commissioner params
Optional<uint16_t> vendorId;
Optional<uint16_t> productId;
Optional<uint32_t> deviceType;
Optional<const char *> deviceName;
// commissionable node params
uint8_t shortDiscriminator = 52;
uint16_t longDiscriminator = 840;
Dnssd::CommissioningMode commissioningMode = Dnssd::CommissioningMode::kDisabled;
Optional<const char *> rotatingId;
Optional<const char *> pairingInstr;
Optional<uint16_t> pairingHint;
// operational params
uint64_t fabricId = 12345;
uint64_t nodeId = 6789;
uint8_t mac[6] = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06 };
} gOptions;
using namespace chip::ArgParser;
constexpr uint16_t kOptionEnableIpV4 = '4';
constexpr uint16_t kOptionAdvertisingMode = 'm';
constexpr uint16_t kOptionCommissioningShortDiscriminator = 's';
constexpr uint16_t kOptionCommissioningLongDiscriminaotr = 'l';
constexpr uint16_t kOptionCommissioningVendorId = 0x100; // v is used by 'version'
constexpr uint16_t kOptionCommissioningProductId = 'p';
constexpr uint16_t kOptionCommissioningPairingInstr = 0x200; // Just use the long format
constexpr uint16_t kOptionCommissioningPairingHint = 0x300;
constexpr uint16_t kOptionCommissioningDeviceType = 0x400;
constexpr uint16_t kOptionCommissioningDeviceName = 0x500;
constexpr uint16_t kOptionCommissioningMode = 0x600;
constexpr uint16_t kOptionCommissioningRotatingId = 0x700;
constexpr uint16_t kOptionOperationalFabricId = 'f';
constexpr uint16_t kOptionOperationalNodeId = 'n';
constexpr uint16_t kOptionTraceTo = 't';
// Only used for argument parsing. Tracing setup owned by the main loop.
chip::CommandLineApp::TracingSetup * tracing_setup_for_argparse = nullptr;
bool HandleOptions(const char * aProgram, OptionSet * aOptions, int aIdentifier, const char * aName, const char * aValue)
{
uint8_t cm;
switch (aIdentifier)
{
case kOptionEnableIpV4:
gOptions.enableIpV4 = true;
return true;
case kOptionTraceTo:
tracing_setup_for_argparse->EnableTracingFor(aValue);
return true;
case kOptionAdvertisingMode:
if (strcmp(aValue, "operational") == 0)
{
gOptions.advertisingMode = AdvertisingMode::kOperational;
}
else if (strcmp(aValue, "operational-multi-admin") == 0)
{
gOptions.advertisingMode = AdvertisingMode::kOperationalMultiAdmin;
}
else if (strcmp(aValue, "commissionable-node") == 0)
{
gOptions.advertisingMode = AdvertisingMode::kCommissionableNode;
}
else if (strcmp(aValue, "commissioner") == 0)
{
gOptions.advertisingMode = AdvertisingMode::kCommissioner;
}
else
{
PrintArgError("%s: Invalid advertising mode %s\n", aProgram, aValue);
return false;
}
return true;
case kOptionCommissioningShortDiscriminator:
gOptions.shortDiscriminator = static_cast<uint8_t>(atoi(aValue));
return true;
case kOptionCommissioningLongDiscriminaotr:
gOptions.longDiscriminator = static_cast<uint16_t>(atoi(aValue));
return true;
case kOptionCommissioningVendorId:
gOptions.vendorId = Optional<uint16_t>::Value(static_cast<uint16_t>(atoi(aValue)));
return true;
case kOptionCommissioningProductId:
gOptions.productId = Optional<uint16_t>::Value(static_cast<uint16_t>(atoi(aValue)));
return true;
case kOptionCommissioningMode:
cm = static_cast<uint8_t>(atoi(aValue));
if (cm == 1)
{
gOptions.commissioningMode = Dnssd::CommissioningMode::kEnabledBasic;
}
if (cm == 2)
{
gOptions.commissioningMode = Dnssd::CommissioningMode::kEnabledEnhanced;
}
return true;
case kOptionCommissioningDeviceType:
gOptions.deviceType = Optional<uint32_t>::Value(static_cast<uint32_t>(atoi(aValue)));
return true;
case kOptionCommissioningDeviceName:
gOptions.deviceName = Optional<const char *>::Value(static_cast<const char *>(aValue));
return true;
case kOptionCommissioningRotatingId:
gOptions.rotatingId = Optional<const char *>::Value(static_cast<const char *>(aValue));
return true;
case kOptionCommissioningPairingInstr:
gOptions.pairingInstr = Optional<const char *>::Value(static_cast<const char *>(aValue));
return true;
case kOptionCommissioningPairingHint:
gOptions.pairingHint = Optional<uint16_t>::Value(static_cast<uint16_t>(atoi(aValue)));
return true;
case kOptionOperationalFabricId:
if (sscanf(aValue, "%" SCNx64, &gOptions.fabricId) != 1)
{
PrintArgError("%s: Invalid fabric id: %s\n", aProgram, aValue);
return false;
}
return true;
case kOptionOperationalNodeId:
if (sscanf(aValue, "%" SCNx64, &gOptions.nodeId) != 1)
{
PrintArgError("%s: Invalid node id: %s\n", aProgram, aValue);
return false;
}
return true;
default:
PrintArgError("%s: INTERNAL ERROR: Unhandled option: %s\n", aProgram, aName);
return false;
}
}
OptionDef cmdLineOptionsDef[] = {
#if INET_CONFIG_ENABLE_IPV4
{ "enable-ip-v4", kNoArgument, kOptionEnableIpV4 },
#endif
{ "advertising-mode", kArgumentRequired, kOptionAdvertisingMode },
{ "short-discriminator", kArgumentRequired, kOptionCommissioningShortDiscriminator },
{ "long-discriminator", kArgumentRequired, kOptionCommissioningLongDiscriminaotr },
{ "vendor-id", kArgumentRequired, kOptionCommissioningVendorId },
{ "product-id", kArgumentRequired, kOptionCommissioningProductId },
{ "commissioning-mode", kNoArgument, kOptionCommissioningMode },
{ "device-type", kArgumentRequired, kOptionCommissioningDeviceType },
{ "device-name", kArgumentRequired, kOptionCommissioningDeviceName },
{ "rotating-id", kArgumentRequired, kOptionCommissioningRotatingId },
{ "pairing-instruction", kArgumentRequired, kOptionCommissioningPairingInstr },
{ "pairing-hint", kArgumentRequired, kOptionCommissioningPairingHint },
{ "fabric-id", kArgumentRequired, kOptionOperationalFabricId },
{ "node-id", kArgumentRequired, kOptionOperationalNodeId },
{ "trace-to", kArgumentRequired, kOptionTraceTo },
{},
};
OptionSet cmdLineOptions = {
HandleOptions, cmdLineOptionsDef, "PROGRAM OPTIONS",
#if INET_CONFIG_ENABLE_IPV4
" -4\n"
" --enable-ip-v4\n"
" enable listening on IPv4\n"
#endif
" -m <mode>\n"
" --advertising-mode <mode>\n"
" Advertise in this mode (operational, operational-multi-admin, commissionable-node or commissioner).\n"
" --short-discriminator <value>\n"
" -s <value>\n"
" Commissioning/commissionable short discriminator.\n"
" --long-discriminator <value>\n"
" -l <value>\n"
" Commissioning/commissionable long discriminator.\n"
" --vendor-id <value>\n"
" Commissioning/commissionable vendor id.\n"
" --product-id <value>\n"
" -p <value>\n"
" Commissioning/commissionable product id.\n"
" --commissioning-mode <value>\n"
" Commissioning Mode (0=disabled, 1=basic, 2=enhanced).\n"
" --device-type <value>\n"
" Device type id.\n"
" --device-name <value>\n"
" Name of device.\n"
" --rotating-id <value>\n"
" Rotating Id.\n"
" --pairing-instruction <value>\n"
" Commissionable pairing instruction.\n"
" --pairing-hint <value>\n"
" Commissionable pairing hint.\n"
" --fabric-id <value>\n"
" -f <value>\n"
" Operational fabric id.\n"
" --node-id <value>\n"
" -n <value>\n"
" Operational node id.\n"
" -t <dest>\n"
" --trace-to <dest>\n"
" trace to the given destination (supported: " SUPPORTED_COMMAND_LINE_TRACING_TARGETS ").\n"
"\n"
};
HelpOptions helpOptions("advertiser", "Usage: advertiser [options]", "1.0");
OptionSet * allOptions[] = { &cmdLineOptions, &helpOptions, nullptr };
void StopSignalHandler(int signal)
{
DeviceLayer::PlatformMgr().StopEventLoopTask();
}
} // namespace
int main(int argc, char ** args)
{
if (Platform::MemoryInit() != CHIP_NO_ERROR)
{
fprintf(stderr, "FAILED to initialize memory\n");
return 1;
}
if (DeviceLayer::PlatformMgr().InitChipStack() != CHIP_NO_ERROR)
{
fprintf(stderr, "FAILED to initialize chip stack\n");
return 1;
}
chip::CommandLineApp::TracingSetup tracing_setup;
tracing_setup_for_argparse = &tracing_setup;
if (!chip::ArgParser::ParseArgs(args[0], argc, args, allOptions))
{
return 1;
}
tracing_setup_for_argparse = nullptr;
if (chip::Dnssd::ServiceAdvertiser::Instance().Init(DeviceLayer::UDPEndPointManager()) != CHIP_NO_ERROR)
{
fprintf(stderr, "FAILED to start MDNS advertisement\n");
return 1;
}
CHIP_ERROR err;
if (gOptions.advertisingMode == AdvertisingMode::kCommissionableNode)
{
printf("Advertise Commissionable Node\n");
err = chip::Dnssd::ServiceAdvertiser::Instance().Advertise(chip::Dnssd::CommissionAdvertisingParameters()
.EnableIpV4(gOptions.enableIpV4)
.SetPort(CHIP_PORT)
.SetShortDiscriminator(gOptions.shortDiscriminator)
.SetLongDiscriminator(gOptions.longDiscriminator)
.SetMac(chip::ByteSpan(gOptions.mac, 6))
.SetVendorId(gOptions.vendorId)
.SetProductId(gOptions.productId)
.SetCommissioningMode(gOptions.commissioningMode)
.SetDeviceType(gOptions.deviceType)
.SetDeviceName(gOptions.deviceName)
.SetRotatingDeviceId(gOptions.rotatingId)
.SetPairingInstruction(gOptions.pairingInstr)
.SetPairingHint(gOptions.pairingHint));
}
else if (gOptions.advertisingMode == AdvertisingMode::kOperational)
{
err = chip::Dnssd::ServiceAdvertiser::Instance().Advertise(
chip::Dnssd::OperationalAdvertisingParameters()
.EnableIpV4(gOptions.enableIpV4)
.SetPort(CHIP_PORT)
.SetMac(chip::ByteSpan(gOptions.mac, 6))
.SetPeerId(PeerId().SetCompressedFabricId(gOptions.fabricId).SetNodeId(gOptions.nodeId)));
}
else if (gOptions.advertisingMode == AdvertisingMode::kOperationalMultiAdmin)
{
err = chip::Dnssd::ServiceAdvertiser::Instance().Advertise(
chip::Dnssd::OperationalAdvertisingParameters()
.EnableIpV4(gOptions.enableIpV4)
.SetPort(CHIP_PORT)
.SetMac(chip::ByteSpan(gOptions.mac, 6))
.SetPeerId(PeerId().SetCompressedFabricId(gOptions.fabricId).SetNodeId(gOptions.nodeId)));
if (err == CHIP_NO_ERROR)
{
err = chip::Dnssd::ServiceAdvertiser::Instance().Advertise(
chip::Dnssd::OperationalAdvertisingParameters()
.EnableIpV4(gOptions.enableIpV4)
.SetPort(CHIP_PORT + 1)
.SetMac(chip::ByteSpan(gOptions.mac, 6))
.SetPeerId(PeerId().SetCompressedFabricId(gOptions.fabricId + 1).SetNodeId(gOptions.nodeId + 1)));
}
if (err == CHIP_NO_ERROR)
{
err = chip::Dnssd::ServiceAdvertiser::Instance().Advertise(
chip::Dnssd::OperationalAdvertisingParameters()
.EnableIpV4(gOptions.enableIpV4)
.SetPort(CHIP_PORT + 2)
.SetMac(chip::ByteSpan(gOptions.mac, 6))
.SetPeerId(PeerId().SetCompressedFabricId(gOptions.fabricId + 2).SetNodeId(gOptions.nodeId + 2)));
}
}
else if (gOptions.advertisingMode == AdvertisingMode::kCommissioner)
{
printf("Advertise Commissioner\n");
err = chip::Dnssd::ServiceAdvertiser::Instance().Advertise(
chip::Dnssd::CommissionAdvertisingParameters()
.EnableIpV4(gOptions.enableIpV4)
.SetPort(CHIP_PORT)
.SetMac(chip::ByteSpan(gOptions.mac, 6))
.SetVendorId(gOptions.vendorId)
.SetProductId(gOptions.productId)
.SetDeviceType(gOptions.deviceType)
.SetDeviceName(gOptions.deviceName)
.SetCommissionAdvertiseMode(chip::Dnssd::CommssionAdvertiseMode::kCommissioner));
}
else
{
fprintf(stderr, "FAILED to determine advertising type.\n");
return 1;
}
if (err != CHIP_NO_ERROR)
{
fprintf(stderr, "FAILED to setup advertisement parameters err=%s\n", chip::ErrorStr(err));
return 1;
}
signal(SIGTERM, StopSignalHandler);
signal(SIGINT, StopSignalHandler);
DeviceLayer::PlatformMgr().RunEventLoop();
tracing_setup.StopTracing();
Dnssd::Resolver::Instance().Shutdown();
DeviceLayer::PlatformMgr().Shutdown();
printf("Done...\n");
return 0;
}