dsparse.c

 
#include <winpr/config.h>
 
#include <winpr/dsparse.h>
#include <winpr/assert.h>
 
#if !defined(_WIN32) || defined(_UWP)
 
DWORD DsMakeSpnW(LPCWSTR ServiceClass, LPCWSTR ServiceName, LPCWSTR InstanceName,
                 USHORT InstancePort, LPCWSTR Referrer, DWORD* pcSpnLength, LPWSTR pszSpn)
{
  DWORD res = ERROR_OUTOFMEMORY;
  char* ServiceClassA = NULL;
  char* ServiceNameA = NULL;
  char* InstanceNameA = NULL;
  char* ReferrerA = NULL;
  char* pszSpnA = NULL;
  size_t length = 0;
 
  WINPR_ASSERT(ServiceClass);
  WINPR_ASSERT(ServiceName);
  WINPR_ASSERT(pcSpnLength);
 
  length = *pcSpnLength;
  if ((length > 0) && pszSpn)
    pszSpnA = calloc(length + 1, sizeof(char));
 
  if (ServiceClass)
  {
    ServiceClassA = ConvertWCharToUtf8Alloc(ServiceClass, NULL);
    if (!ServiceClassA)
      goto fail;
  }
  if (ServiceName)
  {
    ServiceNameA = ConvertWCharToUtf8Alloc(ServiceName, NULL);
    if (!ServiceNameA)
      goto fail;
  }
  if (InstanceName)
  {
    InstanceNameA = ConvertWCharToUtf8Alloc(InstanceName, NULL);
    if (!InstanceNameA)
      goto fail;
  }
  if (Referrer)
  {
    ReferrerA = ConvertWCharToUtf8Alloc(Referrer, NULL);
    if (!ReferrerA)
      goto fail;
  }
  res = DsMakeSpnA(ServiceClassA, ServiceNameA, InstanceNameA, InstancePort, ReferrerA,
                   pcSpnLength, pszSpnA);
 
  if (res == ERROR_SUCCESS)
  {
    if (ConvertUtf8NToWChar(pszSpnA, *pcSpnLength, pszSpn, length) < 0)
      res = ERROR_OUTOFMEMORY;
  }
 
fail:
  free(ServiceClassA);
  free(ServiceNameA);
  free(InstanceNameA);
  free(ReferrerA);
  free(pszSpnA);
  return res;
}
 
DWORD DsMakeSpnA(LPCSTR ServiceClass, LPCSTR ServiceName, LPCSTR InstanceName, USHORT InstancePort,
                 LPCSTR Referrer, DWORD* pcSpnLength, LPSTR pszSpn)
{
  DWORD SpnLength = 0;
  DWORD ServiceClassLength = 0;
  DWORD ServiceNameLength = 0;
 
  WINPR_ASSERT(ServiceClass);
  WINPR_ASSERT(ServiceName);
  WINPR_ASSERT(pcSpnLength);
 
  WINPR_UNUSED(InstanceName);
  WINPR_UNUSED(InstancePort);
  WINPR_UNUSED(Referrer);
 
  if ((*pcSpnLength != 0) && (pszSpn == NULL))
    return ERROR_INVALID_PARAMETER;
 
  ServiceClassLength = (DWORD)strlen(ServiceClass);
  ServiceNameLength = (DWORD)strlen(ServiceName);
 
  SpnLength = ServiceClassLength + 1 + ServiceNameLength + 1;
 
  if ((*pcSpnLength == 0) || (*pcSpnLength < SpnLength))
  {
    *pcSpnLength = SpnLength;
    return ERROR_BUFFER_OVERFLOW;
  }
 
  (void)sprintf_s(pszSpn, *pcSpnLength, "%s/%s", ServiceClass, ServiceName);
 
  return ERROR_SUCCESS;
}
 
#endif