jeff/NSspi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fixed namespaces; Working on server-side UI.

Browse Source
This commit is contained in:
antiduh
2014-06-25 21:49:35 +00:00
parent 2e6b847434
commit 81ed80a4d0
8 changed files with 205 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
NSspi.sln
View File

@@ -7,6 +7,8 @@ Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "TestClient", "TestClient\Te
EndProject
Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "TestProtocol", "TestProtocol\TestProtocol.csproj", "{9BFD94E1-D9FB-44D7-A6E7-8BAC2620E535}"
EndProject
Project("{FAE04EC0-301F-11D3-BF4B-00C04F79EFBC}") = "TestServer", "TestServer\TestServer.csproj", "{35B8280A-8EB1-4FB5-B448-B4E9F132317F}"
EndProject
Global
GlobalSection(SolutionConfigurationPlatforms) = preSolution
Debug|Any CPU = Debug|Any CPU
@@ -25,6 +27,10 @@ Global
{9BFD94E1-D9FB-44D7-A6E7-8BAC2620E535}.Debug|Any CPU.Build.0 = Debug|Any CPU
{9BFD94E1-D9FB-44D7-A6E7-8BAC2620E535}.Release|Any CPU.ActiveCfg = Release|Any CPU
{9BFD94E1-D9FB-44D7-A6E7-8BAC2620E535}.Release|Any CPU.Build.0 = Release|Any CPU
{35B8280A-8EB1-4FB5-B448-B4E9F132317F}.Debug|Any CPU.ActiveCfg = Debug|Any CPU
{35B8280A-8EB1-4FB5-B448-B4E9F132317F}.Debug|Any CPU.Build.0 = Debug|Any CPU
{35B8280A-8EB1-4FB5-B448-B4E9F132317F}.Release|Any CPU.ActiveCfg = Release|Any CPU
{35B8280A-8EB1-4FB5-B448-B4E9F132317F}.Release|Any CPU.Build.0 = Release|Any CPU
EndGlobalSection
GlobalSection(SolutionProperties) = preSolution
HideSolutionNode = FALSE

3
TestClient/Form1.cs
View File

@@ -11,9 +11,12 @@ using System.Windows.Forms;
using NSspi;
using NSspi.Contexts;
using NSspi.Credentials;
using TestProtocol;
namespace TestClient
{
using Message = TestProtocol.Message;
public partial class Form1 : Form
{
private ClientContext context;

2
TestProtocol/CustomConnection.cs
View File

@@ -8,7 +8,7 @@ using System.Threading.Tasks;
using System.Windows.Forms;
using NSspi;
namespace TestClient
namespace TestProtocol
{
public class CustomConnection
{

191
TestProtocol/CustomServer.cs Normal file
View File

@@ -0,0 +1,191 @@
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace TestProtocol
{
using System;
using System.Collections.Generic;
using System.Linq;
using System.Net.Sockets;
using System.Text;
using System.Threading;
using System.Threading.Tasks;
using System.Windows.Forms;
using NSspi;
public class CustomServer
{
private Thread receiveThread;
private Socket serverSocket;
private Socket readSocket;
private bool running;
public CustomServer()
{
this.running = false;
}
public delegate void ReceivedAction( Message message );
public event ReceivedAction Received;
public void StartServer( int port )
{
if( this.running )
{
throw new InvalidOperationException( "Already running" );
}
this.serverSocket = new Socket( SocketType.Stream, ProtocolType.Tcp );
this.running = true;
this.receiveThread = new Thread( ReceiveThreadEntry );
this.receiveThread.Name = "SSPI Server Receive Thread";
this.receiveThread.Start();
}
public void Stop()
{
if( this.running == false )
{
throw new InvalidOperationException( "Already stopped" );
}
this.serverSocket.Close();
if( this.readSocket != null )
{
this.readSocket.Close();
}
this.receiveThread.Join();
}
public void Send( Message message )
{
if( this.running == false )
{
throw new InvalidOperationException( "Not connected" );
}
byte[] outBuffer = new byte[message.Data.Length + 8];
ByteWriter.WriteInt32_BE( (int)message.Operation, outBuffer, 0 );
ByteWriter.WriteInt32_BE( message.Data.Length, outBuffer, 4 );
Array.Copy( message.Data, 0, outBuffer, 8, message.Data.Length );
this.serverSocket.Send( outBuffer, 0, outBuffer.Length, SocketFlags.None );
}
private void ReceiveThreadEntry()
{
try
{
while( this.running )
{
try
{
this.readSocket = this.serverSocket.Accept();
}
catch( SocketException e )
{
if( e.SocketErrorCode == SocketError.ConnectionAborted ||
e.SocketErrorCode == SocketError.Interrupted ||
e.SocketErrorCode == SocketError.OperationAborted ||
e.SocketErrorCode == SocketError.Shutdown )
{
// Shutting down.
break;
}
else
{
throw;
}
}
ReadLoop();
}
}
catch( Exception e )
{
MessageBox.Show( "The SspiConnection receive thread crashed:\r\n\r\n" + e.ToString() );
this.running = false;
}
}
private void ReadLoop()
{
byte[] readBuffer = new byte[65536];
ProtocolOp operation;
int length;
while( this.running )
{
try
{
// |--4 bytes--|--4 bytes--|---N--|
// Every command is a TLV - | Operation | Length | Data |
// Read the operation.
this.serverSocket.Receive( readBuffer, 4, SocketFlags.None );
// Check if we popped out of a receive call after we were shut down.
if( this.running == false ) { break; }
operation = (ProtocolOp)ByteWriter.ReadInt32_BE( readBuffer, 0 );
// Read the length
this.serverSocket.Receive( readBuffer, 4, SocketFlags.None );
length = ByteWriter.ReadInt32_BE( readBuffer, 0 );
// Read the data
this.serverSocket.Receive( readBuffer, length, SocketFlags.None );
}
catch( SocketException e )
{
if( e.SocketErrorCode == SocketError.ConnectionAborted ||
e.SocketErrorCode == SocketError.Interrupted ||
e.SocketErrorCode == SocketError.OperationAborted ||
e.SocketErrorCode == SocketError.Shutdown )
{
// Shutting down.
break;
}
else
{
throw;
}
}
if( this.Received != null )
{
byte[] dataCopy = new byte[length];
Array.Copy( readBuffer, 0, dataCopy, 0, length );
Message message = new Message( operation, dataCopy );
try
{
this.Received( message );
}
catch( Exception e )
{ }
}
}
}
}
}

2
TestProtocol/Message.cs
View File

@@ -4,7 +4,7 @@ using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace TestClient
namespace TestProtocol
{
public class Message
{

2
TestProtocol/ProtocolOp.cs
View File

@@ -4,7 +4,7 @@ using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace TestClient
namespace TestProtocol
{
public enum ProtocolOp : uint
{

1
TestProtocol/TestProtocol.csproj
View File

@@ -47,6 +47,7 @@
<Reference Include="System.Xml" />
</ItemGroup>
<ItemGroup>
<Compile Include="CustomServer.cs" />
<Compile Include="CustomConnection.cs" />
<Compile Include="Message.cs" />
<Compile Include="Properties\AssemblyInfo.cs" />

2
TestServer/TestServer.csproj
View File

@@ -4,7 +4,7 @@
<PropertyGroup>
<Configuration Condition=" '$(Configuration)' == '' ">Debug</Configuration>
<Platform Condition=" '$(Platform)' == '' ">AnyCPU</Platform>
<ProjectGuid>ccc5f303-782b-427d-855d-b86452c40033</ProjectGuid>
<ProjectGuid>{35B8280A-8EB1-4FB5-B448-B4E9F132317F}</ProjectGuid>
<OutputType>WinExe</OutputType>
<AppDesignerFolder>Properties</AppDesignerFolder>
<RootNamespace>TestServer</RootNamespace>