TestPathCchAddBackslash.c

 
#include <stdio.h>
#include <winpr/crt.h>
#include <winpr/path.h>
#include <winpr/tchar.h>
#include <winpr/winpr.h>
 
static const TCHAR testPathBackslash[] = _T("C:\\Program Files\\");
static const TCHAR testPathNoBackslash[] = _T("C:\\Program Files");
 
int TestPathCchAddBackslash(int argc, char* argv[])
{
  HRESULT status = 0;
  TCHAR Path[PATHCCH_MAX_CCH] = { 0 };
 
  WINPR_UNUSED(argc);
  WINPR_UNUSED(argv);
 
  _tcsncpy(Path, testPathNoBackslash, ARRAYSIZE(Path));
 
  /* Add a backslash to a path without a trailing backslash, expect S_OK */
 
  status = PathCchAddBackslash(Path, PATHCCH_MAX_CCH);
 
  if (status != S_OK)
  {
    _tprintf(_T("PathCchAddBackslash status: 0x%08") _T(PRIX32) _T("\n"), status);
    return -1;
  }
 
  if (_tcsncmp(Path, testPathBackslash, ARRAYSIZE(Path)) != 0)
  {
    _tprintf(_T("Path Mismatch: Actual: %s, Expected: %s\n"), Path, testPathBackslash);
    return -1;
  }
 
  /* Add a backslash to a path with a trailing backslash, expect S_FALSE */
 
  _tcsncpy(Path, testPathBackslash, ARRAYSIZE(Path));
 
  status = PathCchAddBackslash(Path, PATHCCH_MAX_CCH);
 
  if (status != S_FALSE)
  {
    _tprintf(_T("PathCchAddBackslash status: 0x%08") _T(PRIX32) _T("\n"), status);
    return -1;
  }
 
  if (_tcsncmp(Path, testPathBackslash, ARRAYSIZE(Path)) != 0)
  {
    _tprintf(_T("Path Mismatch: Actual: %s, Expected: %s\n"), Path, testPathBackslash);
    return -1;
  }
 
  /* Use NULL PSTR, expect FAILED(status) */
 
  status = PathCchAddBackslash(NULL, PATHCCH_MAX_CCH);
 
  if (SUCCEEDED(status))
  {
    _tprintf(
        _T("PathCchAddBackslash unexpectedly succeeded with null buffer. Status: 0x%08") _T(
            PRIX32) _T("\n"),
        status);
    return -1;
  }
 
  /* Use insufficient size value, expect FAILED(status)  */
 
  _tcsncpy(Path, _T("C:\\tmp"), ARRAYSIZE(Path));
 
  status = PathCchAddBackslash(Path, 7);
 
  if (SUCCEEDED(status))
  {
    _tprintf(_T("PathCchAddBackslash unexpectedly succeeded with insufficient buffer size. ")
             _T("Status: 0x%08") _T(PRIX32) _T("\n"),
             status);
    return -1;
  }
 
  /* Use minimum required size value, expect S_OK  */
 
  _tcsncpy(Path, _T("C:\\tmp"), ARRAYSIZE(Path));
 
  status = PathCchAddBackslash(Path, 8);
 
  if (status != S_OK)
  {
    _tprintf(_T("PathCchAddBackslash failed with status: 0x%08") _T(PRIX32) _T("\n"), status);
    return -1;
  }
 
  return 0;
}