/*
 * Copyright (C) 2011-2018 Intel Corporation. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 *   * Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *   * Redistributions in binary form must reproduce the above copyright
 *     notice, this list of conditions and the following disclaimer in
 *     the documentation and/or other materials provided with the
 *     distribution.
 *   * Neither the name of Intel Corporation nor the names of its
 *     contributors may be used to endorse or promote products derived
 *     from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 */

/** 
* File: 
*     cse_provision_tool.cpp
*Description: 
*     provision CSE Fw
* 
*/

#include <stdio.h>
#include <stdint.h>
#include <dlfcn.h>
#include <stdarg.h>
#include <unistd.h>
#include <sys/types.h>

static char ICLS_LIB_NAME[] = "libiclsclient.so";
static char ICLS_INIT_FUNC_NAME[] = "iclsInit";
typedef  uint32_t(*iclsInit_t)(const void*);
#define STATUS_OK  (0)
#define PARA_ERROR (1)
#define ICLS_MISS  (2)
#define ICLS_ERROR (3)

static char USAGE_STRING[] = \
    "Usage: cse_provision_tool\n"\
    "Invoke iclsclient to provision CSE Fw. Root privilege is required.\n";

static char MISSING_STRING[] =  \
    "libiclsclient.so or iclsInit() cannot be found.\n"\
    "Trusted platform service is unavailable. Refer to README for details.\n";

static char ERR_STRING[] =  \
    "iclsInit() returned error.\n"\
    "Trusted platform service is unavailable. "\
    "Check log in /opt/Intel/iclsClient/log/iclsClient.log.\n";

uint32_t upse_iclsInit()
{
    uint32_t status = STATUS_OK;
    void *hmodule = NULL;
    do{
        //For this to work iclsClient needs to be installed in the system
        hmodule = dlopen(ICLS_LIB_NAME, RTLD_LAZY);
        if(NULL == hmodule)
        {
            status = ICLS_MISS;
            break;
        }

        //clear exist error code
        dlerror();

        iclsInit_t iclsInit = (iclsInit_t)dlsym(hmodule, ICLS_INIT_FUNC_NAME);
        if ( NULL != dlerror() || NULL == iclsInit)
        {
            status = ICLS_MISS;
            break;
        }
        //If you get error, check /opt/Intel/iclsClient/log/iclsClient.log
        uint32_t status_provision = iclsInit(0);
        if (status_provision != STATUS_OK)
        {
            status = ICLS_ERROR;
            break;
        }
    } while(0);

    if(hmodule)
        dlclose(hmodule);
    return status;
}


int main(int argc, char* argv[])
{
    (void)argv;
    if(argc != 1 || getuid() != 0)
    {
        fprintf(stderr, "%s", USAGE_STRING);
        return PARA_ERROR;
    }
    uint32_t status_provision = upse_iclsInit();
    switch (status_provision)
    {
        case STATUS_OK:
            break;
        case ICLS_MISS:
            fprintf(stderr, "%s", MISSING_STRING);
            break;
        case ICLS_ERROR:
            fprintf(stderr, "%s", ERR_STRING);
            break;
    }
    return status_provision;
}