1 files changed, 137 insertions, 0 deletions

diff --git a/src/dutil/inetutil.cpp b/src/dutil/inetutil.cpp
new file mode 100644
index 00000000..69a0176a
--- /dev/null
+++ b/src/dutil/inetutil.cpp
@@ -0,0 +1,137 @@
+// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.
+
+#include "precomp.h"
+
+
+/*******************************************************************
+ InternetGetSizeByHandle - returns size of file by url handle
+
+*******************************************************************/
+extern "C" HRESULT DAPI InternetGetSizeByHandle(
+    __in HINTERNET hiFile,
+    __out LONGLONG* pllSize
+    )
+{
+    Assert(pllSize);
+
+    HRESULT hr = S_OK;
+    DWORD dwSize;
+    DWORD cb;
+
+    cb = sizeof(dwSize);
+    if (!::HttpQueryInfoW(hiFile, HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER, reinterpret_cast<LPVOID>(&dwSize), &cb, NULL))
+    {
+        ExitOnLastError(hr, "Failed to get size for internet file handle");
+    }
+
+    *pllSize = dwSize;
+LExit:
+    return hr;
+}
+
+
+/*******************************************************************
+ InetGetCreateTimeByHandle - returns url creation time
+
+******************************************************************/
+extern "C" HRESULT DAPI InternetGetCreateTimeByHandle(
+    __in HINTERNET hiFile,
+    __out LPFILETIME pft
+    )
+{
+    Assert(pft);
+
+    HRESULT hr = S_OK;
+    SYSTEMTIME st = {0 };
+    DWORD cb = sizeof(SYSTEMTIME);
+
+    if (!::HttpQueryInfoW(hiFile, HTTP_QUERY_LAST_MODIFIED | HTTP_QUERY_FLAG_SYSTEMTIME, reinterpret_cast<LPVOID>(&st), &cb, NULL))
+    {
+        ExitWithLastError(hr, "failed to get create time for internet file handle");
+    }
+
+    if (!::SystemTimeToFileTime(&st, pft))
+    {
+        ExitWithLastError(hr, "failed to convert system time to file time");
+    }
+
+LExit:
+    return hr;
+}
+
+
+/*******************************************************************
+ InternetQueryInfoString - query info string
+
+*******************************************************************/
+extern "C" HRESULT DAPI InternetQueryInfoString(
+    __in HINTERNET hRequest,
+    __in DWORD dwInfo,
+    __deref_out_z LPWSTR* psczValue
+    )
+{
+    HRESULT hr = S_OK;
+    DWORD_PTR cbValue = 0;
+    DWORD dwIndex = 0;
+
+    // If nothing was provided start off with some arbitrary size.
+    if (!*psczValue)
+    {
+        hr = StrAlloc(psczValue, 64);
+        ExitOnFailure(hr, "Failed to allocate memory for value.");
+    }
+
+    hr = StrSize(*psczValue, &cbValue);
+    ExitOnFailure(hr, "Failed to get size of value.");
+
+    if (!::HttpQueryInfoW(hRequest, dwInfo, static_cast<void*>(*psczValue), reinterpret_cast<DWORD*>(&cbValue), &dwIndex))
+    {
+        DWORD er = ::GetLastError();
+        if (ERROR_INSUFFICIENT_BUFFER == er)
+        {
+            cbValue += sizeof(WCHAR); // add one character for the null terminator.
+
+            hr = StrAlloc(psczValue, cbValue / sizeof(WCHAR));
+            ExitOnFailure(hr, "Failed to allocate value.");
+
+            if (!::HttpQueryInfoW(hRequest, dwInfo, static_cast<void*>(*psczValue), reinterpret_cast<DWORD*>(&cbValue), &dwIndex))
+            {
+                er = ::GetLastError();
+            }
+            else
+            {
+                er = ERROR_SUCCESS;
+            }
+        }
+
+        hr = HRESULT_FROM_WIN32(er);
+        ExitOnRootFailure(hr, "Failed to get query information.");
+    }
+
+LExit:
+    return hr;
+}
+
+
+/*******************************************************************
+ InternetQueryInfoNumber - query info number
+
+*******************************************************************/
+extern "C" HRESULT DAPI InternetQueryInfoNumber(
+    __in HINTERNET hRequest,
+    __in DWORD dwInfo,
+    __out LONG* plInfo
+    )
+{
+    HRESULT hr = S_OK;
+    DWORD cbCode = sizeof(LONG);
+    DWORD dwIndex = 0;
+
+    if (!::HttpQueryInfoW(hRequest, dwInfo | HTTP_QUERY_FLAG_NUMBER, static_cast<void*>(plInfo), &cbCode, &dwIndex))
+    {
+        ExitWithLastError(hr, "Failed to get query information.");
+    }
+
+LExit:
+    return hr;
+}