add start_single to pro API manager

myalcin81 · Sep 2, 2014 · beef96a · beef96a
1 parent 6ac2f04
commit beef96a
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 6 deletions.
diff --git a/metasploit-sharp/MetasploitProManager.cs b/metasploit-sharp/MetasploitProManager.cs
@@ -187,7 +187,12 @@ public Dictionary<string, object> StartLootCollection(Dictionary<string, object>
 		{
 			return _session.Execute("pro.start_collect", options);
 		}
-
+
+		public Dictionary<string, object> StartSingle (Dictionary<string, object> options)
+		{
+			return _session.Execute("pro.start_single", options);
+		}
+
 		public Dictionary<string, object> StartReport(Dictionary<string, object> options)
 		{
 			return _session.Execute("pro.start_report", options);

diff --git a/metasploit-sharp/MetasploitSession.cs b/metasploit-sharp/MetasploitSession.cs
@@ -65,6 +65,7 @@ public Dictionary<string, object> Execute (string method, params object[] args)
 			request.KeepAlive = true;
 
 			Stream requestStream = request.GetRequestStream ();
+
 			Packer msgpackWriter = Packer.Create(requestStream);
 
 			msgpackWriter.PackArrayHeader (args.Length + 1 + (string.IsNullOrEmpty (_token) ? 0 : 1));
@@ -99,11 +100,13 @@ public Dictionary<string, object> Execute (string method, params object[] args)
 			}
 			}
 			catch (WebException ex) {
-				string res = string.Empty;
-				using (StreamReader rdr = new StreamReader(ex.Response.GetResponseStream()))
-					res = rdr.ReadToEnd();
+				if (ex.Response != null) {
+					string res = string.Empty;
+					using (StreamReader rdr = new StreamReader(ex.Response.GetResponseStream()))
+						res = rdr.ReadToEnd();
 
-				Console.WriteLine(res) ;
+					Console.WriteLine(res) ;
+				}
 			}
 
 			mstream.Position = 0;

diff --git a/metasploit-sharp/metasploit-sharp.csproj b/metasploit-sharp/metasploit-sharp.csproj
@@ -9,6 +9,7 @@
     <OutputType>Library</OutputType>
     <RootNamespace>metasploitsharp</RootNamespace>
     <AssemblyName>metasploit-sharp</AssemblyName>
+    <TargetFrameworkVersion>v4.5</TargetFrameworkVersion>
   </PropertyGroup>
   <PropertyGroup Condition=" '$(Configuration)|$(Platform)' == 'Debug|x86' ">
     <DebugSymbols>true</DebugSymbols>
@@ -41,7 +42,7 @@
   </ItemGroup>
   <Import Project="$(MSBuildBinPath)\Microsoft.CSharp.targets" />
   <ItemGroup>
-    <ProjectReference Include="..\..\msgpack\src\MsgPack\MsgPack.csproj">
+    <ProjectReference Include="..\..\msgpack-cli\src\MsgPack\MsgPack.csproj">
       <Project>{5BCEC32E-990E-4DE5-945F-BD27326A7418}</Project>
       <Name>MsgPack</Name>
     </ProjectReference>