TestDsMakeSpn.c

 
#include <stdio.h>
#include <winpr/crt.h>
#include <winpr/winpr.h>
#include <winpr/tchar.h>
#include <winpr/dsparse.h>
 
static BOOL test_DsMakeSpnA(void)
{
  LPCSTR testServiceClass = "HTTP";
  LPCSTR testServiceName = "LAB1-W2K8R2-GW.lab1.awake.local";
  LPCSTR testSpn = "HTTP/LAB1-W2K8R2-GW.lab1.awake.local";
  BOOL rc = FALSE;
  CHAR Spn[100] = { 0 };
  DWORD status = 0;
  DWORD SpnLength = -1;
 
  status = DsMakeSpnA(testServiceClass, testServiceName, NULL, 0, NULL, &SpnLength, NULL);
 
  if (status != ERROR_INVALID_PARAMETER)
  {
    printf("DsMakeSpnA: expected ERROR_INVALID_PARAMETER\n");
    goto fail;
  }
 
  SpnLength = 0;
  status = DsMakeSpnA(testServiceClass, testServiceName, NULL, 0, NULL, &SpnLength, NULL);
 
  if (status != ERROR_BUFFER_OVERFLOW)
  {
    printf("DsMakeSpnA: expected ERROR_BUFFER_OVERFLOW\n");
    goto fail;
  }
 
  if (SpnLength != 37)
  {
    printf("DsMakeSpnA: SpnLength mismatch: Actual: %" PRIu32 ", Expected: 37\n", SpnLength);
    goto fail;
  }
 
  status = DsMakeSpnA(testServiceClass, testServiceName, NULL, 0, NULL, &SpnLength, Spn);
 
  if (status != ERROR_SUCCESS)
  {
    printf("DsMakeSpnA: expected ERROR_SUCCESS\n");
    goto fail;
  }
 
  if (strcmp(Spn, testSpn) != 0)
  {
    printf("DsMakeSpnA: SPN mismatch: Actual: %s, Expected: %s\n", Spn, testSpn);
    goto fail;
  }
 
  printf("DsMakeSpnA: %s\n", Spn);
  rc = TRUE;
fail:
  return rc;
}
 
static BOOL test_DsMakeSpnW(void)
{
  const CHAR ctestServiceClass[] = { 'H', 'T', 'T', 'P', '\0' };
  const CHAR ctestServiceName[] = { 'L', 'A', 'B', '1', '-', 'W', '2', 'K', '8', 'R', '2',
                                  '-', 'G', 'W', '.', 'l', 'a', 'b', '1', '.', 'a', 'w',
                                  'a', 'k', 'e', '.', 'l', 'o', 'c', 'a', 'l', '\0' };
  const CHAR ctestSpn[] = { 'H', 'T', 'T', 'P', '/', 'L', 'A', 'B', '1', '-', 'W', '2', 'K',
                          '8', 'R', '2', '-', 'G', 'W', '.', 'l', 'a', 'b', '1', '.', 'a',
                          'w', 'a', 'k', 'e', '.', 'l', 'o', 'c', 'a', 'l', '\0' };
  WCHAR testServiceClass[ARRAYSIZE(ctestServiceClass)] = { 0 };
  WCHAR testServiceName[ARRAYSIZE(ctestServiceName)] = { 0 };
  WCHAR testSpn[ARRAYSIZE(ctestSpn)] = { 0 };
 
  BOOL rc = FALSE;
  WCHAR Spn[100] = { 0 };
  DWORD status = 0;
  DWORD SpnLength = -1;
 
  (void)ConvertUtf8NToWChar(ctestServiceClass, ARRAYSIZE(ctestServiceClass), testServiceClass,
                            ARRAYSIZE(testServiceClass));
  (void)ConvertUtf8NToWChar(ctestServiceName, ARRAYSIZE(ctestServiceName), testServiceName,
                            ARRAYSIZE(testServiceName));
  (void)ConvertUtf8NToWChar(ctestSpn, ARRAYSIZE(ctestSpn), testSpn, ARRAYSIZE(testSpn));
 
  status = DsMakeSpnW(testServiceClass, testServiceName, NULL, 0, NULL, &SpnLength, NULL);
 
  if (status != ERROR_INVALID_PARAMETER)
  {
    printf("DsMakeSpnW: expected ERROR_INVALID_PARAMETER\n");
    goto fail;
  }
 
  SpnLength = 0;
  status = DsMakeSpnW(testServiceClass, testServiceName, NULL, 0, NULL, &SpnLength, NULL);
 
  if (status != ERROR_BUFFER_OVERFLOW)
  {
    printf("DsMakeSpnW: expected ERROR_BUFFER_OVERFLOW\n");
    goto fail;
  }
 
  if (SpnLength != 37)
  {
    printf("DsMakeSpnW: SpnLength mismatch: Actual: %" PRIu32 ", Expected: 37\n", SpnLength);
    goto fail;
  }
 
  status = DsMakeSpnW(testServiceClass, testServiceName, NULL, 0, NULL, &SpnLength, Spn);
 
  if (status != ERROR_SUCCESS)
  {
    printf("DsMakeSpnW: expected ERROR_SUCCESS\n");
    goto fail;
  }
 
  if (_wcscmp(Spn, testSpn) != 0)
  {
    char buffer1[8192] = { 0 };
    char buffer2[8192] = { 0 };
    char* SpnA = buffer1;
    char* testSpnA = buffer2;
 
    (void)ConvertWCharToUtf8(Spn, SpnA, ARRAYSIZE(buffer1));
    (void)ConvertWCharToUtf8(testSpn, testSpnA, ARRAYSIZE(buffer2));
    printf("DsMakeSpnW: SPN mismatch: Actual: %s, Expected: %s\n", SpnA, testSpnA);
    goto fail;
  }
 
  {
    char buffer[8192] = { 0 };
    char* SpnA = buffer;
 
    (void)ConvertWCharToUtf8(Spn, SpnA, ARRAYSIZE(buffer));
    printf("DsMakeSpnW: %s\n", SpnA);
  }
 
  rc = TRUE;
fail:
  return rc;
}
int TestDsMakeSpn(int argc, char* argv[])
{
  WINPR_UNUSED(argc);
  WINPR_UNUSED(argv);
 
  if (!test_DsMakeSpnA())
    return -1;
  if (!test_DsMakeSpnW())
    return -2;
  return 0;
}