/*
 * Copyright 2022 Hans Leidekker for CodeWeavers
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
 */

#if 0
#pragma makedep unix
#endif

#include <stdarg.h>
#include <stdlib.h>
#include "ntstatus.h"
#define WIN32_NO_STATUS
#include "windef.h"
#include "winternl.h"
#include "winbase.h"

#include "wine/unixlib.h"
#include "unixlib.h"

#ifdef __APPLE__
LONG SCardEstablishContext( UINT32, const void *, const void *, UINT32 * );
LONG SCardReleaseContext( UINT32 );
LONG SCardIsValidContext( UINT32 );
LONG SCardGetStatusChange( UINT32, UINT32, void *, UINT32 );
LONG SCardCancel( UINT32 );
LONG SCardListReaders( UINT32, const char *, char *, UINT32 * );
LONG SCardListReaderGroups( UINT32, char *, UINT32 * );
LONG SCardConnect( UINT32, const char *, UINT32, UINT32, UINT32 *, UINT32 * );
LONG SCardStatus( UINT32, char *, UINT32 *, UINT32 *, UINT32 *, BYTE *, UINT32 * );
LONG SCardDisconnect( UINT32, UINT32 );
LONG SCardReconnect( UINT32, UINT32, UINT32, UINT32, UINT32 * );
LONG SCardBeginTransaction( UINT32 );
LONG SCardEndTransaction( UINT32, UINT32 );
LONG SCardTransmit( UINT32, const void *, const BYTE *, UINT32, void *, BYTE *, UINT32 * );
LONG SCardControl132( UINT32, UINT32, const void *, UINT32, void *, UINT32, UINT32 * );
LONG SCardGetAttrib( UINT32, UINT32, BYTE *, UINT32 * );
LONG SCardSetAttrib( UINT32, UINT32, const BYTE *, UINT32 );
#else
LONG SCardEstablishContext( UINT64, const void *, const void *, UINT64 * );
LONG SCardReleaseContext( UINT64 );
LONG SCardIsValidContext( UINT64 );
LONG SCardGetStatusChange( UINT64, UINT64, void *, UINT64 );
LONG SCardCancel( UINT64 );
LONG SCardListReaders( UINT64, const char *, char *, UINT64 * );
LONG SCardListReaderGroups( UINT64, char *, UINT64 * );
LONG SCardConnect( UINT64, const char *, UINT64, UINT64, UINT64 *, UINT64 * );
LONG SCardStatus( UINT64, char *, UINT64 *, UINT64 *, UINT64 *, BYTE *, UINT64 * );
LONG SCardDisconnect( UINT64, UINT64 );
LONG SCardReconnect( UINT64, UINT64, UINT64, UINT64, UINT64 * );
LONG SCardBeginTransaction( UINT64 );
LONG SCardEndTransaction( UINT64, UINT64 );
LONG SCardTransmit( UINT64, const void *, const BYTE *, UINT64, void *, BYTE *, UINT64 * );
LONG SCardControl( UINT64, UINT64, const void *, UINT64, void *, UINT64, UINT64 * );
LONG SCardGetAttrib( UINT64, UINT64, BYTE *, UINT64 * );
LONG SCardSetAttrib( UINT64, UINT64, const BYTE *, UINT64 );
#endif

static NTSTATUS scard_establish_context( void *args )
{
    struct scard_establish_context_params *params = args;
#ifdef __APPLE__
    NTSTATUS status;
    UINT32 handle;
    if (!(status = SCardEstablishContext( params->scope, NULL, NULL, &handle ))) *params->handle = handle;
    return status;
#else
    return SCardEstablishContext( params->scope, NULL, NULL, params->handle );
#endif
}

static NTSTATUS scard_release_context( void *args )
{
    struct scard_release_context_params *params = args;
    return SCardReleaseContext( params->handle );
}

static NTSTATUS scard_is_valid_context( void *args )
{
    struct scard_is_valid_context_params *params = args;
    return SCardIsValidContext( params->handle );
}

#ifdef __APPLE__
struct reader_state_macos
{
    UINT64 reader;
    UINT64 userdata;
    UINT32 current_state;
    UINT32 event_state;
    UINT32 atr_size;
    unsigned char atr[MAX_ATR_SIZE];
};

static struct reader_state_macos *map_states_in( const struct reader_state *state, unsigned int count )
{
    struct reader_state_macos *ret;
    unsigned int i;

    if (!(ret = malloc( sizeof(*ret) * count ))) return NULL;
    for (i = 0; i < count; i++)
    {
        ret[i].reader = state[i].reader;
        ret[i].userdata = state[i].userdata;
        ret[i].current_state = state[i].current_state;
        ret[i].event_state = state[i].event_state;
        memcpy( ret[i].atr, state[i].atr, state[i].atr_size );
        ret[i].atr_size = state[i].atr_size;
    }
    return ret;
}

static void map_states_out( const struct reader_state_macos *state_macos, struct reader_state *state, unsigned int count )
{
    unsigned int i;
    for (i = 0; i < count; i++)
    {
        state[i].current_state = state_macos[i].current_state;
        state[i].event_state = state_macos[i].event_state;
        memcpy( state[i].atr, state_macos[i].atr, state_macos[i].atr_size );
        state[i].atr_size = state_macos[i].atr_size;
    }
}
#endif

static NTSTATUS scard_get_status_change( void *args )
{
    struct scard_get_status_change_params *params = args;
#ifdef __APPLE__
    NTSTATUS status;
    struct reader_state_macos *states = map_states_in( params->states, params->count );

    if (!states) return STATUS_NO_MEMORY;
    if (!(status = SCardGetStatusChange( params->handle, params->timeout, states, params->count )))
        map_states_out( states, params->states, params->count );
    free( states );
    return status;
#else
    return SCardGetStatusChange( params->handle, params->timeout, params->states, params->count );
#endif
}

static NTSTATUS scard_cancel( void *args )
{
    struct scard_cancel_params *params = args;
    return SCardCancel( params->handle );
}

static NTSTATUS scard_list_readers( void *args )
{
    struct scard_list_readers_params *params = args;
#ifdef __APPLE__
    NTSTATUS status;
    UINT32 len = *params->readers_len;
    if (!(status = SCardListReaders( params->handle, params->groups, params->readers, &len ))) *params->readers_len = len;
    return status;
#else
    return SCardListReaders( params->handle, params->groups, params->readers, params->readers_len );
#endif
}

static NTSTATUS scard_list_reader_groups( void *args )
{
    struct scard_list_reader_groups_params *params = args;
#ifdef __APPLE__
    NTSTATUS status;
    UINT32 len = *params->groups_len;
    if (!(status = SCardListReaderGroups( params->handle, params->groups, &len ))) *params->groups_len = len;
    return status;
#else
    return SCardListReaderGroups( params->handle, params->groups, params->groups_len );
#endif
}

static NTSTATUS scard_connect( void *args )
{
    struct scard_connect_params *params = args;
#ifdef __APPLE__
    NTSTATUS status;
    UINT32 handle, protocol;
    if (!(status = SCardConnect( params->context_handle, params->reader, params->share_mode,
                                 params->preferred_protocols, &handle, &protocol )))
    {
        *params->connect_handle = handle;
        *params->protocol = protocol;
    }
    return status;
#else
    return SCardConnect( params->context_handle, params->reader, params->share_mode, params->preferred_protocols,
                         params->connect_handle, params->protocol );
#endif
}

static NTSTATUS scard_status( void *args )
{
    struct scard_status_params *params = args;
#ifdef __APPLE__
    NTSTATUS status;
    UINT32 names_len = params->names_len ? *params->names_len : 0;
    UINT32 atr_len = params->atr_len ? *params->atr_len : 0;
    UINT32 state, protocol;

    if (!(status = SCardStatus( params->handle, params->names, params->names_len ? &names_len : NULL,
                                params->state ? &state : NULL, params->protocol ? &protocol : NULL, params->atr,
                                params->atr_len ? &atr_len : NULL )))
    {
        if (params->names_len) *params->names_len = names_len;
        if (params->state) *params->state = state;
        if (params->protocol) *params->protocol = protocol;
        if (params->atr_len) *params->atr_len = atr_len;
    }
    return status;
#else
    return SCardStatus( params->handle, params->names, params->names_len, params->state, params->protocol,
                        params->atr, params->atr_len );
#endif
}

static NTSTATUS scard_reconnect( void *args )
{
    struct scard_reconnect_params *params = args;
#ifdef __APPLE__
    NTSTATUS status;
    UINT32 protocol;
    if (!(status = SCardReconnect( params->handle, params->share_mode, params->preferred_protocols,
                                   params->initialization, &protocol ))) *params->protocol = protocol;
    return status;
#else
    return SCardReconnect( params->handle, params->share_mode, params->preferred_protocols, params->initialization,
                           params->protocol );
#endif
}

static NTSTATUS scard_disconnect( void *args )
{
    struct scard_disconnect_params *params = args;
    return SCardDisconnect( params->handle, params->disposition );
}

static NTSTATUS scard_begin_transaction( void *args )
{
    struct scard_begin_transaction_params *params = args;
    return SCardBeginTransaction( params->handle );
}

static NTSTATUS scard_end_transaction( void *args )
{
    struct scard_end_transaction_params *params = args;
    return SCardEndTransaction( params->handle, params->disposition );
}

static NTSTATUS scard_transmit( void *args )
{
    struct scard_transmit_params *params = args;
#ifdef __APPLE__
    NTSTATUS status;
    UINT32 len = *params->recv_buflen;
    struct
    {
        UINT32 protocol;
        UINT32 pci_len;
    } send, recv;

    send.protocol = params->send->protocol;
    send.pci_len = params->send->pci_len;
    if (params->recv)
    {
        recv.protocol = params->recv->protocol;
        recv.pci_len = params->recv->pci_len;
    }
    if (!(status = SCardTransmit( params->handle, &send, params->send_buf, params->send_buflen, &recv,
                                  params->recv_buf, &len )))
    {
        if (params->recv)
        {
            params->recv->protocol = recv.protocol;
            params->recv->pci_len = recv.pci_len;
        }
        *params->recv_buflen = len;
    }
    return status;
#else
    return SCardTransmit( params->handle, params->send, params->send_buf, params->send_buflen, params->recv,
                          params->recv_buf, params->recv_buflen );
#endif
}

static NTSTATUS scard_control( void *args )
{
    struct scard_control_params *params = args;
#ifdef __APPLE__
    NTSTATUS status;
    UINT32 len;
    if (!(status = SCardControl132( params->handle, params->code, params->in_buf, params->in_buflen, params->out_buf,
                                    params->out_buflen, &len )))
        *params->ret_len = len;
    return status;
#else
    return SCardControl( params->handle, params->code, params->in_buf, params->in_buflen, params->out_buf,
                         params->out_buflen, params->ret_len );
#endif
}

static NTSTATUS scard_get_attrib( void *args )
{
    struct scard_get_attrib_params *params = args;
#ifdef __APPLE__
    NTSTATUS status;
    UINT32 len = *params->attr_len;
    if (!(status = SCardGetAttrib( params->handle, params->id, params->attr, &len ))) *params->attr_len = len;
    return status;
#else
    return SCardGetAttrib( params->handle, params->id, params->attr, params->attr_len );
#endif
}

static NTSTATUS scard_set_attrib( void *args )
{
    struct scard_set_attrib_params *params = args;
    return SCardSetAttrib( params->handle, params->id, params->attr, params->attr_len );
}

const unixlib_entry_t __wine_unix_call_funcs[] =
{
    scard_establish_context,
    scard_release_context,
    scard_is_valid_context,
    scard_get_status_change,
    scard_cancel,
    scard_list_readers,
    scard_list_reader_groups,
    scard_connect,
    scard_status,
    scard_reconnect,
    scard_disconnect,
    scard_begin_transaction,
    scard_end_transaction,
    scard_transmit,
    scard_control,
    scard_get_attrib,
    scard_set_attrib,
};

#ifdef _WIN64

typedef ULONG PTR32;

static NTSTATUS wow64_scard_establish_context( void *args )
{
    struct
    {
        UINT64 scope;
        PTR32 handle;
    } const *params32 = args;

    struct scard_establish_context_params params =
    {
        params32->scope,
        ULongToPtr(params32->handle)
    };
    return scard_establish_context( &params );
}

static NTSTATUS wow64_scard_release_context( void *args )
{
    struct
    {
        UINT64 handle;
    } const *params32 = args;

    struct scard_release_context_params params =
    {
        params32->handle
    };
    return scard_release_context( &params );
}

static NTSTATUS wow64_scard_is_valid_context( void *args )
{
    struct
    {
        UINT64 handle;
    } const *params32 = args;

    struct scard_is_valid_context_params params =
    {
        params32->handle
    };
    return scard_is_valid_context( &params );
}

static NTSTATUS wow64_scard_get_status_change( void *args )
{
    struct
    {
        UINT64 handle;
        UINT64 timeout;
        PTR32 states;
        UINT64 count;
    } const *params32 = args;

    struct scard_get_status_change_params params =
    {
        params32->handle,
        params32->timeout,
        ULongToPtr(params32->states),
        params32->count,
    };
    return scard_get_status_change( &params );
}

static NTSTATUS wow64_scard_cancel( void *args )
{
    struct
    {
        UINT64 handle;
    } const *params32 = args;

    struct scard_cancel_params params =
    {
        params32->handle
    };
    return scard_cancel( &params );
}

static NTSTATUS wow64_scard_list_readers( void *args )
{
    struct
    {
        UINT64 handle;
        PTR32 groups;
        PTR32 readers;
        PTR32 readers_len;
    } *params32 = args;

    struct scard_list_readers_params params =
    {
        params32->handle,
        ULongToPtr(params32->groups),
        ULongToPtr(params32->readers),
        ULongToPtr(params32->readers_len)
    };
    return scard_list_readers( &params );
};

static NTSTATUS wow64_scard_list_reader_groups( void *args )
{
    struct
    {
        UINT64 handle;
        PTR32 groups;
        PTR32 groups_len;
    } *params32 = args;

    struct scard_list_reader_groups_params params =
    {
        params32->handle,
        ULongToPtr(params32->groups),
        ULongToPtr(params32->groups_len)
    };
    return scard_list_reader_groups( &params );
}

static NTSTATUS wow64_scard_connect( void *args )
{
    struct
    {
        UINT64 context_handle;
        PTR32 reader;
        UINT64 share_mode;
        UINT64 preferred_protocols;
        PTR32 connect_handle;
        PTR32 protocol;
    } *params32 = args;

    struct scard_connect_params params =
    {
        params32->context_handle,
        ULongToPtr(params32->reader),
        params32->share_mode,
        params32->preferred_protocols,
        ULongToPtr(params32->connect_handle),
        ULongToPtr(params32->protocol)
    };
    return scard_connect( &params );
}

static NTSTATUS wow64_scard_status( void *args )
{
    struct
    {
        UINT64 handle;
        PTR32 names;
        PTR32 names_len;
        PTR32 state;
        PTR32 protocol;
        PTR32 atr;
        PTR32 atr_len;
    } *params32 = args;

    struct scard_status_params params =
    {
        params32->handle,
        ULongToPtr(params32->names),
        ULongToPtr(params32->names_len),
        ULongToPtr(params32->state),
        ULongToPtr(params32->protocol),
        ULongToPtr(params32->atr),
        ULongToPtr(params32->atr_len)
    };
    return scard_status( &params );
};

static NTSTATUS wow64_scard_reconnect( void *args )
{
    struct
    {
        UINT64 handle;
        UINT64 share_mode;
        UINT64 preferred_protocols;
        UINT64 initialization;
        PTR32 protocol;
    } *params32 = args;

    struct scard_reconnect_params params =
    {
        params32->handle,
        params32->share_mode,
        params32->preferred_protocols,
        params32->initialization,
        ULongToPtr(params32->protocol)
    };
    return scard_reconnect( &params );
}

static NTSTATUS wow64_scard_disconnect( void *args )
{
    struct
    {
        UINT64 handle;
        UINT64 disposition;
    } *params32 = args;

    struct scard_disconnect_params params =
    {
        params32->handle,
        params32->disposition
    };
    return scard_disconnect( &params );
}

static NTSTATUS wow64_scard_begin_transaction( void *args )
{
    struct
    {
        UINT64 handle;
    } *params32 = args;

    struct scard_begin_transaction_params params =
    {
        params32->handle
    };
    return scard_begin_transaction( &params );
}

static NTSTATUS wow64_scard_end_transaction( void *args )
{
    struct
    {
        UINT64 handle;
        UINT64 disposition;
    } *params32 = args;

    struct scard_end_transaction_params params =
    {
        params32->handle,
        params32->disposition
    };
    return scard_end_transaction( &params );
}

static NTSTATUS wow64_scard_transmit( void *args )
{
    struct
    {
        UINT64 handle;
        PTR32 send;
        PTR32 send_buf;
        UINT64 send_buflen;
        PTR32 recv;
        PTR32 recv_buf;
        PTR32 recv_buflen;
    } *params32 = args;

    struct scard_transmit_params params =
    {
        params32->handle,
        ULongToPtr(params32->send),
        ULongToPtr(params32->send_buf),
        params32->send_buflen,
        ULongToPtr(params32->recv),
        ULongToPtr(params32->recv_buf),
        ULongToPtr(params32->recv_buflen)
    };
    return scard_transmit( &params );
};

static NTSTATUS wow64_scard_control( void *args )
{
    struct
    {
        UINT64 handle;
        UINT64 code;
        PTR32 in_buf;
        UINT64 in_buflen;
        PTR32 out_buf;
        UINT64 out_buflen;
        PTR32 ret_len;
    } *params32 = args;

    struct scard_control_params params =
    {
        params32->handle,
        params32->code,
        ULongToPtr(params32->in_buf),
        params32->in_buflen,
        ULongToPtr(params32->out_buf),
        params32->out_buflen,
        ULongToPtr(params32->ret_len)
    };
    return scard_control( &params );
};

static NTSTATUS wow64_scard_get_attrib( void *args )
{
    struct
    {
        UINT64 handle;
        UINT64 id;
        PTR32 attr;
        PTR32 attr_len;
    } *params32 = args;

    struct scard_get_attrib_params params =
    {
        params32->handle,
        params32->id,
        ULongToPtr(params32->attr),
        ULongToPtr(params32->attr_len)
    };
    return scard_get_attrib( &params );
}

static NTSTATUS wow64_scard_set_attrib( void *args )
{
    struct
    {
        UINT64 handle;
        UINT64 id;
        PTR32 attr;
        UINT64 attr_len;
    } *params32 = args;

    struct scard_set_attrib_params params =
    {
        params32->handle,
        params32->id,
        ULongToPtr(params32->attr),
        params32->attr_len
    };
    return scard_set_attrib( &params );
}

const unixlib_entry_t __wine_unix_call_wow64_funcs[] =
{
    wow64_scard_establish_context,
    wow64_scard_release_context,
    wow64_scard_is_valid_context,
    wow64_scard_get_status_change,
    wow64_scard_cancel,
    wow64_scard_list_readers,
    wow64_scard_list_reader_groups,
    wow64_scard_connect,
    wow64_scard_status,
    wow64_scard_reconnect,
    wow64_scard_disconnect,
    wow64_scard_begin_transaction,
    wow64_scard_end_transaction,
    wow64_scard_transmit,
    wow64_scard_control,
    wow64_scard_get_attrib,
    wow64_scard_set_attrib,
};

#endif  /* _WIN64 */