src/setup_payload/SetupPayloadHelper.cpp

/*
 *
 *    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 "SetupPayloadHelper.h"
#include "ManualSetupPayloadGenerator.h"
#include "QRCodeSetupPayloadGenerator.h"
#include "SetupPayload.h"
#include <fstream>

#include <support/CodeUtils.h>

using namespace chip;

namespace chip {

enum SetupPayloadKey
{
    SetupPayloadKey_Version,
    SetupPayloadKey_VendorID,
    SetupPayloadKey_ProductID,
    SetupPayloadKey_RequiresCustomFlowTrue,
    SetupPayloadKey_RendezVousInformation,
    SetupPayloadKey_Discriminator,
    SetupPayloadKey_SetupPINCode,
};

struct SetupPayloadParameter
{
    SetupPayloadKey key;
    string stringValue;
    uint64_t uintValue;
};

static CHIP_ERROR resolveSetupPayloadParameter(SetupPayloadParameter & parameter, string key, string value)
{
    bool isUnsignedInt   = true;
    bool shouldHaveValue = true;
    if (key.compare("version") == 0)
    {
        parameter.key = SetupPayloadKey_Version;
    }
    else if (key.compare("vendorID") == 0)
    {
        parameter.key = SetupPayloadKey_VendorID;
    }
    else if (key.compare("productID") == 0)
    {
        parameter.key = SetupPayloadKey_ProductID;
    }
    else if (key.compare("requiresCustomFlowTrue") == 0)
    {
        parameter.key   = SetupPayloadKey_RequiresCustomFlowTrue;
        shouldHaveValue = false;
    }
    else if (key.compare("rendezVousInformation") == 0)
    {
        parameter.key = SetupPayloadKey_RendezVousInformation;
    }
    else if (key.compare("discriminator") == 0)
    {
        parameter.key = SetupPayloadKey_Discriminator;
    }
    else if (key.compare("setUpPINCode") == 0)
    {
        parameter.key = SetupPayloadKey_SetupPINCode;
    }
    else
    {
        return CHIP_ERROR_INVALID_ARGUMENT;
    }
    if (shouldHaveValue && value.length() > 0)
    {
        if (isUnsignedInt)
        {
            parameter.uintValue = stoul(value, nullptr, 10);
        }
        else
        {
            parameter.stringValue = value;
        }
    }
    else if (shouldHaveValue)
    {
        return CHIP_ERROR_INVALID_ARGUMENT;
    }
    return CHIP_NO_ERROR;
}

static CHIP_ERROR addParameter(SetupPayload & setupPayload, SetupPayloadParameter parameter)
{
    switch (parameter.key)
    {
    case SetupPayloadKey_Version:
        printf("Loaded version: %u\n", (uint8_t) parameter.uintValue);
        setupPayload.version = (uint8_t) parameter.uintValue;
        break;
    case SetupPayloadKey_VendorID:
        printf("Loaded vendorID: %u\n", (uint16_t) parameter.uintValue);
        setupPayload.vendorID = (uint16_t) parameter.uintValue;
        break;
    case SetupPayloadKey_ProductID:
        printf("Loaded productID: %u\n", (uint16_t) parameter.uintValue);
        setupPayload.productID = (uint16_t) parameter.uintValue;
        break;
    case SetupPayloadKey_RequiresCustomFlowTrue:
        printf("Requires custom flow was set to true\n");
        setupPayload.requiresCustomFlow = true;
        break;
    case SetupPayloadKey_RendezVousInformation:
        printf("Loaded rendezvousInfo: %u\n", (uint16_t) parameter.uintValue);
        setupPayload.rendezvousInformation = (uint16_t) parameter.uintValue;
        break;
    case SetupPayloadKey_Discriminator:
        printf("Loaded discriminator: %u\n", (uint16_t) parameter.uintValue);
        setupPayload.discriminator = (uint16_t) parameter.uintValue;
        break;
    case SetupPayloadKey_SetupPINCode:
        printf("Loaded setupPinCode: %lu\n", (unsigned long) parameter.uintValue);
        setupPayload.setUpPINCode = (uint32_t) parameter.uintValue;
        break;
    default:
        return CHIP_ERROR_INVALID_ARGUMENT;
    }
    return CHIP_NO_ERROR;
}

CHIP_ERROR loadPayloadFromFile(SetupPayload & setupPayload, string filePath)
{
    CHIP_ERROR err = CHIP_NO_ERROR;
    ifstream fileStream(filePath);
    VerifyOrExit(!fileStream.fail(), err = CHIP_ERROR_INVALID_ARGUMENT);

    while (fileStream)
    {
        string key;
        string value;
        SetupPayloadParameter parameter;

        getline(fileStream, key, ' ');
        fileStream >> value;
        fileStream.ignore();

        if (key.length() == 0)
        {
            continue;
        }
        err = resolveSetupPayloadParameter(parameter, key, value);
        SuccessOrExit(err);

        err = addParameter(setupPayload, parameter);
        SuccessOrExit(err);
    }
exit:
    return err;
}

CHIP_ERROR generateQRCodeFromFilePath(string filePath, string & outCode)
{
    SetupPayload setupPayload;
    CHIP_ERROR err = loadPayloadFromFile(setupPayload, filePath);
    if (err != CHIP_NO_ERROR)
    {
        return err;
    }
    QRCodeSetupPayloadGenerator generator(setupPayload);
    err = generator.payloadBase41Representation(outCode);
    return err;
}

CHIP_ERROR generateManualCodeFromFilePath(string filePath, string & outCode)
{
    SetupPayload setupPayload;
    CHIP_ERROR err = loadPayloadFromFile(setupPayload, filePath);
    if (err != CHIP_NO_ERROR)
    {
        return err;
    }
    ManualSetupPayloadGenerator generator(setupPayload);
    err = generator.payloadDecimalStringRepresentation(outCode);
    return err;
}
} // namespace chip