1 files changed, 216 insertions, 0 deletions

diff --git a/private/oleutest/trktest/trktest.cxx b/private/oleutest/trktest/trktest.cxx
new file mode 100644
index 000000000..ab4d41003
--- /dev/null
+++ b/private/oleutest/trktest/trktest.cxx
@@ -0,0 +1,216 @@
+#include <windows.h>
+#include <stdlib.h>
+#include <stdio.h>
+
+WCHAR *Convert(char *psz)
+{	int c;
+	WCHAR *pwsz = new WCHAR [ c = strlen(psz) + 1 ];
+	if (pwsz)
+		mbstowcs(pwsz, psz, c);
+	return pwsz;
+}
+
+#define CHECKNE(hr, stat, string, errRet) \
+if (hr != stat) { iLine = __LINE__; pszErr = string; goto errRet; }
+
+HRESULT TouchFile(WCHAR *pwszSource)
+{
+    FILETIME ftc;
+    SYSTEMTIME st;
+
+    GetSystemTime(&st);
+    SystemTimeToFileTime(&st, &ftc);
+    return StgSetTimes(pwszSource, &ftc, NULL, NULL);
+}
+
+HRESULT CreateLink(WCHAR *pwszClient, WCHAR *pwszSource)
+{
+	HRESULT hr;
+	IBindCtx *pbc=NULL;
+	ULONG ulEaten;
+	IMoniker *pmkSource=NULL;
+	IStorage *pstg=NULL;
+	IOleObject *poo=NULL;
+	IMoniker *pmkThis=NULL;
+	IMoniker *pmkLink=NULL;
+	IPersistStorage *pps=NULL;
+	char *pszErr;
+	int iLine=0;
+
+	hr = CreateFileMoniker(pwszClient, &pmkThis);
+	CHECKNE(hr, S_OK, "CreateFileMoniker", errRet);
+	hr = CreateItemMoniker(L"!", L"Link", &pmkLink);
+	CHECKNE(hr, S_OK, "CreateItemMoniker", errRet);
+	hr = StgCreateDocfile(pwszClient, STGM_CREATE | STGM_READWRITE | STGM_SHARE_EXCLUSIVE,
+		0, &pstg);
+	CHECKNE(hr, S_OK, "StgCreateDocfile", errRet);
+	hr = CreateBindCtx(0, &pbc);
+	CHECKNE(hr, S_OK, "CreateBindCtx", errRet);
+	hr = MkParseDisplayName(pbc, pwszSource, &ulEaten, &pmkSource);
+	CHECKNE(hr, S_OK, "MkParseDisplayName", errRet);
+	hr = OleCreateLink(pmkSource, IID_IOleObject, OLERENDER_NONE, NULL, NULL, pstg, (void**)&poo);
+	CHECKNE(hr, S_OK, "OleCreateLink", errRet);
+	hr = poo->SetMoniker(OLEWHICHMK_CONTAINER, pmkThis);
+	CHECKNE(hr, S_OK, "SetMonikerOLEWHICHMK_CONTAINER", errRet);
+	hr = poo->SetMoniker(OLEWHICHMK_OBJREL, pmkLink);
+	CHECKNE(hr, S_OK, "SetMonikerOLEWHICHMK_OBJREL", errRet);
+	hr = poo->QueryInterface(IID_IPersistStorage, (void**)&pps);
+	CHECKNE(hr, S_OK, "QueryInterface(IID_IPersistStorage", errRet);
+	hr = OleSave(pps, pstg, TRUE);
+	CHECKNE(hr, S_OK, "OleSave", errRet);
+
+errRet:
+	if (hr != S_OK)
+            printf("CreateLink at %s line %d with hr=%08x\n", pszErr, iLine, hr);
+
+	if (pbc) pbc->Release();
+	if (pmkSource) pmkSource->Release();
+	if (pstg) pstg->Release();
+	if (poo) poo->Release();
+	if (pmkThis) pmkThis->Release();
+	if (pmkLink) pmkLink->Release();
+	if (pps) pps->Release();
+
+	return hr;
+}
+
+HRESULT ResolveLink(WCHAR *pwszClient, WCHAR *pwszSource, BOOL fReSave)
+{
+	HRESULT hr;
+	IStorage *pstg = NULL;
+	IOleLink *pol = NULL;
+	WCHAR *pwszSourceFound = NULL;
+	IPersistStorage *pps = NULL;
+	char *pszErr;
+	int iLine;
+
+	hr = StgOpenStorage(pwszClient, NULL, STGM_READWRITE | STGM_SHARE_EXCLUSIVE, 
+		NULL, 0, &pstg);
+	CHECKNE(hr, S_OK, "StgOpenStorage(pwszClient", errRet);
+	hr = OleLoad(pstg, IID_IOleLink, NULL, (void**)&pol);
+	CHECKNE(hr, S_OK, "OleLoad(pstg, IID_IOleLink", errRet);
+	hr = pol->BindToSource(0, NULL);
+	CHECKNE(hr, S_OK, "pol->BindToSource", errRet);
+	hr = pol->GetSourceDisplayName(&pwszSourceFound);
+	CHECKNE(hr, S_OK, "pol->GetSourceDisplayName", errRet);
+	if (wcsicmp(pwszSourceFound, pwszSource) != 0)
+	{
+		printf("SourceFound=%ls, SourceExpected=%ls\n", pwszSourceFound, pwszSource);
+                pszErr = "'path compare'";
+		hr = E_FAIL;
+		goto errRet;
+	}
+
+	if (fReSave)
+	{
+		hr = pol->QueryInterface(IID_IPersistStorage, (void**)&pps);
+		CHECKNE(hr, S_OK, "pol->QueryInterface(IID_IPersistStorage", errRet);
+		hr = OleSave(pps, pstg, TRUE);
+		CHECKNE(hr, S_OK, "OleSave(pps, pstg, TRUE)", errRet);
+	}
+
+errRet:
+	if (hr != S_OK)
+            printf("ResolveLink at %s line %d with hr=%08x\n", pszErr, iLine, hr);
+
+	CoTaskMemFree(pwszSourceFound);
+	if (pol) pol->Release();
+	if (pstg) pstg->Release();
+	if (pps) pps->Release();
+	return hr;
+}
+
+int __cdecl main(int argc, char **argv)
+{
+	char * pszExit = "TEST FAILED.";
+	int iExit = 1;
+	WCHAR *pwszClient = NULL;
+	WCHAR *pwszSource = NULL;
+	HRESULT hr=E_FAIL;
+	BOOL fArgsOk = FALSE;
+        HRESULT hrExpected=S_OK;
+
+	CoInitialize(NULL);
+	OleInitialize(NULL);
+
+        if (argc >= 3)
+        {
+		pwszClient = Convert(argv[2]);
+        }
+
+        if (argc >= 5)
+        {
+		pwszSource = Convert(argv[3]);
+                sscanf(argv[4], "%x", &hrExpected);
+		if (!pwszClient || !pwszSource)
+		{
+			pszExit = "Out of memory.";
+			goto errExit;
+		}
+        }
+
+        if (argc == 3)
+        {
+		if (stricmp(argv[1], "/t") == 0 || stricmp(argv[1], "-t") == 0)
+		{
+			// touch file creation date
+			hr = TouchFile(pwszClient);
+			fArgsOk = TRUE;
+		}
+        }
+        else
+	if (argc == 5)
+	{
+		if (stricmp(argv[1], "/c") == 0 || stricmp(argv[1], "-c") == 0)
+		{
+			// create link
+			hr = CreateLink(pwszClient, pwszSource);
+			fArgsOk = TRUE;
+		}
+	}
+	else
+	if (argc == 6)
+	{
+		if (stricmp(argv[1], "/r") == 0 || stricmp(argv[1], "-r") == 0)
+		{
+			// resolve link
+			hr = ResolveLink(pwszClient, pwszSource, argv[5][0] == 's');
+			fArgsOk = TRUE;
+		}
+	}
+
+	if (!fArgsOk)
+	{
+		pszExit = "Usage: /c client source expected_status -- create .lts file linked to source\n"
+			  "       /r client expected_source expected_status save/nosave  -- resolve .lts file, checked linked to source\n"
+                          "       expected_source is ignored(but required) if not found during resolve.\n"
+                          "       /t source -- change the creation date\n";
+
+	}
+        else
+        {
+            if (hrExpected != hr)
+            {
+                printf("Expected status = %08X, actual status = %08X\n",
+                    hrExpected,
+                    hr);
+            }
+            else
+            {
+                pszExit = "TEST PASSED.";
+                iExit = 0;
+            }
+        }
+	
+errExit:
+
+	OleUninitialize();
+	CoUninitialize();
+
+	delete pwszClient;
+	delete pwszSource;
+
+	puts(pszExit);
+
+	return iExit;
+}