TestPathCchAddExtension.c

 
#include <stdio.h>
#include <winpr/crt.h>
#include <winpr/path.h>
#include <winpr/tchar.h>
#include <winpr/winpr.h>
 
static const TCHAR testExtDot[] = _T(".exe");
static const TCHAR testExtNoDot[] = _T("exe");
static const TCHAR testPathNoExtension[] = _T("C:\\Windows\\System32\\cmd");
static const TCHAR testPathExtension[] = _T("C:\\Windows\\System32\\cmd.exe");
 
int TestPathCchAddExtension(int argc, char* argv[])
{
  HRESULT status = 0;
  TCHAR Path[PATHCCH_MAX_CCH] = { 0 };
 
  WINPR_UNUSED(argc);
  WINPR_UNUSED(argv);
 
  /* Path: no extension, Extension: dot */
 
  _tcsncpy(Path, testPathNoExtension, ARRAYSIZE(Path));
 
  status = PathCchAddExtension(Path, PATHCCH_MAX_CCH, testExtDot);
 
  if (status != S_OK)
  {
    _tprintf(_T("PathCchAddExtension status: 0x%08") _T(PRIX32) _T("\n"), status);
    return -1;
  }
 
  if (_tcsncmp(Path, testPathExtension, ARRAYSIZE(Path)) != 0)
  {
    _tprintf(_T("Path Mismatch: Actual: %s, Expected: %s\n"), Path, testPathExtension);
    return -1;
  }
 
  /* Path: no extension, Extension: no dot */
 
  _tcsncpy(Path, testPathNoExtension, ARRAYSIZE(Path));
 
  status = PathCchAddExtension(Path, PATHCCH_MAX_CCH, testExtNoDot);
 
  if (status != S_OK)
  {
    _tprintf(_T("PathCchAddExtension status: 0x%08") _T(PRIX32) _T("\n"), status);
    return -1;
  }
 
  if (_tcsncmp(Path, testPathExtension, ARRAYSIZE(Path)) != 0)
  {
    _tprintf(_T("Path Mismatch: Actual: %s, Expected: %s\n"), Path, testPathExtension);
    return -1;
  }
 
  /* Path: extension, Extension: dot */
 
  _tcsncpy(Path, testPathExtension, ARRAYSIZE(Path));
 
  status = PathCchAddExtension(Path, PATHCCH_MAX_CCH, testExtDot);
 
  if (status != S_FALSE)
  {
    _tprintf(_T("PathCchAddExtension status: 0x%08") _T(PRIX32) _T("\n"), status);
    return -1;
  }
 
  if (_tcsncmp(Path, testPathExtension, ARRAYSIZE(Path)) != 0)
  {
    _tprintf(_T("Path Mismatch: Actual: %s, Expected: %s\n"), Path, testPathExtension);
    return -1;
  }
 
  /* Path: extension, Extension: no dot */
 
  _tcsncpy(Path, testPathExtension, ARRAYSIZE(Path));
 
  status = PathCchAddExtension(Path, PATHCCH_MAX_CCH, testExtDot);
 
  if (status != S_FALSE)
  {
    _tprintf(_T("PathCchAddExtension status: 0x%08") _T(PRIX32) _T("\n"), status);
    return -1;
  }
 
  if (_tcsncmp(Path, testPathExtension, ARRAYSIZE(Path)) != 0)
  {
    _tprintf(_T("Path Mismatch: Actual: %s, Expected: %s\n"), Path, testPathExtension);
    return -1;
  }
 
  /* Path: NULL */
 
  status = PathCchAddExtension(NULL, PATHCCH_MAX_CCH, testExtDot);
  if (status != E_INVALIDARG)
  {
    _tprintf(_T("PathCchAddExtension with null buffer returned status: 0x%08") _T(
                 PRIX32) _T(" (expected E_INVALIDARG)\n"),
             status);
    return -1;
  }
 
  /* Extension: NULL */
 
  status = PathCchAddExtension(Path, PATHCCH_MAX_CCH, NULL);
  if (status != E_INVALIDARG)
  {
    _tprintf(_T("PathCchAddExtension with null extension returned status: 0x%08") _T(
                 PRIX32) _T(" (expected E_INVALIDARG)\n"),
             status);
    return -1;
  }
 
  /* Insufficient Buffer size */
 
  _tcsncpy(Path, _T("C:\\456789"), ARRAYSIZE(Path));
  status = PathCchAddExtension(Path, 9 + 4, _T(".jpg"));
  if (SUCCEEDED(status))
  {
    _tprintf(_T("PathCchAddExtension with insufficient buffer unexpectedly succeeded with ")
             _T("status: 0x%08") _T(PRIX32) _T("\n"),
             status);
    return -1;
  }
 
  /* Minimum required buffer size */
 
  _tcsncpy(Path, _T("C:\\456789"), ARRAYSIZE(Path));
  status = PathCchAddExtension(Path, 9 + 4 + 1, _T(".jpg"));
  if (FAILED(status))
  {
    _tprintf(_T("PathCchAddExtension with sufficient buffer unexpectedly failed with status: ")
             _T("0x%08") _T(PRIX32) _T("\n"),
             status);
    return -1;
  }
 
  return 0;
}