1 files changed, 65 insertions, 0 deletions

diff --git a/src/balutil/inc/balretry.h b/src/balutil/inc/balretry.h
new file mode 100644
index 00000000..040ab4ae
--- /dev/null
+++ b/src/balutil/inc/balretry.h
@@ -0,0 +1,65 @@
+#pragma once
+// 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.
+
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+typedef enum BALRETRY_TYPE
+{
+    BALRETRY_TYPE_CACHE,
+    BALRETRY_TYPE_EXECUTE,
+} BALRETRY_TYPE;
+
+/*******************************************************************
+ BalRetryInitialize - initialize the retry count and timeout between
+                      retries (in milliseconds).
+********************************************************************/
+DAPI_(void) BalRetryInitialize(
+    __in DWORD dwMaxRetries,
+    __in DWORD dwTimeout
+    );
+
+/*******************************************************************
+ BalRetryUninitialize - call to cleanup any memory allocated during
+                        use of the retry utility.
+********************************************************************/
+DAPI_(void) BalRetryUninitialize();
+
+/*******************************************************************
+ BalRetryStartPackage - call when a package begins to be modified. If
+                        the package is being retried, the function will
+                        wait the specified timeout.
+********************************************************************/
+DAPI_(void) BalRetryStartPackage(
+    __in BALRETRY_TYPE type,
+    __in_z_opt LPCWSTR wzPackageId,
+    __in_z_opt LPCWSTR wzPayloadId
+    );
+
+/*******************************************************************
+ BalRetryErrorOccured - call when an error occurs for the retry utility
+                        to consider.
+********************************************************************/
+DAPI_(void) BalRetryErrorOccurred(
+    __in_z_opt LPCWSTR wzPackageId,
+    __in DWORD dwError
+    );
+
+/*******************************************************************
+ BalRetryEndPackage - returns IDRETRY is a retry is recommended or 
+                      IDNOACTION if a retry is not recommended.
+********************************************************************/
+DAPI_(HRESULT) BalRetryEndPackage(
+    __in BALRETRY_TYPE type,
+    __in_z_opt LPCWSTR wzPackageId,
+    __in_z_opt LPCWSTR wzPayloadId,
+    __in HRESULT hrError,
+    __inout BOOL* pfRetry
+    );
+
+
+#ifdef __cplusplus
+}
+#endif