138 lines
3.3 KiB
C
138 lines
3.3 KiB
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;
|
||
|
TCHAR Path[PATHCCH_MAX_CCH];
|
||
|
|
||
|
/* Path: no extension, Extension: dot */
|
||
|
|
||
|
_tcscpy(Path, testPathNoExtension);
|
||
|
|
||
|
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 (_tcscmp(Path, testPathExtension) != 0)
|
||
|
{
|
||
|
_tprintf(_T("Path Mismatch: Actual: %s, Expected: %s\n"), Path, testPathExtension);
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
/* Path: no extension, Extension: no dot */
|
||
|
|
||
|
_tcscpy(Path, testPathNoExtension);
|
||
|
|
||
|
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 (_tcscmp(Path, testPathExtension) != 0)
|
||
|
{
|
||
|
_tprintf(_T("Path Mismatch: Actual: %s, Expected: %s\n"), Path, testPathExtension);
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
/* Path: extension, Extension: dot */
|
||
|
|
||
|
_tcscpy(Path, testPathExtension);
|
||
|
|
||
|
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 (_tcscmp(Path, testPathExtension) != 0)
|
||
|
{
|
||
|
_tprintf(_T("Path Mismatch: Actual: %s, Expected: %s\n"), Path, testPathExtension);
|
||
|
return -1;
|
||
|
}
|
||
|
|
||
|
/* Path: extension, Extension: no dot */
|
||
|
|
||
|
_tcscpy(Path, testPathExtension);
|
||
|
|
||
|
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 (_tcscmp(Path, testPathExtension) != 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 */
|
||
|
|
||
|
_tcscpy(Path, _T("C:\\456789"));
|
||
|
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 */
|
||
|
|
||
|
_tcscpy(Path, _T("C:\\456789"));
|
||
|
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;
|
||
|
}
|