WolfSSL
/
wolfssl
mirror of https://github.com/wolfSSL/wolfssl.git
1
0
Fork 0
Code Issues Releases Wiki Activity

C# Wrapper : TCP check connection termination

pull/583/head
Jacob Barthelmeh 2016-09-28 15:00:30 -06:00
parent be86308a33
commit 1ed06b53df
1 changed files with 21 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
wrapper/CSharp/wolfSSL_CSharp/wolfSSL.cs
View File

@ -32,6 +32,9 @@ namespace wolfSSL.CSharp {
{ {
private const string wolfssl_dll = "wolfssl.dll"; private const string wolfssl_dll = "wolfssl.dll";
/* wait for 6 seconds default on TCP socket state poll if timeout not set */
private const int WC_WAIT = 6000000;
/******************************** /********************************
* Class for DTLS connections * Class for DTLS connections
*/ */
@ -336,9 +339,17 @@ namespace wolfSSL.CSharp {
System.Runtime.InteropServices.GCHandle gch; System.Runtime.InteropServices.GCHandle gch;
gch = GCHandle.FromIntPtr(ctx); gch = GCHandle.FromIntPtr(ctx);
Socket con = (System.Net.Sockets.Socket)gch.Target; Socket con = (System.Net.Sockets.Socket)gch.Target;
Byte[] msg = new Byte[sz]; Byte[] msg = new Byte[sz];
amtRecv = con.Receive(msg, msg.Length, 0); amtRecv = con.Receive(msg, msg.Length, 0);
if (amtRecv == 0)
{
/* No data received so check for a response to see if connection is still open */
if (con.Poll((con.ReceiveTimeout > 0) ? con.ReceiveTimeout : WC_WAIT, SelectMode.SelectRead))
{
log(ERROR_LOG, "socket connection issue, suspected connection termination.");
return wolfssl.CBIO_ERR_CONN_CLOSE;
}
}
Marshal.Copy(msg, 0, buf, sz); Marshal.Copy(msg, 0, buf, sz);
} }
catch (Exception e) catch (Exception e)
@ -373,10 +384,17 @@ namespace wolfSSL.CSharp {
gch = GCHandle.FromIntPtr(ctx); gch = GCHandle.FromIntPtr(ctx);
Socket con = (System.Net.Sockets.Socket)gch.Target; Socket con = (System.Net.Sockets.Socket)gch.Target;
Byte[] msg = new Byte[sz]; Byte[] msg = new Byte[sz];
Marshal.Copy(buf, msg, 0, sz); Marshal.Copy(buf, msg, 0, sz);
con.Send(msg, 0, msg.Length, SocketFlags.None); if (con.Send(msg, 0, msg.Length, SocketFlags.None) == 0 && sz !=0)
{
/* no data sent and msg size is larger then 0, check for lost connection */
if (con.Poll((con.SendTimeout > 0) ? con.SendTimeout : WC_WAIT, SelectMode.SelectWrite))
{
log(ERROR_LOG, "socket connection issue, suspect connection termination");
return wolfssl.CBIO_ERR_CONN_CLOSE;
}
}
return sz; return sz;
} }
catch (Exception e) catch (Exception e)