/*
 * Copyright (C) 2020 The Android Open Source Project
 *
 * 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.
 */
#define LOG_TAG "android.hardware.usb.gadget@1.1-service.rosemary"
#include "UsbGadget.h"
#include <dirent.h>
#include <fcntl.h>
#include <stdio.h>
#include <sys/inotify.h>
#include <sys/mount.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
namespace android {
namespace hardware {
namespace usb {
namespace gadget {
namespace V1_1 {
namespace implementation {
UsbGadget::UsbGadget() {
    if (access(OS_DESC_PATH, R_OK) != 0) {
        ALOGE("configfs setup not done yet");
        abort();
    }
}
void currentFunctionsAppliedCallback(bool functionsApplied, void* payload) {
    UsbGadget* gadget = (UsbGadget*)payload;
    gadget->mCurrentUsbFunctionsApplied = functionsApplied;
}
Return<void> UsbGadget::getCurrentUsbFunctions(const sp<V1_0::IUsbGadgetCallback>& callback) {
    Return<void> ret = callback->getCurrentUsbFunctionsCb(
            mCurrentUsbFunctions, mCurrentUsbFunctionsApplied ? Status::FUNCTIONS_APPLIED
                                                              : Status::FUNCTIONS_NOT_APPLIED);
    if (!ret.isOk()) ALOGE("Call to getCurrentUsbFunctionsCb failed %s", ret.description().c_str());
    return Void();
}
V1_0::Status UsbGadget::tearDownGadget() {
    if (resetGadget() != V1_0::Status::SUCCESS) return V1_0::Status::ERROR;
    return V1_0::Status::SUCCESS;
}
Return<Status> UsbGadget::reset() {
    if (!WriteStringToFile("none", PULLUP_PATH)) {
        ALOGI("Gadget cannot be pulled down");
        return Status::ERROR;
    }
    return Status::SUCCESS;
}
static V1_0::Status validateAndSetVidPid(uint64_t functions) {
    V1_0::Status ret = V1_0::Status::SUCCESS;
    switch (functions) {
        case static_cast<uint64_t>(V1_0::GadgetFunction::MTP):
            ret = setVidPid("0x2717", "0xFF40");
            WriteStringToFile("mtp", CONFIG_STR_PATH);
            WriteStringToFile("1", DESC_USE_PATH);
            break;
        case V1_0::GadgetFunction::ADB | V1_0::GadgetFunction::MTP:
            ret = setVidPid("0x2717", "0xFF48");
            WriteStringToFile("mtp_adb", CONFIG_STR_PATH);
            WriteStringToFile("1", DESC_USE_PATH);
            break;
        case static_cast<uint64_t>(V1_0::GadgetFunction::RNDIS):
            ret = setVidPid("0x2717", "0xFF80");
            WriteStringToFile("rndis", CONFIG_STR_PATH);
            break;
        case V1_0::GadgetFunction::ADB | V1_0::GadgetFunction::RNDIS:
            ret = setVidPid("0x2717", "0xFF88");
            WriteStringToFile("rndis_adb", CONFIG_STR_PATH);
            break;
        case static_cast<uint64_t>(V1_0::GadgetFunction::PTP):
            ret = setVidPid("0x2717", "0xFF10");
            WriteStringToFile("ptp", CONFIG_STR_PATH);
            break;
        case V1_0::GadgetFunction::ADB | V1_0::GadgetFunction::PTP:
            ret = setVidPid("0x2717", "0xFF18");
            WriteStringToFile("ptp_adb", CONFIG_STR_PATH);
            break;
        case static_cast<uint64_t>(V1_0::GadgetFunction::ADB):
            ret = setVidPid("0x18d1", "0x4ee7");
            WriteStringToFile("adb", CONFIG_STR_PATH);
            break;
        case static_cast<uint64_t>(V1_0::GadgetFunction::MIDI):
            ret = setVidPid("0x0E8D", "0x2046");
            break;
        case V1_0::GadgetFunction::ADB | V1_0::GadgetFunction::MIDI:
            ret = setVidPid("0x0E8D", "0x2048");
            break;
        case static_cast<uint64_t>(V1_0::GadgetFunction::ACCESSORY):
            ret = setVidPid("0x18d1", "0x2d00");
            break;
        case V1_0::GadgetFunction::ADB | V1_0::GadgetFunction::ACCESSORY:
            ret = setVidPid("0x18d1", "0x2d01");
            break;
        case static_cast<uint64_t>(V1_0::GadgetFunction::AUDIO_SOURCE):
            ret = setVidPid("0x18d1", "0x2d02");
            break;
        case V1_0::GadgetFunction::ADB | V1_0::GadgetFunction::AUDIO_SOURCE:
            ret = setVidPid("0x18d1", "0x2d03");
            break;
        case V1_0::GadgetFunction::ACCESSORY | V1_0::GadgetFunction::AUDIO_SOURCE:
            ret = setVidPid("0x18d1", "0x2d04");
            break;
        case V1_0::GadgetFunction::ADB | V1_0::GadgetFunction::ACCESSORY |
                V1_0::GadgetFunction::AUDIO_SOURCE:
            ret = setVidPid("0x18d1", "0x2d05");
            break;
        default:
            ALOGE("Combination not supported");
            ret = V1_0::Status::CONFIGURATION_NOT_SUPPORTED;
    }
    return ret;
}
V1_0::Status UsbGadget::setupFunctions(uint64_t functions,
                                       const sp<V1_0::IUsbGadgetCallback>& callback,
                                       uint64_t timeout) {
    int i = 1;
    if (addGenericAndroidFunctions(functions, &i) !=
        V1_0::Status::SUCCESS)
        return V1_0::Status::ERROR;

    if (!WriteStringToFile(kGadgetName, PULLUP_PATH)) return V1_0::Status::ERROR;
    mCurrentUsbFunctionsApplied = true;
    if (callback) callback->setCurrentUsbFunctionsCb(functions, V1_0::Status::SUCCESS);
    return V1_0::Status::SUCCESS;
}

Return<void> UsbGadget::setCurrentUsbFunctions(uint64_t functions,
                                               const sp<V1_0::IUsbGadgetCallback>& callback,
                                               uint64_t timeout) {
    std::unique_lock<std::mutex> lk(mLockSetCurrentFunction);
    mCurrentUsbFunctions = functions;
    mCurrentUsbFunctionsApplied = false;
    // Unlink the gadget
    V1_0::Status status = tearDownGadget();
    if (status != V1_0::Status::SUCCESS) {
        goto error;
    }
    ALOGI("Returned from tearDown gadget");
    // Leave the gadget pulled down to give time for the host to sense disconnect.
    usleep(kDisconnectWaitUs);
    if (functions == static_cast<uint64_t>(V1_0::GadgetFunction::NONE)) {
        if (callback == NULL) return Void();
        Return<void> ret = callback->setCurrentUsbFunctionsCb(functions, V1_0::Status::SUCCESS);
        if (!ret.isOk())
            ALOGE("Error while calling setCurrentUsbFunctionsCb %s", ret.description().c_str());
        return Void();
    }
    status = validateAndSetVidPid(functions);
    if (status != V1_0::Status::SUCCESS) {
        goto error;
    }
    status = setupFunctions(functions, callback, timeout);
    if (status != V1_0::Status::SUCCESS) {
        goto error;
    }
    ALOGI("Usb Gadget setcurrent functions called successfully");
    return Void();
error:
    ALOGI("Usb Gadget setcurrent functions failed");
    if (callback == NULL) return Void();
    Return<void> ret = callback->setCurrentUsbFunctionsCb(functions, status);
    if (!ret.isOk())
        ALOGE("Error while calling setCurrentUsbFunctionsCb %s", ret.description().c_str());
    return Void();
}
}  // namespace implementation
}  // namespace V1_1
}  // namespace gadget
}  // namespace usb
}  // namespace hardware
}  // namespace android