Logs

Red4Sec · Nov 28, 2019 · 70023ee · 70023ee
1 parent 6668377
commit 70023ee
Show file tree

Hide file tree

Showing 5 changed files with 88 additions and 7 deletions.
diff --git a/src/TuringMachine.Core/Helpers/EnumeratorHelper.cs b/src/TuringMachine.Core/Helpers/EnumeratorHelper.cs
@@ -0,0 +1,23 @@
+using System.Collections.Generic;
+using System.Linq;
+
+namespace TuringMachine.Core.Helpers
+{
+	public static class EnumeratorHelper
+	{
+		/// <summary>
+		/// To arryy
+		/// </summary>
+		/// <typeparam name="T">Type</typeparam>
+		/// <param name="index">Index</param>
+		/// <param name="count">Count</param>
+		/// <returns>Array</returns>
+		public static T[] ToArray<T>(this IEnumerable<T> input, int index = -1, int count = -1)
+		{
+			if (index > 0) input = input.Skip(index);
+			if (count >= 0) input = input.Take(count);
+
+			return Enumerable.ToArray(input);
+		}
+	}
+}
diff --git a/src/TuringMachine/RPC/Controllers/ConfigsController.cs b/src/TuringMachine/RPC/Controllers/ConfigsController.cs
@@ -2,6 +2,7 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
+using TuringMachine.Core.Helpers;
 using TuringMachine.Core.Interfaces;
 using TuringMachine.Core.Logs;
 
@@ -42,10 +43,16 @@ public ActionResult<bool> Remove(Guid id)
 			return false;
 		}
 
+		[HttpGet("count")]
+		public ActionResult<int> Count()
+		{
+			return _server.Server.Configurations.Count;
+		}
+
 		[HttpGet("all")]
-		public ActionResult<IEnumerable<FuzzingConfigBase>> GetAll()
+		public ActionResult<IEnumerable<FuzzingConfigBase>> GetAll(int index = -1, int count = -1)
 		{
-			return _server.Server.Configurations.Values.Select(u => u.Source).ToArray();
+			return _server.Server.Configurations.Values.Select(u => u.Source).ToArray(index, count);
 		}
 
 		[HttpGet("stat")]

diff --git a/src/TuringMachine/RPC/Controllers/ConnectionsController.cs b/src/TuringMachine/RPC/Controllers/ConnectionsController.cs
@@ -1,8 +1,9 @@
-using Microsoft.AspNetCore.Mvc;
+using Microsoft.AspNetCore.Mvc;
 using System;
 using System.Collections.Generic;
 using System.Linq;
 using TuringMachine.Core.Fuzzers;
+using TuringMachine.Core.Helpers;
 using TuringMachine.Core.Logs;
 
 namespace TuringMachine.RPC.Controllers
@@ -23,10 +24,16 @@ public ConnectionsController(RpcServer server)
 			_server = server;
 		}
 
+		[HttpGet("count")]
+		public ActionResult<int> Count()
+		{
+			return _server.Server.Connections.Count;
+		}
+
 		[HttpGet("all")]
-		public ActionResult<IEnumerable<FuzzerClientInfo>> GetAll()
+		public ActionResult<IEnumerable<FuzzerClientInfo>> GetAll(int index = -1, int count = -1)
 		{
-			return _server.Server.Connections.Values.Select(u => u.Source).ToArray();
+			return _server.Server.Connections.Values.Select(u => u.Source).ToArray(index, count);
 		}
 
 		[HttpGet("stat")]

diff --git a/src/TuringMachine/RPC/Controllers/InputsController.cs b/src/TuringMachine/RPC/Controllers/InputsController.cs
@@ -2,6 +2,7 @@
 using System;
 using System.Collections.Generic;
 using System.Linq;
+using TuringMachine.Core.Helpers;
 using TuringMachine.Core.Interfaces;
 using TuringMachine.Core.Logs;
 
@@ -42,10 +43,16 @@ public ActionResult<bool> Remove(Guid id)
 			return false;
 		}
 
+		[HttpGet("count")]
+		public ActionResult<int> Count()
+		{
+			return _server.Server.Inputs.Count;
+		}
+
 		[HttpGet("all")]
-		public ActionResult<IEnumerable<FuzzingInputBase>> GetAll()
+		public ActionResult<IEnumerable<FuzzingInputBase>> GetAll(int index = -1, int count = -1)
 		{
-			return _server.Server.Inputs.Values.Select(u => u.Source).ToArray();
+			return _server.Server.Inputs.Values.Select(u => u.Source).ToArray(index, count);
 		}
 
 		[HttpGet("stat")]

diff --git a/src/TuringMachine/RPC/Controllers/ResultsController.cs b/src/TuringMachine/RPC/Controllers/ResultsController.cs
@@ -0,0 +1,37 @@
+using Microsoft.AspNetCore.Mvc;
+using System.Collections.Generic;
+using System.Linq;
+using TuringMachine.Core.Helpers;
+using TuringMachine.Core.Logs;
+
+namespace TuringMachine.RPC.Controllers
+{
+	[ApiController, Route("results")]
+	public class ResultsController : ControllerBase
+	{
+		/// <summary>
+		/// Server
+		/// </summary>
+		private readonly RpcServer _server;
+
+		/// <summary>
+		/// Controller
+		/// </summary>
+		public ResultsController(RpcServer server)
+		{
+			_server = server;
+		}
+
+		[HttpGet("count")]
+		public ActionResult<int> Count()
+		{
+			return _server.Server.Logs.Count;
+		}
+
+		[HttpGet("all")]
+		public ActionResult<IEnumerable<FuzzerLog>> GetAll(int index = -1, int count = -1)
+		{
+			return _server.Server.Logs.Values.ToArray(index, count);
+		}
+	}
+}