From 87c4a5592f4c150b6f77aa9b35464de86ab9a15a Mon Sep 17 00:00:00 2001
From: Surbhi Kakarya <surbhi@edera.dev>
Date: Fri, 21 Nov 2025 20:01:27 +0000
Subject: [PATCH 1/3] This patch includes: 1. Device State: CLEAN, DIRTY,
 SCRUB, BUSY 2. DeviceStateUpdate RPC 3. DeviceChanged Event

Signed-off-by: Surbhi Kakarya <surbhi@edera.dev>
---
 protect/control/v1/common.proto  | 11 +++++++++++
 protect/control/v1/control.proto | 13 +++++++++++++
 2 files changed, 24 insertions(+)

diff --git a/protect/control/v1/common.proto b/protect/control/v1/common.proto
index 11c2f45..e91c9e4 100644
--- a/protect/control/v1/common.proto
+++ b/protect/control/v1/common.proto
@@ -532,3 +532,14 @@ message OciRegistryAuthentication {
     string registry_token = 4;
   }
 }
+
+message PciDevice {
+        PciDeviceState state = 1;
+}
+
+enum PciDeviceState {
+        CLEAN = 0;
+        DIRTY = 1;
+        SCRUB = 2;
+        BUSY = 3;
+}
diff --git a/protect/control/v1/control.proto b/protect/control/v1/control.proto
index e86f66d..2eb3416 100644
--- a/protect/control/v1/control.proto
+++ b/protect/control/v1/control.proto
@@ -55,6 +55,7 @@ service ControlService {
   rpc SetHostPowerManagementPolicy(SetHostPowerManagementPolicyRequest) returns (SetHostPowerManagementPolicyReply);
 
   rpc DialNetworkSocket(stream DialNetworkSocketRequest) returns (stream DialNetworkSocketReply);
+  rpc DeviceStateUpdate(DeviceStateRequest) returns (DeviceStateReply);
 }
 
 message GetHostStatusRequest {}
@@ -68,6 +69,13 @@ message GetHostStatusReply {
   string host_mac = 6;
 }
 
+message DeviceStateRequest {
+  string name = 1;
+  protect.common.v1.PciDevice state = 2;
+}
+
+message DeviceStateReply {}
+
 message CreateZoneRequest {
   ZoneSpec spec = 1;
 }
@@ -393,6 +401,7 @@ message WatchEventsReply {
   oneof event {
     ZoneChangedEvent zone_changed = 1;
     WorkloadChangedEvent workload_changed = 2;
+    DeviceChangedEvent device_changed = 3;
   }
 }
 
@@ -404,6 +413,10 @@ message WorkloadChangedEvent {
   Workload workload = 1;
 }
 
+message DeviceChangedEvent {
+  protect.common.v1.DeviceReferenceSpec device = 1;
+}
+
 message ReadZoneMetricsRequest {
   string zone_id = 1;
 }

From 622cf862645017a36ea9ce9ddf7a82f0696e538d Mon Sep 17 00:00:00 2001
From: Surbhi Kakarya <surbhi@edera.dev>
Date: Fri, 21 Nov 2025 20:21:54 +0000
Subject: [PATCH 2/3] Fixing the failure

Signed-off-by: Surbhi Kakarya <surbhi@edera.dev>
---
 protect/control/v1/control.proto | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/protect/control/v1/control.proto b/protect/control/v1/control.proto
index 2eb3416..dbc367d 100644
--- a/protect/control/v1/control.proto
+++ b/protect/control/v1/control.proto
@@ -71,7 +71,7 @@ message GetHostStatusReply {
 
 message DeviceStateRequest {
   string name = 1;
-  protect.common.v1.PciDevice state = 2;
+  PciDevice state = 2;
 }
 
 message DeviceStateReply {}
@@ -414,7 +414,7 @@ message WorkloadChangedEvent {
 }
 
 message DeviceChangedEvent {
-  protect.common.v1.DeviceReferenceSpec device = 1;
+  DeviceReferenceSpec device = 1;
 }
 
 message ReadZoneMetricsRequest {

From 8f8a16cb73e3baf7f457b654902f103b80f5d204 Mon Sep 17 00:00:00 2001
From: Surbhi Kakarya <surbhi@edera.dev>
Date: Fri, 21 Nov 2025 20:58:08 +0000
Subject: [PATCH 3/3] Fixing more failure

Signed-off-by: Surbhi Kakarya <surbhi@edera.dev>
---
 protect/control/v1/common.proto | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/protect/control/v1/common.proto b/protect/control/v1/common.proto
index e91c9e4..40fcff4 100644
--- a/protect/control/v1/common.proto
+++ b/protect/control/v1/common.proto
@@ -534,12 +534,12 @@ message OciRegistryAuthentication {
 }
 
 message PciDevice {
-        PciDeviceState state = 1;
+  PciDeviceState state = 1;
 }
 
 enum PciDeviceState {
-        CLEAN = 0;
-        DIRTY = 1;
-        SCRUB = 2;
-        BUSY = 3;
+  PCI_DEVICE_STATE_CLEAN = 0;
+  PCI_DEVICE_STATE_DIRTY = 1;
+  PCI_DEVICE_STATE_SCRUB = 2;
+  PCI_DEVICE_STATE_BUSY = 3;
 }