162 lines
4.5 KiB
C#
162 lines
4.5 KiB
C#
// SocketEx.cs implementation by J. Arturo <webmaster at komodosoft dot net>
|
|
//
|
|
// This library is free software; you can redistribute it and/or
|
|
// modify it under the terms of the GNU Lesser General Public
|
|
// License as published by the Free Software Foundation; either
|
|
// version 2.1 of the License, or (at your option) any later version.
|
|
//
|
|
// This library is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
// Lesser General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU Lesser General Public
|
|
// License along with this library; if not, write to the Free Software
|
|
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
|
|
using System;
|
|
using System.Net;
|
|
using System.Net.Sockets;
|
|
using System.Threading;
|
|
|
|
namespace SharpCifs.Util.Sharpen
|
|
{
|
|
public class SocketEx : Socket
|
|
{
|
|
private int _soTimeOut = -1;
|
|
|
|
public int SoTimeOut
|
|
{
|
|
get
|
|
{
|
|
return _soTimeOut;
|
|
}
|
|
|
|
set
|
|
{
|
|
if (value > 0)
|
|
{
|
|
_soTimeOut = value;
|
|
}
|
|
else
|
|
{
|
|
_soTimeOut = -1;
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
public SocketEx(AddressFamily addressFamily, SocketType socketType, ProtocolType protocolType)
|
|
: base(addressFamily, socketType, protocolType)
|
|
{
|
|
|
|
}
|
|
|
|
public void Connect(IPEndPoint endPoint, int timeOut)
|
|
{
|
|
using (var evt = new ManualResetEventSlim(false))
|
|
{
|
|
using (var args = new SocketAsyncEventArgs
|
|
{
|
|
RemoteEndPoint = endPoint
|
|
})
|
|
{
|
|
args.Completed += delegate
|
|
{
|
|
evt.Set();
|
|
};
|
|
|
|
ConnectAsync(args);
|
|
|
|
if (!evt.Wait(timeOut))
|
|
{
|
|
CancelConnectAsync(args);
|
|
throw new ConnectException("Can't connect to end point.");
|
|
}
|
|
if (args.SocketError != SocketError.Success)
|
|
{
|
|
throw new ConnectException("Can't connect to end point.");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public void Bind2(EndPoint ep)
|
|
{
|
|
if (ep == null)
|
|
Bind(new IPEndPoint(IPAddress.Any, 0));
|
|
else
|
|
Bind(ep);
|
|
}
|
|
|
|
|
|
public int Receive(byte[] buffer, int offset, int count)
|
|
{
|
|
using (var evt = new ManualResetEventSlim(false))
|
|
{
|
|
using (var args = new SocketAsyncEventArgs
|
|
{
|
|
UserToken = this
|
|
})
|
|
{
|
|
args.SetBuffer(buffer, offset, count);
|
|
|
|
args.Completed += delegate
|
|
{
|
|
evt.Set();
|
|
};
|
|
|
|
if (ReceiveAsync(args))
|
|
{
|
|
if (!evt.Wait(_soTimeOut))
|
|
{
|
|
throw new TimeoutException("No data received.");
|
|
}
|
|
}
|
|
|
|
return args.BytesTransferred;
|
|
}
|
|
}
|
|
}
|
|
|
|
public void Send(byte[] buffer, int offset, int length, EndPoint destination = null)
|
|
{
|
|
using (var evt = new ManualResetEventSlim(false))
|
|
{
|
|
using (SocketAsyncEventArgs args = new SocketAsyncEventArgs
|
|
{
|
|
UserToken = this
|
|
})
|
|
{
|
|
args.SetBuffer(buffer, offset, length);
|
|
|
|
args.Completed += delegate
|
|
{
|
|
evt.Set();
|
|
};
|
|
|
|
args.RemoteEndPoint = destination ?? RemoteEndPoint;
|
|
|
|
|
|
SendToAsync(args);
|
|
if (!evt.Wait(_soTimeOut))
|
|
{
|
|
throw new TimeoutException("No data sent.");
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
public InputStream GetInputStream()
|
|
{
|
|
return new NetworkStream(this);
|
|
}
|
|
|
|
public OutputStream GetOutputStream()
|
|
{
|
|
return new NetworkStream(this);
|
|
}
|
|
|
|
}
|
|
}
|