drivers_peripheral/usb/sample/device/liteos/prop_test.c

/*
 * Copyright (c) 2021 Huawei Device Co., Ltd.
 * 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 <stdio.h>
#include <sys/time.h>
#include <unistd.h>

#include "hdf_io_service_if.h"
#include "hdf_log.h"
#include "osal_mem.h"
#include "osal_mutex.h"
#include "osal_thread.h"
#include "osal_time.h"
#include "securec.h"
#include "usb_dev_test.h"

#define HDF_LOG_TAG      hcs_prop
#define ACM_SERVICE_NAME "usbfn_cdcacm"

enum UsbSerialCmd {
    USB_SERIAL_OPEN = 0,
    USB_SERIAL_CLOSE,
    USB_SERIAL_READ,
    USB_SERIAL_WRITE,
    USB_SERIAL_GET_BAUDRATE,
    USB_SERIAL_SET_BAUDRATE,
    USB_SERIAL_SET_PROP,
    USB_SERIAL_GET_PROP,
    USB_SERIAL_REGIST_PROP,
};

static struct HdfSBuf *g_data;
static struct HdfSBuf *g_reply;
static struct HdfIoService *g_acmService;

static void ShowUsage(void)
{
    HDF_LOGE("Usage options:\n");
    HDF_LOGE("g : name of getting prop, as: g idProduct");
    HDF_LOGE("s : name of setting prop, as: s idProduct 0xa4b7");
    HDF_LOGE("r : register prop, as: r testa aaaaa");
    HDF_LOGE("h : show this help message");
}

static int32_t DispatcherInit(void)
{
    g_acmService = HdfIoServiceBind("usbfn_cdcacm");
    if (g_acmService == NULL) {
        HDF_LOGE("%s: GetService err", __func__);
        return HDF_FAILURE;
    }

    g_data = HdfSbufObtainDefaultSize();
    g_reply = HdfSbufObtainDefaultSize();
    if (g_data == NULL || g_reply == NULL) {
        HDF_LOGE("%s: GetService err", __func__);
        return HDF_FAILURE;
    }
    return HDF_SUCCESS;
}

static void DispatcherDeInit(void)
{
    HdfSbufRecycle(g_data);
    HdfSbufRecycle(g_reply);
}

static int32_t TestPropGet(const char *propName)
{
    int32_t status = -1;
    const char *propVal = NULL;
    if (!HdfSbufWriteString(g_data, propName)) {
        HDF_LOGE("%s:failed to write result", __func__);
        goto FAIL;
    }
    status = g_acmService->dispatcher->Dispatch(&g_acmService->object, USB_SERIAL_GET_PROP, g_data, g_reply);
    if (status != HDF_SUCCESS) {
        HDF_LOGE("%s: Dispatch USB_SERIAL_GET_PROP failed status = %d", __func__, status);
        goto FAIL;
    }
    propVal = HdfSbufReadString(g_reply);
    if (propVal == NULL) {
        HDF_LOGE("%s:failed to write result", __func__);
        goto FAIL;
    }
    printf("%s: %s = %s\n", __func__, propName, propVal);

FAIL:
    return status;
}

static int32_t TestPropSet(const char *propName, const char *propValue)
{
    int32_t status = -1;
    if (!HdfSbufWriteString(g_data, propName)) {
        HDF_LOGE("%s:failed to write propName : %s", __func__, propName);
        goto FAIL;
    }
    if (!HdfSbufWriteString(g_data, propValue)) {
        HDF_LOGE("%s:failed to write propValue : %s", __func__, propValue);
        goto FAIL;
    }
    status = g_acmService->dispatcher->Dispatch(&g_acmService->object, USB_SERIAL_SET_PROP, g_data, g_reply);
    if (status != HDF_SUCCESS) {
        HDF_LOGE("%s: Dispatch USB_SERIAL_SET_PROP failed", __func__);
    }
FAIL:
    return status;
}

static int32_t TestPropRegist(const char *propName, const char *propValue)
{
    int32_t status;

    status = g_acmService->dispatcher->Dispatch(&g_acmService->object, USB_SERIAL_OPEN, g_data, g_reply);
    if (status != HDF_SUCCESS) {
        HDF_LOGE("%s: Dispatch USB_SERIAL_OPEN err", __func__);
        return HDF_FAILURE;
    }
    if (!HdfSbufWriteString(g_data, propName)) {
        HDF_LOGE("%s:failed to write propName : %s", __func__, propName);
        goto FAIL;
    }
    if (!HdfSbufWriteString(g_data, propValue)) {
        HDF_LOGE("%s:failed to write propValue : %s", __func__, propValue);
        goto FAIL;
    }
    status = g_acmService->dispatcher->Dispatch(&g_acmService->object, USB_SERIAL_REGIST_PROP, g_data, g_reply);
    if (status != HDF_SUCCESS) {
        HDF_LOGE("%s: Dispatch USB_SERIAL_SET_PROP failed status = %d", __func__, status);
    }
FAIL:
    status = g_acmService->dispatcher->Dispatch(&g_acmService->object, USB_SERIAL_CLOSE, g_data, g_reply);
    if (status != HDF_SUCCESS) {
        HDF_LOGE("%s: Dispatch USB_SERIAL_CLOSE err", __func__);
        return HDF_FAILURE;
    }

    return status;
}

static int32_t TestProp(const char *propName, const char *propValue, bool setProp, bool getProp, bool registProp)
{
    int32_t ret = HDF_SUCCESS;
    if (DispatcherInit() != HDF_SUCCESS) {
        return HDF_FAILURE;
    }
    if (getProp) {
        ret = TestPropGet(propName);
    } else if (setProp) {
        ret = TestPropSet(propName, propValue);
    } else if (registProp) {
        ret = TestPropRegist(propName, propValue);
    }
    DispatcherDeInit();
    return ret;
}

int32_t PropTest(int32_t argc, const char *argv[])
{
    int32_t ch;
    const char *propName = NULL;
    const char *propValue = NULL;
    bool setProp = false;
    bool getProp = false;
    bool registProp = false;

    ch = *(argv[1]);
    switch (ch) {
        case 'r':
            propName = argv[0x2];
            propValue = argv[0x3];
            registProp = true;
            break;
        case 'g':
            propName = argv[0x2];
            getProp = true;
            break;
        case 's':
            propName = argv[0x2];
            propValue = argv[0x3];
            setProp = true;
            break;
        case 'h':
        case '?':
            ShowUsage();
            return 0;
        default:
            break;
    }
    return TestProp(propName, propValue, setProp, getProp, registProp);
}