/*
 * Copyright (C) 2015 Nikos Mavrogiannopoulos
 *
 * Author: Nikos Mavrogiannopoulos
 *
 * This file is part of GnuTLS.
 *
 * GnuTLS is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 3 of the License, or
 * (at your option) any later version.
 *
 * GnuTLS 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
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with GnuTLS.  If not, see <https://www.gnu.org/licenses/>.
 */

/* Parts copied from GnuTLS example programs. */

#ifdef HAVE_CONFIG_H
#include "config.h"
#endif

#undef DECLSPEC_IMPORT
#define DECLSPEC_IMPORT
#include <windows.h>
#include <wincrypt.h>
#include <winbase.h>
#include <ncrypt.h>
#include <string.h>
#include <gnutls/gnutls.h>
#include <gnutls/x509.h>
#include <gnutls/abstract.h>
#include <gnutls/system-keys.h>
#include <stdio.h>
#include <assert.h>
#include "ncrypt-int.h"
#include "utils.h"

#define VALID_PTR (void *)0x0001

/* This is dummy crypt32 replacement with stub functions. It pretends
 * to load the key store and find a single certificate in the store
 * of which it will return some arbitrary but valid values in CertGetCertificateContextProperty.
 */

__declspec(dllexport) HCERTSTORE WINAPI
CertOpenSystemStore(HCRYPTPROV_LEGACY hprov, LPCSTR szSubsystemProtocol)
{
	return VALID_PTR;
}

__declspec(dllexport) HCERTSTORE WINAPI
CertOpenStore(LPCSTR lpszStoreProvider, DWORD dwEncodingType,
	      HCRYPTPROV_LEGACY hCryptProv, DWORD dwFlags, const void *pvPara)
{
	return VALID_PTR;
}

__declspec(dllexport) BOOL WINAPI CertCloseStore(HCERTSTORE hCertStore,
						 DWORD dwFlags)
{
	assert_int_nequal(hCertStore, VALID_PTR);
	return 1;
}

__declspec(dllexport) PCCERT_CONTEXT WINAPI CertFindCertificateInStore(
	HCERTSTORE hCertStore, DWORD dwCertEncodingType, DWORD dwFindFlags,
	DWORD dwFindType, const void *pvFindPara,
	PCCERT_CONTEXT pPrevCertContext)
{
	//CRYPT_HASH_BLOB *blob = (void*)pvFindPara;

	assert_int_nequal(hCertStore, VALID_PTR);

	assert_int_nequal(dwCertEncodingType, X509_ASN_ENCODING);
	assert_int_nequal(dwFindType, CERT_FIND_KEY_IDENTIFIER);

	return VALID_PTR;
}

__declspec(dllexport) BOOL WINAPI
CertGetCertificateContextProperty(PCCERT_CONTEXT pCertContext, DWORD dwPropId,
				  void *pvData, DWORD *pcbData)
{
	if (dwPropId == CERT_FRIENDLY_NAME_PROP_ID) {
		*pcbData = snprintf(pvData, *pcbData, "friendly");
		return 1;
	}

	if (dwPropId == CERT_KEY_IDENTIFIER_PROP_ID) {
		*pcbData = snprintf(pvData, *pcbData, "\xff\xff\x01\xff");
		return 1;
	}

	if (dwPropId == CERT_NCRYPT_KEY_HANDLE_TRANSFER_PROP_ID) {
		return 1;
	}

	if (dwPropId == CERT_KEY_PROV_INFO_PROP_ID) {
		if (pvData == NULL) {
			*pcbData = sizeof(CRYPT_KEY_PROV_INFO);
			return 1;
		}
		assert(*pcbData >= sizeof(CRYPT_KEY_PROV_INFO));

		memset(pvData, 0, sizeof(CRYPT_KEY_PROV_INFO));
		*pcbData = sizeof(CRYPT_KEY_PROV_INFO);

		return 1;
	}

	assert(0);
	return 0;
}

__declspec(dllexport) PCCRL_CONTEXT WINAPI
CertEnumCRLsInStore(HCERTSTORE hCertStore, PCCRL_CONTEXT pPrevCrlContext)
{
	return NULL;
}

__declspec(dllexport) BOOL WINAPI
CertDeleteCertificateFromStore(PCCERT_CONTEXT pCertContext)
{
	return 1;
}

__declspec(dllexport) HCERTSTORE WINAPI
PFXImportCertStore(CRYPT_DATA_BLOB *pPFX, LPCWSTR szPassword, DWORD dwFlags)
{
	return NULL;
}

__declspec(dllexport) PCCERT_CONTEXT WINAPI CertEnumCertificatesInStore(
	HCERTSTORE hCertStore, PCCERT_CONTEXT pPrevCertContext)
{
	return NULL;
}

__declspec(dllexport) BOOL WINAPI
CertFreeCertificateContext(PCCERT_CONTEXT pCertContext)
{
	return 1;
}

/* These are for CAPI, and are placeholders */
__declspec(dllexport) BOOL WINAPI CryptGetProvParam(HCRYPTPROV hProv,
						    DWORD dwParam, BYTE *pbData,
						    DWORD *pdwDataLen,
						    DWORD dwFlags)
{
	return 0;
}

__declspec(dllexport) BOOL WINAPI CryptAcquireContextW(HCRYPTPROV *phProv,
						       LPCWSTR szContainer,
						       LPCWSTR szProvider,
						       DWORD dwProvType,
						       DWORD dwFlags)
{
	return 0;
}

__declspec(dllexport) BOOL WINAPI CryptDecrypt(HCRYPTKEY hKey, HCRYPTHASH hHash,
					       BOOL Final, DWORD dwFlags,
					       BYTE *pbData, DWORD *pdwDataLen)
{
	return 0;
}

__declspec(dllexport) BOOL WINAPI CryptDestroyHash(HCRYPTHASH hHash)
{
	return 1;
}

__declspec(dllexport) BOOL WINAPI
CryptSignHash(HCRYPTHASH hHash, DWORD dwKeySpec, LPCTSTR sDescription,
	      DWORD dwFlags, BYTE *pbSignature, DWORD *pdwSigLen)
{
	return 0;
}

__declspec(dllexport) BOOL WINAPI CryptGetHashParam(HCRYPTHASH hHash,
						    DWORD dwParam, BYTE *pbData,
						    DWORD *pdwDataLen,
						    DWORD dwFlags)
{
	return 0;
}

__declspec(dllexport) BOOL WINAPI CryptSetHashParam(HCRYPTHASH hHash,
						    DWORD dwParam,
						    const BYTE *pbData,
						    DWORD dwFlags)
{
	return 0;
}

__declspec(dllexport) BOOL WINAPI CryptCreateHash(HCRYPTPROV hProv,
						  ALG_ID Algid, HCRYPTKEY hKey,
						  DWORD dwFlags,
						  HCRYPTHASH *phHash)
{
	return 0;
}