/*

 * COPYRIGHT (c) International Business Machines Corp. 2005-2017

 *

 * This program is provided under the terms of the Common Public License,

 * version 1.0 (CPL-1.0). Any use, reproduction or distribution for this

 * software constitutes recipient's acceptance of CPL-1.0 terms which can be

 * found in the file LICENSE file or at

 * https://opensource.org/licenses/cpl1.0.php

 */

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <memory.h>

#include <dlfcn.h>

#include "pkcs11types.h"

#include "regress.h"

#include "common.c"

int do_SetPIN(CK_FUNCTION_LIST * funcs, CK_SLOT_ID slot_id,

              CK_USER_TYPE userType, char *old, char *new)

{

    CK_RV rc;

    CK_SESSION_HANDLE session;

    CK_FLAGS flags = CKF_SERIAL_SESSION;

    flags |= CKF_RW_SESSION;

    rc = funcs->C_OpenSession(slot_id, flags, NULL, NULL, &session);

    if (rc != CKR_OK) {

        show_error("C_OpenSession", rc);

        return rc;

    }

    rc = funcs->C_Login(session, userType, (CK_CHAR_PTR) old, strlen(old));

    if (rc != CKR_OK) {

        show_error("C_Login", rc);

        return rc;

    }

    printf("Logged in successfully, calling C_SetPIN...\n");

    rc = funcs->C_SetPIN(session, (CK_CHAR_PTR) old, strlen(old),

                         (CK_CHAR_PTR) new, strlen(new));

    if (rc != CKR_OK) {

        show_error("C_SetPIN", rc);

        funcs->C_Logout(session);

        funcs->C_CloseSession(session);

        return rc;

    } else {

        printf("Success.\n");

    }

    rc = funcs->C_Logout(session);

    if (rc != CKR_OK) {

        show_error("C_Logout", rc);

        return rc;

    }

    printf("Logged out.\n");

    rc = funcs->C_CloseSession(session);

    if (rc != CKR_OK) {

        show_error("C_CloseSession", rc);

        return rc;

    }

    return rc;

}

void set_pin_usage(char *argv0)

{

    printf("usage:  %s [-slot <num>] [-h] [-user|-so] -old pass -new pass\n\n",

           argv0);

    printf("By default, Slot #%lu is used, as user\n\n", SLOT_ID);

    exit(-1);

}

int main(int argc, char **argv)

{

    CK_C_INITIALIZE_ARGS cinit_args;

    CK_USER_TYPE userType = CKU_USER;

    CK_RV rc = 0;

    CK_SLOT_ID slot_id = 0;

    char *old = NULL, *new = NULL;

    int i;

    for (i = 1; i < argc; i++) {

        if (strcmp(argv[i], "-old") == 0) {

            ++i;

            old = argv[i];

        } else if (strcmp(argv[i], "-new") == 0) {

            ++i;

            new = argv[i];

        } else if (strcmp(argv[i], "-slot") == 0) {

            ++i;

            slot_id = atoi(argv[i]);

        } else if (strcmp(argv[i], "-so") == 0) {

            userType = CKU_SO;

        } else if (strcmp(argv[i], "-user") == 0) {

            continue;

        } else {

            set_pin_usage(argv[0]);

        }

    }

    if (!old || !new)

        set_pin_usage(argv[0]);

    if (slot_id != SLOT_ID)

        printf("Using user specified slot %lu.\n", slot_id);

    rc = do_GetFunctionList();

    if (funcs == NULL)

        return -1;

    memset(&cinit_args, 0, sizeof(cinit_args));

    cinit_args.flags = CKF_OS_LOCKING_OK;

    rc = funcs->C_Initialize(&cinit_args);

    if (rc != CKR_OK) {

        show_error("C_Initialize", rc);

        return -1;

    }

    rc = do_SetPIN(funcs, slot_id, userType, old, new);

    funcs->C_Finalize(NULL);

    return rc;

}