292 add a callback interval or next callback field to tavern

implants/lib/tavern/graphql/schema.graphql

@@ -17,6 +17,8 @@ type Beacon implements Node {
   agentIdentifier: String
   """Timestamp of when a task was last claimed or updated for the beacon."""
   lastSeenAt: Time
+  """The callback interval of the agent"""
+  interval: Int
   """Host this beacon is running on."""
   host: Host!
   """Tasks that have been assigned to the beacon."""
@@ -121,6 +123,17 @@ input BeaconWhereInput {
   lastSeenAtLTE: Time
   lastSeenAtIsNil: Boolean
   lastSeenAtNotNil: Boolean
+  """interval field predicates"""
+  interval: Int
+  intervalNEQ: Int
+  intervalIn: [Int!]
+  intervalNotIn: [Int!]
+  intervalGT: Int
+  intervalGTE: Int
+  intervalLT: Int
+  intervalLTE: Int
+  intervalIsNil: Boolean
+  intervalNotNil: Boolean
   """host edge predicates"""
   hasHost: Boolean
   hasHostWith: [HostWhereInput!]
@@ -880,6 +893,9 @@ UpdateBeaconInput is used for update Beacon object.
 Input was generated by ent.
 """
 input UpdateBeaconInput {
+  """The callback interval of the agent"""
+  interval: Int
+  clearInterval: Boolean
   hostID: ID
 }
 """

tavern/internal/c2/server.go

@@ -49,6 +49,7 @@ func (srv *Server) ClaimTasks(ctx context.Context, req *c2pb.ClaimTasksRequest)
 	if req.Beacon.Agent.Identifier == "" {
 		return nil, fmt.Errorf("must provide agent identifier")
 	}
+
 	hostPlaform := convertHostPlatform(req.Beacon.Host.Platform)
 
 	// 1. Upsert the host
@@ -72,6 +73,7 @@ func (srv *Server) ClaimTasks(ctx context.Context, req *c2pb.ClaimTasksRequest)
 		SetAgentIdentifier(req.Beacon.Agent.Identifier).
 		SetHostID(hostID).
 		SetLastSeenAt(now).
+		SetInterval(req.Beacon.Interval).
 		OnConflict().
 		UpdateNewValues().
 		ID(ctx)