/src/FreeRDP/libfreerdp/core/smartcardlogon.c

Source (jump to first uncovered line)
/**
 * FreeRDP: A Remote Desktop Protocol Implementation
 * Logging in with smartcards
 *
 * Copyright 2022 David Fort <contact@hardening-consulting.com>
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
#include <string.h>

#include <winpr/error.h>
#include <winpr/ncrypt.h>
#include <winpr/string.h>
#include <winpr/wlog.h>
#include <winpr/crypto.h>
#include <winpr/path.h>

#include <freerdp/log.h>
#include <freerdp/freerdp.h>
#include <winpr/print.h>

#include <freerdp/utils/smartcardlogon.h>
#include <freerdp/crypto/crypto.h>

#include <openssl/obj_mac.h>

#define TAG FREERDP_TAG("smartcardlogon")

struct SmartcardKeyInfo_st
{
  char* certPath;
  char* keyPath;
};

static void delete_file(char* path)
{
  if (!path)
    return;

  /* Overwrite data in files before deletion */
  {
    FILE* fp = winpr_fopen(path, "r+");
    if (fp)
    {
      const char buffer[8192] = { 0 };
      INT64 size = 0;
      int rs = _fseeki64(fp, 0, SEEK_END);
      if (rs == 0)
        size = _ftelli64(fp);
      _fseeki64(fp, 0, SEEK_SET);

      for (INT64 x = 0; x < size; x += sizeof(buffer))
      {
        const size_t dnmemb = (size_t)(size - x);
        const size_t nmemb = MIN(sizeof(buffer), dnmemb);
        const size_t count = fwrite(buffer, nmemb, 1, fp);
        if (count != 1)
          break;
      }

      fclose(fp);
    }
  }

  winpr_DeleteFile(path);
  free(path);
}

static void smartcardKeyInfo_Free(SmartcardKeyInfo* key_info)
{
  if (!key_info)
    return;

  delete_file(key_info->certPath);
  delete_file(key_info->keyPath);

  free(key_info);
}

void smartcardCertInfo_Free(SmartcardCertInfo* scCert)
{
  if (!scCert)
    return;

  free(scCert->csp);
  free(scCert->reader);
  freerdp_certificate_free(scCert->certificate);
  free(scCert->pkinitArgs);
  free(scCert->keyName);
  free(scCert->containerName);
  free(scCert->upn);
  free(scCert->userHint);
  free(scCert->domainHint);
  free(scCert->subject);
  free(scCert->issuer);
  smartcardKeyInfo_Free(scCert->key_info);

  free(scCert);
}

void smartcardCertList_Free(SmartcardCertInfo** cert_list, size_t count)
{
  if (!cert_list)
    return;

  for (size_t i = 0; i < count; i++)
  {
    SmartcardCertInfo* cert = cert_list[i];
    smartcardCertInfo_Free(cert);
  }

  free(cert_list);
}

static BOOL add_cert_to_list(SmartcardCertInfo*** certInfoList, size_t* count,
                             SmartcardCertInfo* certInfo)
{
  size_t curCount = *count;
  SmartcardCertInfo** curInfoList = *certInfoList;

  /* Check if the certificate is already in the list */
  for (size_t i = 0; i < curCount; ++i)
  {
    if (_wcscmp(curInfoList[i]->containerName, certInfo->containerName) == 0)
    {
      smartcardCertInfo_Free(certInfo);
      return TRUE;
    }
  }

  curInfoList = realloc(curInfoList, sizeof(SmartcardCertInfo*) * (curCount + 1));
  if (!curInfoList)
  {
    WLog_ERR(TAG, "unable to reallocate certs");
    return FALSE;
  }

  curInfoList[curCount++] = certInfo;
  *certInfoList = curInfoList;
  *count = curCount;
  return TRUE;
}

static BOOL treat_sc_cert(SmartcardCertInfo* scCert)
{
  WINPR_ASSERT(scCert);

  scCert->upn = freerdp_certificate_get_upn(scCert->certificate);
  if (!scCert->upn)
  {
    WLog_DBG(TAG, "%s has no UPN, trying emailAddress", scCert->keyName);
    scCert->upn = freerdp_certificate_get_email(scCert->certificate);
  }

  if (scCert->upn)
  {
    size_t userLen = 0;
    const char* atPos = strchr(scCert->upn, '@');

    if (!atPos)
    {
      WLog_ERR(TAG, "invalid UPN, for key %s (no @)", scCert->keyName);
      return FALSE;
    }

    userLen = (size_t)(atPos - scCert->upn);
    scCert->userHint = malloc(userLen + 1);
    scCert->domainHint = _strdup(atPos + 1);

    if (!scCert->userHint || !scCert->domainHint)
    {
      WLog_ERR(TAG, "error allocating userHint or domainHint, for key %s", scCert->keyName);
      return FALSE;
    }

    memcpy(scCert->userHint, scCert->upn, userLen);
    scCert->userHint[userLen] = 0;
  }

  scCert->subject = freerdp_certificate_get_subject(scCert->certificate);
  scCert->issuer = freerdp_certificate_get_issuer(scCert->certificate);
  return TRUE;
}

static BOOL set_info_certificate(SmartcardCertInfo* cert, BYTE* certBytes, DWORD cbCertBytes,
                                 const char* userFilter, const char* domainFilter)
{
  if (!winpr_Digest(WINPR_MD_SHA1, certBytes, cbCertBytes, cert->sha1Hash,
                    sizeof(cert->sha1Hash)))
  {
    WLog_ERR(TAG, "unable to compute certificate sha1 for key %s", cert->keyName);
    return FALSE;
  }

  cert->certificate = freerdp_certificate_new_from_der(certBytes, cbCertBytes);
  if (!cert->certificate)
  {
    WLog_ERR(TAG, "unable to parse X509 certificate for key %s", cert->keyName);
    return FALSE;
  }

  if (!freerdp_certificate_check_eku(cert->certificate, NID_ms_smartcard_login))
  {
    WLog_DBG(TAG, "discarding certificate without Smartcard Login EKU for key %s",
             cert->keyName);
    return FALSE;
  }

  if (!treat_sc_cert(cert))
  {
    WLog_DBG(TAG, "error treating cert");
    return FALSE;
  }

  if (userFilter && (!cert->upn || (strcmp(cert->upn, userFilter) != 0)))
  {
    if (cert->userHint && strcmp(cert->userHint, userFilter) != 0)
    {
      WLog_DBG(TAG, "discarding non matching cert by user %s@%s", cert->userHint,
               cert->domainHint);
      return FALSE;
    }
  }

  if (domainFilter && cert->domainHint && strcmp(cert->domainHint, domainFilter) != 0)
  {
    WLog_DBG(TAG, "discarding non matching cert by domain(%s) %s@%s", domainFilter,
             cert->userHint, cert->domainHint);
    return FALSE;
  }

  return TRUE;
}

#ifndef _WIN32
static BOOL build_pkinit_args(NCRYPT_PROV_HANDLE provider, SmartcardCertInfo* scCert)
{
  /* pkinit args only under windows
   *    PKCS11:module_name=opensc-pkcs11.so
   */
  const char* pkModule = winpr_NCryptGetModulePath(provider);
  size_t size = 0;

  if (winpr_asprintf(&scCert->pkinitArgs, &size, "PKCS11:module_name=%s:slotid=%" PRIu16,
                     pkModule, (UINT16)scCert->slotId) <= 0)
    return FALSE;
  return TRUE;
}
#endif /* _WIN32 */

#ifdef _WIN32
static BOOL list_capi_provider_keys(const rdpSettings* settings, LPCWSTR csp, LPCWSTR scope,
                                    const char* userFilter, const char* domainFilter,
                                    SmartcardCertInfo*** pcerts, size_t* pcount)
{
  BOOL ret = FALSE;
  HCRYPTKEY hKey = 0;
  HCRYPTPROV hProvider = 0;
  SmartcardCertInfo* cert = NULL;
  BYTE* certBytes = NULL;
  CHAR* readerName = NULL;

  if (!CryptAcquireContextW(&hProvider, scope, csp, PROV_RSA_FULL, CRYPT_SILENT))
  {
    WLog_DBG(TAG, "Unable to acquire context: %d", GetLastError());
    goto out;
  }

  cert = calloc(1, sizeof(SmartcardCertInfo));
  if (!cert)
    goto out;

  cert->csp = _wcsdup(csp);
  if (!cert->csp)
    goto out;

  /* ====== retrieve key's reader ====== */
  DWORD dwDataLen = 0;
  if (!CryptGetProvParam(hProvider, PP_SMARTCARD_READER, NULL, &dwDataLen, 0))
  {
    WLog_DBG(TAG, "Unable to get provider param: %d", GetLastError());
    goto out;
  }

  readerName = malloc(dwDataLen);
  if (!readerName)
    goto out;

  if (!CryptGetProvParam(hProvider, PP_SMARTCARD_READER, readerName, &dwDataLen, 0))
  {
    WLog_DBG(TAG, "Unable to get reader name: %d", GetLastError());
    goto out;
  }

  cert->reader = ConvertUtf8ToWCharAlloc(readerName, NULL);
  if (!cert->reader)
    goto out;

  /* ====== retrieve key container name ====== */
  dwDataLen = 0;
  if (!CryptGetProvParam(hProvider, PP_CONTAINER, NULL, &dwDataLen, 0))
  {
    WLog_DBG(TAG, "Unable to get provider param: %d", GetLastError());
    goto out;
  }

  cert->keyName = malloc(dwDataLen);
  if (!cert->keyName)
    goto out;

  if (!CryptGetProvParam(hProvider, PP_CONTAINER, cert->keyName, &dwDataLen, 0))
  {
    WLog_DBG(TAG, "Unable to get container name: %d", GetLastError());
    goto out;
  }

  cert->containerName = ConvertUtf8ToWCharAlloc(cert->keyName, NULL);
  if (!cert->containerName)
    goto out;

  /* ========= retrieve the certificate ===============*/
  if (!CryptGetUserKey(hProvider, AT_KEYEXCHANGE, &hKey))
  {
    WLog_DBG(TAG, "Unable to get user key for %s: %d", cert->keyName, GetLastError());
    goto out;
  }

  dwDataLen = 0;
  if (!CryptGetKeyParam(hKey, KP_CERTIFICATE, NULL, &dwDataLen, 0))
  {
    WLog_DBG(TAG, "Unable to get key param for key %s: %d", cert->keyName, GetLastError());
    goto out;
  }

  certBytes = malloc(dwDataLen);
  if (!certBytes)
  {
    WLog_ERR(TAG, "unable to allocate %" PRIu32 " certBytes for key %s", dwDataLen,
             cert->keyName);
    goto out;
  }

  if (!CryptGetKeyParam(hKey, KP_CERTIFICATE, certBytes, &dwDataLen, 0))
  {
    WLog_ERR(TAG, "unable to retrieve certificate for key %s", cert->keyName);
    goto out;
  }

  if (!set_info_certificate(cert, certBytes, dwDataLen, userFilter, domainFilter))
    goto out;

  if (!add_cert_to_list(pcerts, pcount, cert))
    goto out;

  ret = TRUE;

out:
  free(readerName);
  free(certBytes);
  if (hKey)
    CryptDestroyKey(hKey);
  if (hProvider)
    CryptReleaseContext(hProvider, 0);
  if (!ret)
    smartcardCertInfo_Free(cert);
  return ret;
}
#endif /* _WIN32 */

static BOOL list_provider_keys(const rdpSettings* settings, NCRYPT_PROV_HANDLE provider,
                               LPCWSTR csp, LPCWSTR scope, const char* userFilter,
                               const char* domainFilter, SmartcardCertInfo*** pcerts,
                               size_t* pcount)
{
  BOOL ret = FALSE;
  NCryptKeyName* keyName = NULL;
  PVOID enumState = NULL;
  SmartcardCertInfo** cert_list = *pcerts;
  size_t count = *pcount;

  while (NCryptEnumKeys(provider, scope, &keyName, &enumState, NCRYPT_SILENT_FLAG) ==
         ERROR_SUCCESS)
  {
    NCRYPT_KEY_HANDLE phKey = 0;
    PBYTE certBytes = NULL;
    DWORD dwFlags = NCRYPT_SILENT_FLAG;
    DWORD cbOutput = 0;
    SmartcardCertInfo* cert = NULL;
    BOOL haveError = TRUE;
    SECURITY_STATUS status = 0;

    cert = calloc(1, sizeof(SmartcardCertInfo));
    if (!cert)
      goto out;

    cert->keyName = ConvertWCharToUtf8Alloc(keyName->pszName, NULL);
    if (!cert->keyName)
      goto endofloop;

    WLog_DBG(TAG, "opening key %s", cert->keyName);

    status =
        NCryptOpenKey(provider, &phKey, keyName->pszName, keyName->dwLegacyKeySpec, dwFlags);
    if (status != ERROR_SUCCESS)
    {
      WLog_DBG(TAG,
               "unable to NCryptOpenKey(dwLegacyKeySpec=0x%" PRIx32 " dwFlags=0x%" PRIx32
               "), status=%s, skipping",
               status, keyName->dwLegacyKeySpec, keyName->dwFlags,
               winpr_NCryptSecurityStatusError(status));
      goto endofloop;
    }

    cert->csp = _wcsdup(csp);
    if (!cert->csp)
      goto endofloop;

#ifndef _WIN32
    status = NCryptGetProperty(phKey, NCRYPT_WINPR_SLOTID, (PBYTE)&cert->slotId, 4, &cbOutput,
                               dwFlags);
    if (status != ERROR_SUCCESS)
    {
      WLog_ERR(TAG, "unable to retrieve slotId for key %s, status=%s", cert->keyName,
               winpr_NCryptSecurityStatusError(status));
      goto endofloop;
    }
#endif /* _WIN32 */

    /* ====== retrieve key's reader ====== */
    cbOutput = 0;
    status = NCryptGetProperty(phKey, NCRYPT_READER_PROPERTY, NULL, 0, &cbOutput, dwFlags);
    if (status != ERROR_SUCCESS)
    {
      WLog_DBG(TAG, "unable to retrieve reader's name length for key %s", cert->keyName);
      goto endofloop;
    }

    cert->reader = calloc(1, cbOutput + 2);
    if (!cert->reader)
    {
      WLog_ERR(TAG, "unable to allocate reader's name for key %s", cert->keyName);
      goto endofloop;
    }

    status = NCryptGetProperty(phKey, NCRYPT_READER_PROPERTY, (PBYTE)cert->reader, cbOutput + 2,
                               &cbOutput, dwFlags);
    if (status != ERROR_SUCCESS)
    {
      WLog_ERR(TAG, "unable to retrieve reader's name for key %s", cert->keyName);
      goto endofloop;
    }

    /* ====== retrieve key container name ====== */
    /* When using PKCS11, this will try to return what Windows would use for the key's name */
    cbOutput = 0;
    status = NCryptGetProperty(phKey, NCRYPT_NAME_PROPERTY, NULL, 0, &cbOutput, dwFlags);
    if (status == ERROR_SUCCESS)
    {
      cert->containerName = calloc(1, cbOutput + sizeof(WCHAR));
      if (!cert->containerName)
      {
        WLog_ERR(TAG, "unable to allocate key container name for key %s", cert->keyName);
        goto endofloop;
      }

      status = NCryptGetProperty(phKey, NCRYPT_NAME_PROPERTY, (BYTE*)cert->containerName,
                                 cbOutput, &cbOutput, dwFlags);
    }

    if (status != ERROR_SUCCESS)
    {
      WLog_ERR(TAG, "unable to retrieve key container name for key %s", cert->keyName);
      goto endofloop;
    }

    /* ========= retrieve the certificate ===============*/
    cbOutput = 0;
    status = NCryptGetProperty(phKey, NCRYPT_CERTIFICATE_PROPERTY, NULL, 0, &cbOutput, dwFlags);
    if (status != ERROR_SUCCESS)
    {
      /* can happen that key don't have certificates */
      WLog_DBG(TAG, "unable to retrieve certificate property len, status=0x%lx, skipping",
               status);
      goto endofloop;
    }

    certBytes = calloc(1, cbOutput);
    if (!certBytes)
    {
      WLog_ERR(TAG, "unable to allocate %" PRIu32 " certBytes for key %s", cbOutput,
               cert->keyName);
      goto endofloop;
    }

    status = NCryptGetProperty(phKey, NCRYPT_CERTIFICATE_PROPERTY, certBytes, cbOutput,
                               &cbOutput, dwFlags);
    if (status != ERROR_SUCCESS)
    {
      WLog_ERR(TAG, "unable to retrieve certificate for key %s", cert->keyName);
      goto endofloop;
    }

    if (!set_info_certificate(cert, certBytes, cbOutput, userFilter, domainFilter))
      goto endofloop;

#ifndef _WIN32
    if (!build_pkinit_args(provider, cert))
    {
      WLog_ERR(TAG, "error build pkinit args");
      goto endofloop;
    }
#endif
    haveError = FALSE;

  endofloop:
    free(certBytes);
    NCryptFreeBuffer(keyName);
    if (phKey)
      NCryptFreeObject((NCRYPT_HANDLE)phKey);

    if (haveError)
      smartcardCertInfo_Free(cert);
    else
    {
      if (!add_cert_to_list(&cert_list, &count, cert))
        goto out;
    }
  }

  ret = TRUE;
out:
  if (count == 0)
  {
    char cspa[128] = { 0 };

    ConvertWCharToUtf8(csp, cspa, sizeof(cspa));
    char scopea[128] = { 0 };
    ConvertWCharToUtf8(scope, scopea, sizeof(scopea));
    WLog_WARN(TAG, "%s [%s] no certificates found", cspa, scopea);
  }
  *pcount = count;
  *pcerts = cert_list;
  NCryptFreeBuffer(enumState);
  return ret;
}

static BOOL smartcard_hw_enumerateCerts(const rdpSettings* settings, LPCWSTR csp,
                                        const char* reader, const char* userFilter,
                                        const char* domainFilter, SmartcardCertInfo*** scCerts,
                                        size_t* retCount)
{
  BOOL ret = FALSE;
  LPWSTR scope = NULL;
  NCRYPT_PROV_HANDLE provider = 0;
  SECURITY_STATUS status = 0;
  size_t count = 0;
  SmartcardCertInfo** cert_list = NULL;
  const char* Pkcs11Module = freerdp_settings_get_string(settings, FreeRDP_Pkcs11Module);

  WINPR_ASSERT(scCerts);
  WINPR_ASSERT(retCount);

  if (reader)
  {
    size_t readerSz = strlen(reader);
    char* scopeStr = malloc(4 + readerSz + 1 + 1);
    if (!scopeStr)
      goto out;

    _snprintf(scopeStr, readerSz + 5, "\\\\.\\%s\\", reader);
    scope = ConvertUtf8NToWCharAlloc(scopeStr, readerSz + 5, NULL);
    free(scopeStr);

    if (!scope)
      goto out;
  }

  if (Pkcs11Module)
  {
    /* load a unique CSP by pkcs11 module path */
    LPCSTR paths[] = { Pkcs11Module, NULL };

    if (!csp)
      csp = MS_SCARD_PROV;

    status = winpr_NCryptOpenStorageProviderEx(&provider, csp, 0, paths);
    if (status != ERROR_SUCCESS)
    {
      WLog_ERR(TAG, "unable to open provider given by pkcs11 module");
      goto out;
    }

    status = list_provider_keys(settings, provider, csp, scope, userFilter, domainFilter,
                                &cert_list, &count);
    NCryptFreeObject((NCRYPT_HANDLE)provider);
    if (!status)
    {
      WLog_ERR(TAG, "error listing keys from CSP loaded from %s", Pkcs11Module);
      goto out;
    }
  }
  else
  {
    NCryptProviderName* names = NULL;
    DWORD nproviders = 0;

#ifdef _WIN32
    /* On Windows, mstsc first enumerates the legacy CAPI providers for usable certificates. */
    DWORD provType, cbProvName = 0;
    for (DWORD i = 0; CryptEnumProvidersW(i, NULL, 0, &provType, NULL, &cbProvName); ++i)
    {
      char providerNameStr[256] = { 0 };
      LPWSTR szProvName = malloc(cbProvName * sizeof(WCHAR));
      if (!CryptEnumProvidersW(i, NULL, 0, &provType, szProvName, &cbProvName))
      {
        free(szProvName);
        break;
      }

      if (ConvertWCharToUtf8(szProvName, providerNameStr, ARRAYSIZE(providerNameStr)) < 0)
      {
        _snprintf(providerNameStr, sizeof(providerNameStr), "<unknown>");
        WLog_ERR(TAG, "unable to convert provider name to char*, will show it as '%s'",
                 providerNameStr);
      }

      WLog_DBG(TAG, "exploring CSP '%s'", providerNameStr);
      if (provType != PROV_RSA_FULL || (csp && _wcscmp(szProvName, csp) != 0))
      {
        WLog_DBG(TAG, "CSP '%s' filtered out", providerNameStr);
        goto end_of_loop;
      }

      if (!list_capi_provider_keys(settings, szProvName, scope, userFilter, domainFilter,
                                   &cert_list, &count))
        WLog_INFO(TAG, "error when retrieving keys in CSP '%s'", providerNameStr);

    end_of_loop:
      free(szProvName);
    }
#endif

    status = NCryptEnumStorageProviders(&nproviders, &names, NCRYPT_SILENT_FLAG);
    if (status != ERROR_SUCCESS)
    {
      WLog_ERR(TAG, "error listing providers");
      goto out;
    }

    for (DWORD i = 0; i < nproviders; i++)
    {
      char providerNameStr[256] = { 0 };
      const NCryptProviderName* name = &names[i];

      if (ConvertWCharToUtf8(name->pszName, providerNameStr, ARRAYSIZE(providerNameStr)) < 0)
      {
        _snprintf(providerNameStr, sizeof(providerNameStr), "<unknown>");
        WLog_ERR(TAG, "unable to convert provider name to char*, will show it as '%s'",
                 providerNameStr);
      }

      WLog_DBG(TAG, "exploring CSP '%s'", providerNameStr);
      if (csp && _wcscmp(name->pszName, csp) != 0)
      {
        WLog_DBG(TAG, "CSP '%s' filtered out", providerNameStr);
        continue;
      }

      status = NCryptOpenStorageProvider(&provider, name->pszName, 0);
      if (status != ERROR_SUCCESS)
        continue;

      if (!list_provider_keys(settings, provider, name->pszName, scope, userFilter,
                              domainFilter, &cert_list, &count))
        WLog_INFO(TAG, "error when retrieving keys in CSP '%s'", providerNameStr);

      NCryptFreeObject((NCRYPT_HANDLE)provider);
    }

    NCryptFreeBuffer(names);
  }

  *scCerts = cert_list;
  *retCount = count;
  ret = TRUE;

out:
  if (!ret)
    smartcardCertList_Free(cert_list, count);
  free(scope);
  return ret;
}

static char* create_temporary_file(void)
{
  BYTE buffer[32];
  char* hex = NULL;
  char* path = NULL;

  winpr_RAND(buffer, sizeof(buffer));
  hex = winpr_BinToHexString(buffer, sizeof(buffer), FALSE);
  path = GetKnownSubPath(KNOWN_PATH_TEMP, hex);
  free(hex);
  return path;
}

static SmartcardCertInfo* smartcardCertInfo_New(const char* privKeyPEM, const char* certPEM)
{
  size_t size = 0;

  WINPR_ASSERT(privKeyPEM);
  WINPR_ASSERT(certPEM);

  SmartcardCertInfo* cert = calloc(1, sizeof(SmartcardCertInfo));
  if (!cert)
    goto fail;

  SmartcardKeyInfo* info = cert->key_info = calloc(1, sizeof(SmartcardKeyInfo));
  if (!info)
    goto fail;

  cert->certificate = freerdp_certificate_new_from_pem(certPEM);
  if (!cert->certificate)
  {
    WLog_ERR(TAG, "unable to read smartcard certificate");
    goto fail;
  }

  if (!treat_sc_cert(cert))
  {
    WLog_ERR(TAG, "unable to treat smartcard certificate");
    goto fail;
  }

  cert->reader = ConvertUtf8ToWCharAlloc("FreeRDP Emulator", NULL);
  if (!cert->reader)
    goto fail;

  cert->containerName = ConvertUtf8ToWCharAlloc("Private Key 00", NULL);
  if (!cert->containerName)
    goto fail;

  /* compute PKINIT args FILE:<cert file>,<key file>
   *
   * We need files for PKINIT to read, so write the certificate to some
   * temporary location and use that.
   */
  info->keyPath = create_temporary_file();
  WLog_DBG(TAG, "writing PKINIT key to %s", info->keyPath);
  if (!crypto_write_pem(info->keyPath, privKeyPEM, strlen(privKeyPEM)))
    goto fail;

  info->certPath = create_temporary_file();
  WLog_DBG(TAG, "writing PKINIT cert to %s", info->certPath);
  if (!crypto_write_pem(info->certPath, certPEM, strlen(certPEM)))
    goto fail;

  int res = winpr_asprintf(&cert->pkinitArgs, &size, "FILE:%s,%s", info->certPath, info->keyPath);
  if (res <= 0)
    goto fail;

  return cert;
fail:
  smartcardCertInfo_Free(cert);
  return NULL;
}

static BOOL smartcard_sw_enumerateCerts(const rdpSettings* settings, SmartcardCertInfo*** scCerts,
                                        size_t* retCount)
{
  BOOL rc = FALSE;
  SmartcardCertInfo** cert_list = NULL;

  WINPR_ASSERT(settings);
  WINPR_ASSERT(scCerts);
  WINPR_ASSERT(retCount);

  const char* privKeyPEM = freerdp_settings_get_string(settings, FreeRDP_SmartcardPrivateKey);
  const char* certPEM = freerdp_settings_get_string(settings, FreeRDP_SmartcardCertificate);
  if (!privKeyPEM)
  {
    WLog_ERR(TAG, "Invalid smartcard private key PEM, aborting");
    goto out_error;
  }
  if (!certPEM)
  {
    WLog_ERR(TAG, "Invalid smartcard certificate PEM, aborting");
    goto out_error;
  }

  cert_list = calloc(1, sizeof(SmartcardCertInfo*));
  if (!cert_list)
    goto out_error;

  {
    SmartcardCertInfo* cert = smartcardCertInfo_New(privKeyPEM, certPEM);
    if (!cert)
      goto out_error;
    cert_list[0] = cert;
  }

  rc = TRUE;
  *scCerts = cert_list;
  *retCount = 1;

out_error:
  if (!rc)
    smartcardCertList_Free(cert_list, 1);
  return rc;
}

BOOL smartcard_enumerateCerts(const rdpSettings* settings, SmartcardCertInfo*** scCerts,
                              size_t* retCount, BOOL gateway)
{
  BOOL ret = 0;
  LPWSTR csp = NULL;
  const char* ReaderName = freerdp_settings_get_string(settings, FreeRDP_ReaderName);
  const char* CspName = freerdp_settings_get_string(settings, FreeRDP_CspName);
  const char* Username = NULL;
  const char* Domain = NULL;

  if (gateway)
  {
    Username = freerdp_settings_get_string(settings, FreeRDP_GatewayUsername);
    Domain = freerdp_settings_get_string(settings, FreeRDP_GatewayDomain);
  }
  else
  {
    Username = freerdp_settings_get_string(settings, FreeRDP_Username);
    Domain = freerdp_settings_get_string(settings, FreeRDP_Domain);
  }

  WINPR_ASSERT(settings);
  WINPR_ASSERT(scCerts);
  WINPR_ASSERT(retCount);

  if (Domain && !strlen(Domain))
    Domain = NULL;

  if (freerdp_settings_get_bool(settings, FreeRDP_SmartcardEmulation))
    return smartcard_sw_enumerateCerts(settings, scCerts, retCount);

  if (CspName && (!(csp = ConvertUtf8ToWCharAlloc(CspName, NULL))))
  {
    WLog_ERR(TAG, "error while converting CSP to WCHAR");
    return FALSE;
  }

  ret =
      smartcard_hw_enumerateCerts(settings, csp, ReaderName, Username, Domain, scCerts, retCount);
  free(csp);
  return ret;
}

static BOOL set_settings_from_smartcard(rdpSettings* settings, FreeRDP_Settings_Keys_String id,
                                        const char* value)
{
  WINPR_ASSERT(settings);

  if (!freerdp_settings_get_string(settings, id) && value)
    if (!freerdp_settings_set_string(settings, id, value))
      return FALSE;

  return TRUE;
}

BOOL smartcard_getCert(const rdpContext* context, SmartcardCertInfo** cert, BOOL gateway)
{
  WINPR_ASSERT(context);

  const freerdp* instance = context->instance;
  rdpSettings* settings = context->settings;
  SmartcardCertInfo** cert_list = NULL;
  size_t count = 0;

  WINPR_ASSERT(instance);
  WINPR_ASSERT(settings);

  if (!smartcard_enumerateCerts(settings, &cert_list, &count, gateway))
    return FALSE;

  if (count < 1)
  {
    WLog_ERR(TAG, "no suitable smartcard certificates were found");
    return FALSE;
  }

  if (count > UINT32_MAX)
  {
    WLog_ERR(TAG, "smartcard certificate count %" PRIuz " exceeds UINT32_MAX", count);
    return FALSE;
  }

  if (count > 1)
  {
    DWORD index = 0;

    if (!instance->ChooseSmartcard ||
        !instance->ChooseSmartcard(context->instance, cert_list, (UINT32)count, &index,
                                   gateway))
    {
      WLog_ERR(TAG, "more than one suitable smartcard certificate was found");
      smartcardCertList_Free(cert_list, count);
      return FALSE;
    }
    *cert = cert_list[index];

    for (DWORD i = 0; i < index; i++)
      smartcardCertInfo_Free(cert_list[i]);
    for (DWORD i = index + 1; i < count; i++)
      smartcardCertInfo_Free(cert_list[i]);
  }
  else
    *cert = cert_list[0];

  FreeRDP_Settings_Keys_String username_setting =
      gateway ? FreeRDP_GatewayUsername : FreeRDP_Username;
  FreeRDP_Settings_Keys_String domain_setting = gateway ? FreeRDP_GatewayDomain : FreeRDP_Domain;

  free(cert_list);

  if (!set_settings_from_smartcard(settings, username_setting, (*cert)->userHint) ||
      !set_settings_from_smartcard(settings, domain_setting, (*cert)->domainHint))
  {
    WLog_ERR(TAG, "unable to set settings from smartcard!");
    smartcardCertInfo_Free(*cert);
    return FALSE;
  }

  return TRUE;
}

Coverage Report

Created: 2024-05-20 06:11