/*

 * Copyright (c) 2022 Yubico AB. All rights reserved.

 * Use of this source code is governed by a BSD-style

 * license that can be found in the LICENSE file.

 * SPDX-License-Identifier: BSD-2-Clause

 */

#include <assert.h>

#include <stdint.h>

#include <stdlib.h>

#include <string.h>

#include <stdio.h>

#include <winscard.h>

#include "mutator_aux.h"

static const struct blob *reader_list;

static int (*xread)(void *, u_char *, size_t, int);

static int (*xwrite)(void *, const u_char *, size_t);

static void (*xconsume)(const void *, size_t);

LONG __wrap_SCardEstablishContext(DWORD, LPCVOID, LPCVOID, LPSCARDCONTEXT);

LONG __wrap_SCardListReaders(SCARDCONTEXT, LPCSTR, LPSTR, LPDWORD);

LONG __wrap_SCardReleaseContext(SCARDCONTEXT);

LONG __wrap_SCardConnect(SCARDCONTEXT, LPCSTR, DWORD, DWORD, LPSCARDHANDLE,

    LPDWORD);

LONG __wrap_SCardDisconnect(SCARDHANDLE, DWORD);

LONG __wrap_SCardTransmit(SCARDHANDLE, const SCARD_IO_REQUEST *, LPCBYTE,

    DWORD, SCARD_IO_REQUEST *, LPBYTE, LPDWORD);

LONG

__wrap_SCardEstablishContext(DWORD dwScope, LPCVOID pvReserved1,

    LPCVOID pvReserved2, LPSCARDCONTEXT phContext)

{

        assert(dwScope == SCARD_SCOPE_SYSTEM);

        assert(pvReserved1 == NULL);

        assert(pvReserved2 == NULL);

        *phContext = 1;

        if (uniform_random(400) < 1)

                return SCARD_E_NO_SERVICE;

        if (uniform_random(400) < 1)

                return SCARD_E_NO_SMARTCARD;

        if (uniform_random(400) < 1)

                return SCARD_E_NO_MEMORY;

        if (uniform_random(400) < 1)

                *phContext = 0;

        return SCARD_S_SUCCESS;

}

LONG

__wrap_SCardListReaders(SCARDCONTEXT hContext, LPCSTR mszGroups,

    LPSTR mszReaders, LPDWORD pcchReaders)

{

        assert(hContext == 1);

        assert(mszGroups == NULL);

        assert(mszReaders != NULL);

        assert(pcchReaders != 0);

        if (reader_list == NULL || uniform_random(400) < 1)

                return SCARD_E_NO_READERS_AVAILABLE;

        if (uniform_random(400) < 1)

                return SCARD_E_NO_MEMORY;

        memcpy(mszReaders, reader_list->body, reader_list->len > *pcchReaders ?

            *pcchReaders : reader_list->len);

        *pcchReaders = (DWORD)reader_list->len; /* on purpose */

        return SCARD_S_SUCCESS;

}

LONG

__wrap_SCardReleaseContext(SCARDCONTEXT hContext)

{

        assert(hContext == 1);

        return SCARD_S_SUCCESS;

}

LONG

__wrap_SCardConnect(SCARDCONTEXT hContext, LPCSTR szReader, DWORD dwShareMode,

    DWORD dwPreferredProtocols, LPSCARDHANDLE phCard, LPDWORD pdwActiveProtocol)

{

        uint32_t r;

        assert(hContext == 1);

        xconsume(szReader, strlen(szReader) + 1);

        assert(dwShareMode == SCARD_SHARE_SHARED);

        assert(dwPreferredProtocols == SCARD_PROTOCOL_ANY);

        assert(phCard != NULL);

        assert(pdwActiveProtocol != NULL);

        if ((r = uniform_random(400)) < 1)

                return SCARD_E_UNEXPECTED;

        *phCard = 1;

        *pdwActiveProtocol = (r & 1) ? SCARD_PROTOCOL_T0 : SCARD_PROTOCOL_T1;

        if (uniform_random(400) < 1)

                *pdwActiveProtocol = SCARD_PROTOCOL_RAW;

        return SCARD_S_SUCCESS;

}

LONG

__wrap_SCardDisconnect(SCARDHANDLE hCard, DWORD dwDisposition)

{

        assert(hCard == 1);

        assert(dwDisposition == SCARD_LEAVE_CARD);

        return SCARD_S_SUCCESS;

}

extern void consume(const void *body, size_t len);

LONG

__wrap_SCardTransmit(SCARDHANDLE hCard, const SCARD_IO_REQUEST *pioSendPci,

    LPCBYTE pbSendBuffer, DWORD cbSendLength, SCARD_IO_REQUEST *pioRecvPci,

    LPBYTE pbRecvBuffer, LPDWORD pcbRecvLength)

{

        void *ioh = (void *)NFC_DEV_HANDLE;

        int n;

        assert(hCard == 1);

        xconsume(pioSendPci, sizeof(*pioSendPci));

        xwrite(ioh, pbSendBuffer, cbSendLength);

        assert(pioRecvPci == NULL);

        if (uniform_random(400) < 1 ||

            (n = xread(ioh, pbRecvBuffer, *pcbRecvLength, -1)) == -1)

                return SCARD_E_UNEXPECTED;

        *pcbRecvLength = (DWORD)n;

        return SCARD_S_SUCCESS;

}

void

set_pcsc_parameters(const struct blob *reader_list_ptr)

{

        reader_list = reader_list_ptr;

}

void

set_pcsc_io_functions(int (*read_f)(void *, u_char *, size_t, int),

    int (*write_f)(void *, const u_char *, size_t),

    void (*consume_f)(const void *, size_t))

{

        xread = read_f;

        xwrite = write_f;

        xconsume = consume_f;

}