TestKnownHosts.c

 
#include <winpr/path.h>
#include <winpr/file.h>
#include <winpr/sysinfo.h>
 
#include <freerdp/crypto/certificate_store.h>
 
/* Some certificates copied from /usr/share/ca-certificates */
static const char pem1[] = "-----BEGIN CERTIFICATE-----\n"
                           "MIIFWjCCA0KgAwIBAgIQbkepxUtHDA3sM9CJuRz04TANBgkqhkiG9w0BAQwFADBH\n"
                           "MQswCQYDVQQGEwJVUzEiMCAGA1UEChMZR29vZ2xlIFRydXN0IFNlcnZpY2VzIExM\n"
                           "QzEUMBIGA1UEAxMLR1RTIFJvb3QgUjEwHhcNMTYwNjIyMDAwMDAwWhcNMzYwNjIy\n"
                           "MDAwMDAwWjBHMQswCQYDVQQGEwJVUzEiMCAGA1UEChMZR29vZ2xlIFRydXN0IFNl\n"
                           "cnZpY2VzIExMQzEUMBIGA1UEAxMLR1RTIFJvb3QgUjEwggIiMA0GCSqGSIb3DQEB\n"
                           "AQUAA4ICDwAwggIKAoICAQC2EQKLHuOhd5s73L+UPreVp0A8of2C+X0yBoJx9vaM\n"
                           "f/vo27xqLpeXo4xL+Sv2sfnOhB2x+cWX3u+58qPpvBKJXqeqUqv4IyfLpLGcY9vX\n"
                           "mX7wCl7raKb0xlpHDU0QM+NOsROjyBhsS+z8CZDfnWQpJSMHobTSPS5g4M/SCYe7\n"
                           "zUjwTcLCeoiKu7rPWRnWr4+wB7CeMfGCwcDfLqZtbBkOtdh+JhpFAz2weaSUKK0P\n"
                           "fyblqAj+lug8aJRT7oM6iCsVlgmy4HqMLnXWnOunVmSPlk9orj2XwoSPwLxAwAtc\n"
                           "vfaHszVsrBhQf4TgTM2S0yDpM7xSma8ytSmzJSq0SPly4cpk9+aCEI3oncKKiPo4\n"
                           "Zor8Y/kB+Xj9e1x3+naH+uzfsQ55lVe0vSbv1gHR6xYKu44LtcXFilWr06zqkUsp\n"
                           "zBmkMiVOKvFlRNACzqrOSbTqn3yDsEB750Orp2yjj32JgfpMpf/VjsPOS+C12LOO\n"
                           "Rc92wO1AK/1TD7Cn1TsNsYqiA94xrcx36m97PtbfkSIS5r762DL8EGMUUXLeXdYW\n"
                           "k70paDPvOmbsB4om3xPXV2V4J95eSRQAogB/mqghtqmxlbCluQ0WEdrHbEg8QOB+\n"
                           "DVrNVjzRlwW5y0vtOUucxD/SVRNuJLDWcfr0wbrM7Rv1/oFB2ACYPTrIrnqYNxgF\n"
                           "lQIDAQABo0IwQDAOBgNVHQ8BAf8EBAMCAQYwDwYDVR0TAQH/BAUwAwEB/zAdBgNV\n"
                           "HQ4EFgQU5K8rJnEaK0gnhS9SZizv8IkTcT4wDQYJKoZIhvcNAQEMBQADggIBADiW\n"
                           "Cu49tJYeX++dnAsznyvgyv3SjgofQXSlfKqE1OXyHuY3UjKcC9FhHb8owbZEKTV1\n"
                           "d5iyfNm9dKyKaOOpMQkpAWBz40d8U6iQSifvS9efk+eCNs6aaAyC58/UEBZvXw6Z\n"
                           "XPYfcX3v73svfuo21pdwCxXu11xWajOl40k4DLh9+42FpLFZXvRq4d2h9mREruZR\n"
                           "gyFmxhE+885H7pwoHyXa/6xmld01D1zvICxi/ZG6qcz8WpyTgYMpl0p8WnK0OdC3\n"
                           "d8t5/Wk6kjftbjhlRn7pYL15iJdfOBL07q9bgsiG1eGZbYwE8na6SfZu6W0eX6Dv\n"
                           "J4J2QPim01hcDyxC2kLGe4g0x8HYRZvBPsVhHdljUEn2NIVq4BjFbkerQUIpm/Zg\n"
                           "DdIx02OYI5NaAIFItO/Nis3Jz5nu2Z6qNuFoS3FJFDYoOj0dzpqPJeaAcWErtXvM\n"
                           "+SUWgeExX6GjfhaknBZqlxi9dnKlC54dNuYvoS++cJEPqOba+MSSQGwlfnuzCdyy\n"
                           "F62ARPBopY+Udf90WuioAnwMCeKpSwughQtiue+hMZL77/ZRBIls6Kl0obsXs7X9\n"
                           "SQ98POyDGCBDTtWTurQ0sR8WNh8M5mQ5Fkzc4P4dyKliPUDqysU0ArSuiYgzNdws\n"
                           "E3PYJ/HQcu51OyLemGhmW/HGY0dVHLqlCFF1pkgl\n"
                           "-----END CERTIFICATE-----";
 
static const char pem2[] = "-----BEGIN CERTIFICATE-----\n"
                           "MIIFWjCCA0KgAwIBAgIQbkepxlqz5yDFMJo/aFLybzANBgkqhkiG9w0BAQwFADBH\n"
                           "MQswCQYDVQQGEwJVUzEiMCAGA1UEChMZR29vZ2xlIFRydXN0IFNlcnZpY2VzIExM\n"
                           "QzEUMBIGA1UEAxMLR1RTIFJvb3QgUjIwHhcNMTYwNjIyMDAwMDAwWhcNMzYwNjIy\n"
                           "MDAwMDAwWjBHMQswCQYDVQQGEwJVUzEiMCAGA1UEChMZR29vZ2xlIFRydXN0IFNl\n"
                           "cnZpY2VzIExMQzEUMBIGA1UEAxMLR1RTIFJvb3QgUjIwggIiMA0GCSqGSIb3DQEB\n"
                           "AQUAA4ICDwAwggIKAoICAQDO3v2m++zsFDQ8BwZabFn3GTXd98GdVarTzTukk3Lv\n"
                           "CvptnfbwhYBboUhSnznFt+4orO/LdmgUud+tAWyZH8QiHZ/+cnfgLFuv5AS/T3Kg\n"
                           "GjSY6Dlo7JUle3ah5mm5hRm9iYz+re026nO8/4Piy33B0s5Ks40FnotJk9/BW9Bu\n"
                           "XvAuMC6C/Pq8tBcKSOWIm8Wba96wyrQD8Nr0kLhlZPdcTK3ofmZemde4wj7I0BOd\n"
                           "re7kRXuJVfeKH2JShBKzwkCX44ofR5GmdFrS+LFjKBC4swm4VndAoiaYecb+3yXu\n"
                           "PuWgf9RhD1FLPD+M2uFwdNjCaKH5wQzpoeJ/u1U8dgbuak7MkogwTZq9TwtImoS1\n"
                           "mKPV+3PBV2HdKFZ1E66HjucMUQkQdYhMvI35ezzUIkgfKtzra7tEscszcTJGr61K\n"
                           "8YzodDqs5xoic4DSMPclQsciOzsSrZYuxsN2B6ogtzVJV+mSSeh2FnIxZyuWfoqj\n"
                           "x5RWIr9qS34BIbIjMt/kmkRtWVtd9QCgHJvGeJeNkP+byKq0rxFROV7Z+2et1VsR\n"
                           "nTKaG73VululycslaVNVJ1zgyjbLiGH7HrfQy+4W+9OmTN6SpdTi3/UGVN4unUu0\n"
                           "kzCqgc7dGtxRcw1PcOnlthYhGXmy5okLdWTK1au8CcEYof/UVKGFPP0UJAOyh9Ok\n"
                           "twIDAQABo0IwQDAOBgNVHQ8BAf8EBAMCAQYwDwYDVR0TAQH/BAUwAwEB/zAdBgNV\n"
                           "HQ4EFgQUu//KjiOfT5nK2+JopqUVJxce2Q4wDQYJKoZIhvcNAQEMBQADggIBALZp\n"
                           "8KZ3/p7uC4Gt4cCpx/k1HUCCq+YEtN/L9x0Pg/B+E02NjO7jMyLDOfxA325BS0JT\n"
                           "vhaI8dI4XsRomRyYUpOM52jtG2pzegVATX9lO9ZY8c6DR2Dj/5epnGB3GFW1fgiT\n"
                           "z9D2PGcDFWEJ+YF59exTpJ/JjwGLc8R3dtyDovUMSRqodt6Sm2T4syzFJ9MHwAiA\n"
                           "pJiS4wGWAqoC7o87xdFtCjMwc3i5T1QWvwsHoaRc5svJXISPD+AVdyx+Jn7axEvb\n"
                           "pxZ3B7DNdehyQtaVhJ2Gg/LkkM0JR9SLA3DaWsYDQvTtN6LwG1BUSw7YhN4ZKJmB\n"
                           "R64JGz9I0cNv4rBgF/XuIwKl2gBbbZCr7qLpGzvpx0QnRY5rn/WkhLx3+WuXrD5R\n"
                           "RaIRpsyF7gpo8j5QOHokYh4XIDdtak23CZvJ/KRY9bb7nE4Yu5UC56GtmwfuNmsk\n"
                           "0jmGwZODUNKBRqhfYlcsu2xkiAhu7xNUX90txGdj08+JN7+dIPT7eoOboB6BAFDC\n"
                           "5AwiWVIQ7UNWhwD4FFKnHYuTjKJNRn8nxnGbJN7k2oaLDX5rIMHAnuFl2GqjpuiF\n"
                           "izoHCBy69Y9Vmhh1fuXsgWbRIXOhNUQLgD1bnF5vKheW0YMjiGZt5obicDIvUiLn\n"
                           "yOd/xCxgXS/Dr55FBcOEArf9LAhST4Ldo/DUhgkC\n"
                           "-----END CERTIFICATE-----";
 
static const char pem3[] = "-----BEGIN CERTIFICATE-----\n"
                           "MIICDDCCAZGgAwIBAgIQbkepx2ypcyRAiQ8DVd2NHTAKBggqhkjOPQQDAzBHMQsw\n"
                           "CQYDVQQGEwJVUzEiMCAGA1UEChMZR29vZ2xlIFRydXN0IFNlcnZpY2VzIExMQzEU\n"
                           "MBIGA1UEAxMLR1RTIFJvb3QgUjMwHhcNMTYwNjIyMDAwMDAwWhcNMzYwNjIyMDAw\n"
                           "MDAwWjBHMQswCQYDVQQGEwJVUzEiMCAGA1UEChMZR29vZ2xlIFRydXN0IFNlcnZp\n"
                           "Y2VzIExMQzEUMBIGA1UEAxMLR1RTIFJvb3QgUjMwdjAQBgcqhkjOPQIBBgUrgQQA\n"
                           "IgNiAAQfTzOHMymKoYTey8chWEGJ6ladK0uFxh1MJ7x/JlFyb+Kf1qPKzEUURout\n"
                           "736GjOyxfi//qXGdGIRFBEFVbivqJn+7kAHjSxm65FSWRQmx1WyRRK2EE46ajA2A\n"
                           "DDL24CejQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNVHRMBAf8EBTADAQH/MB0GA1Ud\n"
                           "DgQWBBTB8Sa6oC2uhYHP0/EqEr24Cmf9vDAKBggqhkjOPQQDAwNpADBmAjEAgFuk\n"
                           "fCPAlaUs3L6JbyO5o91lAFJekazInXJ0glMLfalAvWhgxeG4VDvBNhcl2MG9AjEA\n"
                           "njWSdIUlUfUk7GRSJFClH9voy8l27OyCbvWFGFPouOOaKaqW04MjyaR7YbPMAuhd\n"
                           "-----END CERTIFICATE-----";
 
static const char pem4[] = "-----BEGIN CERTIFICATE-----\n"
                           "MIICCjCCAZGgAwIBAgIQbkepyIuUtui7OyrYorLBmTAKBggqhkjOPQQDAzBHMQsw\n"
                           "CQYDVQQGEwJVUzEiMCAGA1UEChMZR29vZ2xlIFRydXN0IFNlcnZpY2VzIExMQzEU\n"
                           "MBIGA1UEAxMLR1RTIFJvb3QgUjQwHhcNMTYwNjIyMDAwMDAwWhcNMzYwNjIyMDAw\n"
                           "MDAwWjBHMQswCQYDVQQGEwJVUzEiMCAGA1UEChMZR29vZ2xlIFRydXN0IFNlcnZp\n"
                           "Y2VzIExMQzEUMBIGA1UEAxMLR1RTIFJvb3QgUjQwdjAQBgcqhkjOPQIBBgUrgQQA\n"
                           "IgNiAATzdHOnaItgrkO4NcWBMHtLSZ37wWHO5t5GvWvVYRg1rkDdc/eJkTBa6zzu\n"
                           "hXyiQHY7qca4R9gq55KRanPpsXI5nymfopjTX15YhmUPoYRlBtHci8nHc8iMai/l\n"
                           "xKvRHYqjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNVHRMBAf8EBTADAQH/MB0GA1Ud\n"
                           "DgQWBBSATNbrdP9JNqPV2Py1PsVq8JQdjDAKBggqhkjOPQQDAwNnADBkAjBqUFJ0\n"
                           "CMRw3J5QdCHojXohw0+WbhXRIjVhLfoIN+4Zba3bssx9BzT1YBkstTTZbyACMANx\n"
                           "sbqjYAuG7ZoIapVon+Kz4ZNkfF6Tpt95LY2F45TPI11xzPKwTdb+mciUqXWi4w==\n"
                           "-----END CERTIFICATE-----";
 
static int prepare(const char* currentFileV2)
{
  int rc = -1;
  const char* hosts[] = { "#somecomment\r\n"
                        "someurl 3389 ff:11:22:dd c3ViamVjdA== aXNzdWVy\r\n"
                        " \t#anothercomment\r\n"
                        "otherurl\t3389\taa:bb:cc:dd\tsubject2\tissuer2\r" };
  FILE* fc = NULL;
  fc = winpr_fopen(currentFileV2, "w+");
 
  if (!fc)
    goto finish;
 
  for (size_t i = 0; i < ARRAYSIZE(hosts); i++)
  {
    if (fwrite(hosts[i], strlen(hosts[i]), 1, fc) != 1)
      goto finish;
  }
 
  rc = 0;
finish:
 
  if (fc)
    (void)fclose(fc);
 
  return rc;
}
 
static BOOL setup_config(rdpSettings** settings)
{
  BOOL rc = FALSE;
  char* path = NULL;
  char sname[8192];
  SYSTEMTIME systemTime;
 
  if (!settings)
    goto fail;
  *settings = freerdp_settings_new(0);
  if (!*settings)
    goto fail;
 
  GetSystemTime(&systemTime);
  (void)sprintf_s(sname, sizeof(sname),
                  "TestKnownHostsCurrent-%04" PRIu16 "%02" PRIu16 "%02" PRIu16 "%02" PRIu16
                  "%02" PRIu16 "%02" PRIu16 "%04" PRIu16,
                  systemTime.wYear, systemTime.wMonth, systemTime.wDay, systemTime.wHour,
                  systemTime.wMinute, systemTime.wSecond, systemTime.wMilliseconds);
 
  path = GetKnownSubPath(KNOWN_PATH_TEMP, sname);
  if (!path)
    goto fail;
  if (!winpr_PathFileExists(path))
  {
    if (!CreateDirectoryA(path, NULL))
    {
      (void)fprintf(stderr, "Could not create %s!\n", path);
      goto fail;
    }
  }
 
  rc = freerdp_settings_set_string(*settings, FreeRDP_ConfigPath, path);
fail:
  free(path);
  return rc;
}
 
static BOOL equal(const char* a, const char* b)
{
  if (!a && !b)
    return TRUE;
  if (!a || !b)
    return FALSE;
  return strcmp(a, b) == 0;
}
 
static BOOL compare(const rdpCertificateData* data, const rdpCertificateData* stored)
{
  if (!data || !stored)
    return FALSE;
  if (!equal(freerdp_certificate_data_get_subject(data),
             freerdp_certificate_data_get_subject(stored)))
    return FALSE;
  if (!equal(freerdp_certificate_data_get_issuer(data),
             freerdp_certificate_data_get_issuer(stored)))
    return FALSE;
  if (!equal(freerdp_certificate_data_get_fingerprint(data),
             freerdp_certificate_data_get_fingerprint(stored)))
    return FALSE;
  return TRUE;
}
 
static BOOL pem_equal(const char* a, const char* b)
{
  return strcmp(a, b) == 0;
}
 
static BOOL compare_ex(const rdpCertificateData* data, const rdpCertificateData* stored)
{
  if (!compare(data, stored))
    return FALSE;
  if (!pem_equal(freerdp_certificate_data_get_pem(data),
                 freerdp_certificate_data_get_pem(stored)))
    return FALSE;
 
  return TRUE;
}
 
static BOOL test_get_data(rdpCertificateStore* store, const rdpCertificateData* data)
{
  BOOL res = 0;
  rdpCertificateData* stored = freerdp_certificate_store_load_data(
      store, freerdp_certificate_data_get_host(data), freerdp_certificate_data_get_port(data));
  if (!stored)
    return FALSE;
 
  res = compare(data, stored);
  freerdp_certificate_data_free(stored);
  return res;
}
 
static BOOL test_get_data_ex(rdpCertificateStore* store, const rdpCertificateData* data)
{
  BOOL res = 0;
  rdpCertificateData* stored = freerdp_certificate_store_load_data(
      store, freerdp_certificate_data_get_host(data), freerdp_certificate_data_get_port(data));
  if (!stored)
    return FALSE;
 
  res = compare_ex(data, stored);
  freerdp_certificate_data_free(stored);
  return res;
}
 
static BOOL test_certs_dir(void)
{
  BOOL rc = FALSE;
  rdpSettings* settings = NULL;
  rdpCertificateStore* store = NULL;
  rdpCertificateData* data1 = NULL;
  rdpCertificateData* data2 = NULL;
  rdpCertificateData* data3 = NULL;
  rdpCertificateData* data4 = NULL;
 
  printf("%s\n", __func__);
  if (!setup_config(&settings))
    goto fail;
 
  printf("freerdp_certificate_store_new()\n");
  store = freerdp_certificate_store_new(settings);
  if (!store)
    goto fail;
 
  {
    printf("freerdp_certificate_data_new()\n");
    data1 = freerdp_certificate_data_new_from_pem("somehost", 1234, pem1, strlen(pem1));
    data2 = freerdp_certificate_data_new_from_pem("otherhost", 4321, pem2, strlen(pem2));
    data3 = freerdp_certificate_data_new_from_pem("otherhost4", 444, pem3, strlen(pem3));
    data4 = freerdp_certificate_data_new_from_pem("otherhost", 4321, pem4, strlen(pem4));
    if (!data1 || !data2 || !data3 || !data4)
      goto fail;
 
    /* Find non existing in empty store */
    printf("freerdp_certificate_store_load_data on empty store\n");
    if (test_get_data(store, data1))
      goto fail;
    if (test_get_data_ex(store, data1))
      goto fail;
    if (test_get_data(store, data2))
      goto fail;
    if (test_get_data_ex(store, data2))
      goto fail;
    if (test_get_data(store, data3))
      goto fail;
    if (test_get_data_ex(store, data3))
      goto fail;
 
    /* Add certificates */
    printf("freerdp_certificate_store_save_data\n");
    if (!freerdp_certificate_store_save_data(store, data1))
      goto fail;
    if (!freerdp_certificate_store_save_data(store, data2))
      goto fail;
 
    /* Find non existing in non empty store */
    printf("freerdp_certificate_store_load_data on filled store, non existing value\n");
    if (test_get_data(store, data3))
      goto fail;
    if (test_get_data_ex(store, data3))
      goto fail;
 
    /* Add remaining certs */
    printf("freerdp_certificate_store_save_data\n");
    if (!freerdp_certificate_store_save_data(store, data3))
      goto fail;
 
    /* Check existing can all be found */
    printf("freerdp_certificate_store_load_data on filled store, existing value\n");
    if (!test_get_data(store, data1))
      goto fail;
    if (!test_get_data_ex(store, data1))
      goto fail;
    if (!test_get_data(store, data2))
      goto fail;
    if (!test_get_data_ex(store, data2))
      goto fail;
    if (!test_get_data(store, data3))
      goto fail;
    if (!test_get_data_ex(store, data3))
      goto fail;
 
    /* Modify existing entry */
    printf("freerdp_certificate_store_save_data modify data\n");
    if (!freerdp_certificate_store_save_data(store, data4))
      goto fail;
 
    /* Check new data is in store */
    printf("freerdp_certificate_store_load_data check modified data can be loaded\n");
    if (!test_get_data(store, data4))
      goto fail;
    if (!test_get_data_ex(store, data4))
      goto fail;
 
    /* Check old data is no longer valid */
    printf("freerdp_certificate_store_load_data check original data no longer there\n");
    if (test_get_data(store, data2))
      goto fail;
    if (test_get_data_ex(store, data2))
      goto fail;
 
    /* Delete a cert */
    printf("freerdp_certificate_store_remove_data\n");
    if (!freerdp_certificate_store_remove_data(store, data3))
      goto fail;
    /* Delete non existing, should succeed */
    printf("freerdp_certificate_store_remove_data missing value\n");
    if (!freerdp_certificate_store_remove_data(store, data3))
      goto fail;
 
    printf("freerdp_certificate_store_load_data on filled store, existing value\n");
    if (!test_get_data(store, data1))
      goto fail;
    if (!test_get_data_ex(store, data1))
      goto fail;
    if (!test_get_data(store, data4))
      goto fail;
    if (!test_get_data_ex(store, data4))
      goto fail;
 
    printf("freerdp_certificate_store_load_data on filled store, removed value\n");
    if (test_get_data(store, data3))
      goto fail;
    if (test_get_data_ex(store, data3))
      goto fail;
  }
 
  rc = TRUE;
fail:
  printf("freerdp_certificate_data_free %d\n", rc);
  freerdp_certificate_data_free(data1);
  freerdp_certificate_data_free(data2);
  freerdp_certificate_data_free(data3);
  freerdp_certificate_data_free(data4);
  freerdp_certificate_store_free(store);
  freerdp_settings_free(settings);
  return rc;
}
 
int TestKnownHosts(int argc, char* argv[])
{
  WINPR_UNUSED(argc);
  WINPR_UNUSED(argv);
 
  if (!test_certs_dir())
    return -1;
  return 0;
}