[feature] Include quest name on gRPC task

implants/imix/src/exec.rs

@@ -105,6 +105,7 @@ print(sys.shell(input_params["cmd"])["stdout"])
             .to_string(),
             parameters: HashMap::from([("cmd".to_string(), "echo hello_from_stdout".to_string())]),
             file_names: Vec::new(),
+            quest_name: "test_quest".to_string(),
         };
 
         let runtime = tokio::runtime::Builder::new_multi_thread()
@@ -134,6 +135,7 @@ aoeu
             .to_string(),
             parameters: HashMap::new(),
             file_names: Vec::new(),
+            quest_name: "test_quest".to_string(),
         };
 
         let runtime: tokio::runtime::Runtime = tokio::runtime::Builder::new_multi_thread()

implants/imix/src/tasks.rs

@@ -252,6 +252,7 @@ mod tests {
             eldritch: "print('okay')".to_string(),
             parameters: HashMap::from([("iter".to_string(), "3".to_string())]),
             file_names: Vec::new(),
+            quest_name: "test_quest".to_string(),
         }];
         start_new_tasks(new_tasks, &mut all_exec_futures, debug_start_time).await?;
         assert_eq!(all_exec_futures.len(), 1);

implants/lib/c2/src/c2.rs

@@ -97,6 +97,8 @@ pub struct Task {
     >,
     #[prost(string, repeated, tag = "4")]
     pub file_names: ::prost::alloc::vec::Vec<::prost::alloc::string::String>,
+    #[prost(string, tag = "5")]
+    pub quest_name: ::prost::alloc::string::String,
 }
 /// TaskError provides information when task execution fails.
 #[allow(clippy::derive_partial_eq_without_eq)]

tavern/internal/c2/api_claim_tasks.go

@@ -192,6 +192,7 @@ func (srv *Server) ClaimTasks(ctx context.Context, req *c2pb.ClaimTasksRequest)
 			Eldritch:   claimedTome.Eldritch,
 			Parameters: params,
 			FileNames:  claimedFileNames,
+			QuestName:  claimedQuest.Name,
 		})
 	}
 

tavern/internal/c2/c2.proto

@@ -48,6 +48,7 @@ message Task {
     string eldritch = 2;
     map<string, string> parameters = 3;
     repeated string file_names = 4;
+    string quest_name = 5;
 }
 
 // TaskError provides information when task execution fails.

tavern/internal/c2/c2test/ent.go

@@ -67,6 +67,10 @@ func ConvertTaskToC2PB(t *testing.T, ctx context.Context, task *ent.Task) *c2pb.
 			Eldritch,
 		Parameters: params,
 		FileNames:  fileNames,
+		QuestName: task.
+			QueryQuest().
+			OnlyX(ctx).
+			Name,
 	}
 }