prototype of device database api

Author: azenla

protect/control/v1/common.proto

@@ -151,7 +151,7 @@ message AnnotationSpec {
 }
 
 message DeviceReferenceSpec {
-  string name = 1;
+  string id = 1;
 }
 
 message WorkloadBlockDeviceSpec {
@@ -489,10 +489,41 @@ message OciImageProgressIndicationCompleted {
   bool is_bytes = 3;
 }
 
-message DeviceInfo {
+message PciDeviceSpec {
+  string location = 1;
+  bool permissive = 2;
+}
+
+message BlockDeviceSpec {
+  string device_path = 1;
+}
+
+message DeviceSpec {
   string name = 1;
-  bool claimed = 2;
-  string owner_zone = 3;
+  oneof device_type {
+    PciDeviceSpec pci = 2;
+    BlockDeviceSpec block = 3;
+  }
+  repeated string modules = 4;
+  repeated KernelModuleParameter module_parameters = 5;
+}
+
+enum DeviceState {
+  DEVICE_STATE_UNKNOWN = 0;
+  DEVICE_STATE_AVAILABLE = 1;
+  DEVICE_STATE_TAINTED = 2;
+  DEVICE_STATE_USED = 3;
+}
+
+message DeviceStatus {
+  DeviceState state = 1;
+  string zone_id = 2;
+}
+
+message Device {
+  string id = 1;
+  DeviceSpec spec = 2;
+  DeviceStatus status = 3;
 }
 
 message ZoneScratchDiskSpec {
@@ -564,14 +595,3 @@ message OciRegistryAuthentication {
     string registry_token = 4;
   }
 }
-
-message PciDevice {
-  PciDeviceState state = 1;
-}
-
-enum PciDeviceState {
-  PCI_DEVICE_STATE_CLEAN = 0;
-  PCI_DEVICE_STATE_DIRTY = 1;
-  PCI_DEVICE_STATE_SCRUB = 2;
-  PCI_DEVICE_STATE_BUSY = 3;
-}

protect/control/v1/control.proto

@@ -9,7 +9,11 @@ service ControlService {
   rpc SnoopIdm(SnoopIdmRequest) returns (stream SnoopIdmReply);
   rpc GetHostCpuTopology(GetHostCpuTopologyRequest) returns (GetHostCpuTopologyReply);
 
+  rpc CreateDevice(CreateDeviceRequest) returns (CreateDeviceReply);
   rpc ListDevices(ListDevicesRequest) returns (ListDevicesReply);
+  rpc UpdateDeviceState(UpdateDeviceStateRequest) returns (UpdateDeviceStateReply);
+  rpc DestroyDevice(DestroyDeviceRequest) returns (DestroyDeviceReply);
+  rpc ResolveDeviceIds(ResolveDeviceIdsRequest) returns (ResolveDeviceIdsReply);
 
   rpc CreateNetworkReservation(CreateNetworkReservationRequest) returns (CreateNetworkReservationReply);
   rpc DestroyNetworkReservation(DestroyNetworkReservationRequest) returns (DestroyNetworkReservationReply);
@@ -55,7 +59,6 @@ service ControlService {
   rpc SetHostPowerManagementPolicy(SetHostPowerManagementPolicyRequest) returns (SetHostPowerManagementPolicyReply);
 
   rpc DialNetworkSocket(stream DialNetworkSocketRequest) returns (stream DialNetworkSocketReply);
-  rpc DeviceStateUpdate(DeviceStateRequest) returns (DeviceStateReply);
 }
 
 message GetHostStatusRequest {}
@@ -70,12 +73,12 @@ message GetHostStatusReply {
   optional uint64 hyp_free_mem = 7;
 }
 
-message DeviceStateRequest {
-  string name = 1;
-  PciDevice state = 2;
+message UpdateDeviceStateRequest {
+  string id = 1;
+  DeviceState state = 2;
 }
 
-message DeviceStateReply {}
+message UpdateDeviceStateReply {}
 
 message CreateZoneRequest {
   ZoneSpec spec = 1;
@@ -415,7 +418,7 @@ message WorkloadChangedEvent {
 }
 
 message DeviceChangedEvent {
-  DeviceReferenceSpec device = 1;
+  Device device = 1;
 }
 
 message ReadZoneMetricsRequest {
@@ -489,7 +492,8 @@ message ListImagesReply {
 message ListDevicesRequest {}
 
 message ListDevicesReply {
-  repeated DeviceInfo devices = 1;
+  reserved 1;
+  repeated Device devices = 2;
 }
 
 message GetHostCpuTopologyRequest {}
@@ -626,3 +630,25 @@ message DialNetworkSocketData {
 message DialNetworkSocketReply {
   DialNetworkSocketData data = 1;
 }
+
+message CreateDeviceRequest {
+  DeviceSpec spec = 1;
+}
+
+message CreateDeviceReply {
+  string id = 1;
+}
+
+message DestroyDeviceRequest {
+  string id = 1;
+}
+
+message DestroyDeviceReply {}
+
+message ResolveDeviceIdsRequest {
+  string name = 1;
+}
+
+message ResolveDeviceIdsReply {
+  string device_ids = 1;
+}