1 files changed, 57 insertions, 0 deletions
diff --git a/cloudinit/sources/azure/kvp.py b/cloudinit/sources/azure/kvp.py
new file mode 100644
index 00000000..735c4616
--- /dev/null
+++ b/cloudinit/sources/azure/kvp.py
@@ -0,0 +1,57 @@
+# Copyright (C) 2022 Microsoft Corporation.
+#
+# This file is part of cloud-init. See LICENSE file for license information.
+
+import logging
+from datetime import datetime
+from typing import Optional
+
+from cloudinit import version
+from cloudinit.reporting import handlers, instantiated_handler_registry
+from cloudinit.sources.azure import errors, identity
+
+LOG = logging.getLogger(__name__)
+
+
+def get_kvp_handler() -> Optional[handlers.HyperVKvpReportingHandler]:
+    """Get instantiated KVP telemetry handler."""
+    kvp_handler = instantiated_handler_registry.registered_items.get(
+        "telemetry"
+    )
+    if not isinstance(kvp_handler, handlers.HyperVKvpReportingHandler):
+        return None
+
+    return kvp_handler
+
+
+def report_via_kvp(report: str) -> bool:
+    """Report to host via PROVISIONING_REPORT KVP key."""
+    kvp_handler = get_kvp_handler()
+    if kvp_handler is None:
+        LOG.debug("KVP handler not enabled, skipping host report.")
+        return False
+
+    kvp_handler.write_key("PROVISIONING_REPORT", report)
+    return True
+
+
+def report_failure_to_host(error: errors.ReportableError) -> bool:
+    return report_via_kvp(error.as_encoded_report())
+
+
+def report_success_to_host() -> bool:
+    try:
+        vm_id = identity.query_vm_id()
+    except Exception as id_error:
+        vm_id = f"failed to read vm id: {id_error!r}"
+
+    report = errors.encode_report(
+        [
+            "result=success",
+            f"agent=Cloud-Init/{version.version_string()}",
+            f"timestamp={datetime.utcnow().isoformat()}",
+            f"vm_id={vm_id}",
+        ]
+    )
+
+    return report_via_kvp(report)