Code:
Imports System.ComponentModel
Imports System.ComponentModel.Design
Imports System.Drawing
Imports System.Net
Imports Service.Winsock_Code
Imports Service.Winsock_Code.WinsockReceiveProgressEventArgs
Imports System.Runtime.Serialization.Formatters.Binary
Imports System.IO
'<ToolboxBitmap(GetType(Winsock), "Winsock.png")> _
'<DefaultEvent("ErrorReceived")> _
'<DesignerAttribute(GetType(WinsockDesigner), GetType(IDesigner))> _
<ToolboxBitmap(GetType(Winsock), "Winsock.png"), _
DefaultEvent("ErrorReceived")> _
Public Class Winsock
Inherits Component
Implements IWinsock
Public Sub New()
_localPort = 8080
_remotePort = 8080
_remoteHost = "localhost"
_maxPendingConnections = 1
_legacySupport = False
_protocol = WinsockProtocol.Tcp
_wsState = WinsockStates.Closed
_asSock = New AsyncSocket(Me)
End Sub
#Region " Events "
''' <summary>
''' Triggers an event declared at module level within a class, form, or document in a thread-safe manner.
''' </summary>
''' <param name="ev">The event to be raised.</param>
''' <param name="args">The arguements for the event.</param>
Private Sub RaiseEventSafe(ByVal ev As System.Delegate, ByRef args() As Object)
Dim bFired As Boolean
If Not (ev Is Nothing) Then
For Each singleCast As System.Delegate In ev.GetInvocationList()
bFired = False
Try
Dim syncInvoke As ISynchronizeInvoke = CType(singleCast.Target, ISynchronizeInvoke)
If Not (syncInvoke Is Nothing) AndAlso syncInvoke.InvokeRequired Then
bFired = True
syncInvoke.BeginInvoke(singleCast, args)
Else
bFired = True
singleCast.DynamicInvoke(args)
End If
Catch ex As Exception
If Not bFired Then singleCast.DynamicInvoke(args)
End Try
Next
End If
End Sub
''' <summary>
''' Occurs when connection is achieved (client and server).
''' </summary>
Public Event Connected(ByVal sender As Object, ByVal e As WinsockConnectedEventArgs) Implements IWinsock.Connected
''' <summary>
''' Occurs on the server when a client is attempting to connect.
''' </summary>
''' <remarks>Client registers connected at this point. Server must Accept in order for it to be connected.</remarks>
Public Event ConnectionRequest(ByVal sender As Object, ByVal e As WinsockConnectionRequestEventArgs) Implements IWinsock.ConnectionRequest
''' <summary>
''' Occurs when data arrives on the socket.
''' </summary>
''' <remarks>Raised only after all parts of the data have been collected.</remarks>
Public Event DataArrival(ByVal sender As Object, ByVal e As WinsockDataArrivalEventArgs) Implements IWinsock.DataArrival
''' <summary>
''' Occurs when disconnected from the remote computer (client and server).
''' </summary>
Public Event Disconnected(ByVal sender As Object, ByVal e As System.EventArgs) Implements IWinsock.Disconnected
''' <summary>
''' Occurs when an error is detected in the socket.
''' </summary>
''' <remarks>May also be raised on disconnected (depending on disconnect circumstance).</remarks>
Public Event ErrorReceived(ByVal sender As Object, ByVal e As WinsockErrorReceivedEventArgs) Implements IWinsock.ErrorReceived
''' <summary>
''' Occurs while the receive buffer is being filled with data.
''' </summary>
Public Event ReceiveProgress(ByVal sender As Object, ByVal e As WinsockReceiveProgressEventArgs) Implements IWinsock.ReceiveProgress
''' <summary>
''' Occurs when sending of data is completed.
''' </summary>
Public Event SendComplete(ByVal sender As Object, ByVal e As WinsockSendEventArgs) Implements IWinsock.SendComplete
''' <summary>
''' Occurs when the send buffer has been sent but not all the data has been sent yet.
''' </summary>
Public Event SendProgress(ByVal sender As Object, ByVal e As WinsockSendEventArgs) Implements IWinsock.SendProgress
''' <summary>
''' Occurs when the state of the socket changes.
''' </summary>
Public Event StateChanged(ByVal sender As Object, ByVal e As WinsockStateChangedEventArgs) Implements IWinsock.StateChanged
''' <summary>
''' Raises the Connected event.
''' </summary>
Public Sub OnConnected(ByVal e As WinsockConnectedEventArgs) Implements IWinsock.OnConnected
RaiseEventSafe(ConnectedEvent, New Object() {Me, e})
End Sub
''' <summary>
''' Raises the ConnectionRequest event.
''' </summary>
Public Sub OnConnectionRequest(ByVal e As WinsockConnectionRequestEventArgs) Implements IWinsock.OnConnectionRequest
RaiseEventSafe(ConnectionRequestEvent, New Object() {Me, e})
If e.Cancel Then
e.Client.Close()
End If
End Sub
''' <summary>
''' Raises the DataArrival event.
''' </summary>
Public Sub OnDataArrival(ByVal e As WinsockDataArrivalEventArgs) Implements IWinsock.OnDataArrival
RaiseEventSafe(DataArrivalEvent, New Object() {Me, e})
End Sub
''' <summary>
''' Raises the Disconnected event.
''' </summary>
Public Sub OnDisconnected() Implements IWinsock.OnDisconnected
RaiseEventSafe(DisconnectedEvent, New Object() {Me, New System.EventArgs})
End Sub
''' <summary>
''' Raises the ErrorReceived event.
''' </summary>
Public Sub OnErrorReceived(ByVal e As WinsockErrorReceivedEventArgs) Implements IWinsock.OnErrorReceived
RaiseEventSafe(ErrorReceivedEvent, New Object() {Me, e})
End Sub
''' <summary>
''' Raises the ReceiveProgress event.
''' </summary>
Public Sub OnReceiveProgress(ByVal e As WinsockReceiveProgressEventArgs) Implements IWinsock.OnReceiveProgress
RaiseEventSafe(ReceiveProgressEvent, New Object() {Me, e})
End Sub
''' <summary>
''' Raises the SendComplete event.
''' </summary>
Public Sub OnSendComplete(ByVal e As WinsockSendEventArgs) Implements IWinsock.OnSendComplete
RaiseEventSafe(SendCompleteEvent, New Object() {Me, e})
End Sub
''' <summary>
''' Raises the SendProgress event.
''' </summary>
Public Sub OnSendProgress(ByVal e As WinsockSendEventArgs) Implements IWinsock.OnSendProgress
RaiseEventSafe(SendProgressEvent, New Object() {Me, e})
End Sub
''' <summary>
''' Raises the StateChanged event.
''' </summary>
Private Sub OnStateChanged(ByVal e As WinsockStateChangedEventArgs) 'Implements IWinsock.OnStateChanged
If _wsState <> e.New_State Then
_wsState = e.New_State
RaiseEventSafe(StateChangedEvent, New Object() {Me, e})
End If
End Sub
#End Region
#Region " Private Members "
Private _asSock As AsyncSocket
Private _wsState As WinsockStates
Private _localPort As Integer
Private _maxPendingConnections As Integer
Private _remoteHost As String
Private _remotePort As Integer
Private _legacySupport As Boolean
Private _protocol As WinsockProtocol
#End Region
#Region " Helper Methods "
''' <summary>
''' Encapsulates the OnStateChanged methods so the AsyncSocket
''' doesn't have to build the EventArgs parameter all the time.
''' </summary>
''' <param name="new_state">The new state of the Winsock.</param>
Protected Friend Sub ChangeState(ByVal new_state As WinsockStates) Implements IWinsock.ChangeState
OnStateChanged(New WinsockStateChangedEventArgs(_wsState, new_state))
End Sub
''' <summary>
''' When the port is set dynamically by using port 0, the socket can now update the property of the component.
''' </summary>
''' <param name="new_port">The port we are now listening on.</param>
Protected Friend Sub ChangeLocalPort(ByVal new_port As Integer) Implements IWinsock.ChangeLocalPort
_localPort = new_port
End Sub
#End Region
#Region " Public Properties "
''' <summary>
''' Gets or sets a value indicating the interal size of the byte buffers.
''' </summary>
Public Property BufferSize() As Integer
Get
Return _asSock.BufferSize
End Get
Set(ByVal value As Integer)
_asSock.BufferSize = value
End Set
End Property
''' <summary>
''' Gets a value indicating whether the buffer has data for retrieval.
''' </summary>
<Browsable(False)> _
Public ReadOnly Property HasData() As Boolean
Get
Return _asSock.BufferCount > 0
End Get
End Property
''' <summary>
''' Gets or sets a value indicating if Legacy support should be used or not.
''' </summary>
''' <remarks>Legacy support is to support older winsock style connections.</remarks>
Public Property LegacySupport() As Boolean Implements IWinsock.LegacySupport
Get
Return _legacySupport
End Get
Set(ByVal value As Boolean)
If Not value AndAlso Protocol = WinsockProtocol.Udp Then
Throw New Exception("LegacySupport is required for UDP connections.")
End If
_legacySupport = value
End Set
End Property
''' <summary>
''' Gets the local machine's IP address(es).
''' </summary>
<Browsable(False)> _
Public ReadOnly Property LocalIP() As String()
Get
Dim h As System.Net.IPHostEntry = System.Net.Dns.Resolve(System.Net.Dns.GetHostName)
Dim s(h.AddressList.Length - 1) As String
For i As Integer = 0 To h.AddressList.Length - 1
s(i) = CType(h.AddressList.GetValue(i), Net.IPAddress).ToString
Next
Return s
End Get
End Property
''' <summary>
''' Gets or sets a value indicating the port the <see cref="Winsock" /> control should listen on.
''' </summary>
''' <remarks>Cannot change while listening, connected, or connecting - but can change while closing.</remarks>
Public Property LocalPort() As Integer
Get
Return _localPort
End Get
Set(ByVal value As Integer)
Select Case State
Case WinsockStates.Listening
Throw New Exception("Cannot change the local port while already listening on a port.")
Case WinsockStates.Connected
Throw New Exception("Cannot change the local port of a connection that is already active.")
Case Else
If State <> WinsockStates.Closed AndAlso State <> WinsockStates.Closing Then
Throw New Exception("Cannot change the local port while the component is processing, it may have adverse effects on the connection process.")
End If
End Select
_localPort = value
End Set
End Property
''' <summary>
''' Gets or sets a value that control the length of the maximum length of the pending connections queue.
''' </summary>
''' <remarks>Cannot change while listening.</remarks>
Public Property MaxPendingConnections() As Integer
Get
Return _maxPendingConnections
End Get
Set(ByVal value As Integer)
Select Case State
Case WinsockStates.Listening
Throw New Exception("Cannot change the pending connections value while already listening.")
End Select
_maxPendingConnections = value
End Set
End Property
''' <summary>
''' Gets a NetworkStream that this Winsock object is based on.
''' </summary>
<Browsable(False)> _
Public ReadOnly Property NetworkStream() As System.Net.Sockets.NetworkStream
Get
Return _asSock.UnderlyingStream
End Get
End Property
''' <summary>
''' Gets or sets the winsock protocol to use when communicating with the remote computer.
''' </summary>
<RefreshProperties(RefreshProperties.All)> _
Public Property Protocol() As WinsockProtocol Implements IWinsock.Protocol
Get
Return _protocol
End Get
Set(ByVal value As WinsockProtocol)
If State <> WinsockStates.Closed Then
Throw New Exception("Cannot change the protocol while listening or connected to a remote computer.")
End If
_protocol = value
End Set
End Property
''' <summary>
''' Gets or sets a value that determines what remote computer to connect to, or is currently connected to.
''' </summary>
''' <remarks>Can only change if closed or listening.</remarks>
Public Property RemoteHost() As String Implements IWinsock.RemoteHost
Get
Return _remoteHost
End Get
Set(ByVal value As String)
If State <> WinsockStates.Closed AndAlso State <> WinsockStates.Listening Then
Throw New Exception("Cannot change the remote host while already connected to a remote computer.")
End If
_remoteHost = value
End Set
End Property
''' <summary>
''' Gets or sets a value that determines which port on the remote computer to connect on, or is currently connected on.
''' </summary>
''' <remarks>Can only change if closed or listening.</remarks>
Public Property RemotePort() As Integer Implements IWinsock.RemotePort
Get
Return _remotePort
End Get
Set(ByVal value As Integer)
If State <> WinsockStates.Closed AndAlso State <> WinsockStates.Listening Then
Throw New Exception("Cannot change the remote port while already connected to a remote computer.")
End If
_remotePort = value
End Set
End Property
''' <summary>
''' Gets the state of the <see cref="Winsock">Winsock</see> control.
''' </summary>
<Browsable(False)> _
Public ReadOnly Property State() As WinsockStates Implements IWinsock.State
Get
Return _wsState
End Get
End Property
#End Region
#Region " Public Methods "
''' <summary>
''' Places a <see cref="Winsock">Winsock</see> in a listening state.
''' </summary>
Public Sub Listen()
_asSock.Listen(LocalPort, MaxPendingConnections)
End Sub
''' <summary>
''' Places a <see cref="Winsock">Winsock</see> in a listening state.
''' </summary>
''' <param name="port">The port <see cref="Winsock">Winsock</see> should listen on.</param>
Public Sub Listen(ByVal port As Integer)
If port < 0 Then
Throw New ArgumentException("Port cannot be less than zero.", "port")
End If
LocalPort = port
Listen()
End Sub
''' <summary>
''' Places a <see cref="Winsock">Winsock</see> in a listening state.
''' </summary>
''' <param name="ip">The IP address the <see cref="Winsock">Winsock</see> should listen on. This must be an ip address.</param>
Public Sub Listen(ByVal ip As String)
If ip Is Nothing Then
Listen()
Else
Dim ipAddr As IPAddress = Nothing
Try
ipAddr = IPAddress.Parse(ip)
Catch ex As Exception
Throw New ArgumentException("IP address specified is not a valid IP address.", "ip")
End Try
_asSock.Listen(LocalPort, MaxPendingConnections, ipAddr)
End If
End Sub
''' <summary>
''' Places a <see cref="Winsock">Winsock</see> in a listening state.
''' </summary>
''' <param name="ip">The IP address the <see cref="Winsock">Winsock</see> should listen on.</param>
''' <param name="port">The port <see cref="Winsock">Winsock</see> should listen on.</param>
Public Sub Listen(ByVal ip As String, ByVal port As Integer)
If port < 0 Then
Throw New ArgumentException("Port cannot be less than zero.", "port")
End If
LocalPort = port
If ip Is Nothing Then
Listen()
Else
Dim ipAddr As IPAddress = Nothing
Try
ipAddr = IPAddress.Parse(ip)
Catch ex As Exception
Throw New ArgumentException("IP address specified is not a valid IP address.", "ip")
End Try
_asSock.Listen(LocalPort, MaxPendingConnections, ipAddr)
End If
End Sub
''' <summary>
''' Accepts a client connect as valid and begins to monitor it for incoming data.
''' </summary>
''' <param name="client">A <see cref="System.Net.Sockets.Socket">System.Net.Sockets.Socket</see> that represents the client being accepted.</param>
Public Sub Accept(ByVal client As Sockets.Socket)
If _asSock.Accept(client) Then
_localPort = _asSock.LocalPort
' also set remote host and port
End If
End Sub
''' <summary>
''' Creates an new <see cref="Winsock">Winsock</see> and accepts the client connection on it.
''' </summary>
''' <param name="client">A <see cref="System.Net.Sockets.Socket">System.Net.Sockets.Socket</see> that represents the client being accepted.</param>
''' <remarks>
''' This was created to be used by the listener, to keep the listener listening while
''' also accepting a connection.
''' </remarks>
Public Function AcceptNew(ByVal client As Sockets.Socket) As Winsock
Dim wskReturn As New Winsock
wskReturn.Protocol = Me.Protocol
wskReturn.LegacySupport = Me.LegacySupport
wskReturn.Accept(client)
Return wskReturn
End Function
''' <summary>
''' Closes an open <see cref="Winsock">Winsock</see> connection.
''' </summary>
Public Sub Close()
_asSock.Close()
End Sub
''' <summary>
''' Establishes a connection to a remote host.
''' </summary>
Public Sub Connect()
_asSock.Connect(RemoteHost, RemotePort)
End Sub
''' <summary>
''' Establishes a connection to a remote host.
''' </summary>
''' <param name="remoteHostOrIP">A <see cref="System.String">System.String</see> containing the Hostname or IP address of the remote host.</param>
''' <param name="remote_port">A value indicating the port on the remote host to connect to.</param>
Public Sub Connect(ByVal remoteHostOrIP As String, ByVal remote_port As Integer)
RemoteHost = remoteHostOrIP
RemotePort = remote_port
Connect()
End Sub
''' <summary>
''' Sends an object to a connected socket on a remote computer.
''' </summary>
''' <param name="data">The object to send.</param>
''' <remarks>
''' The object is first serialized using a BinaryFormatter - unless
''' it is already a byte array, in which case it just sends the byte array.
''' </remarks>
Public Sub Send(ByVal data As Object)
Dim byt() As Byte
If LegacySupport AndAlso data.GetType Is GetType(String) Then
byt = System.Text.Encoding.Default.GetBytes(CStr(data))
Else
byt = ObjectPacker.GetBytes(data)
End If
_asSock.Send(byt)
End Sub
''' <summary>
''' Sends a file to a connected socket on a remote computer.
''' </summary>
''' <param name="filename">The full path to the file you want to send.</param>
''' <remarks>
''' Creates a special file object to send, so the receiving end knows what to do with it.
''' </remarks>
Public Sub SendFile(ByVal filename As String)
Dim wsf As New WinsockFileData
Try
If Not wsf.ReadFile(filename) Then
Throw New Exception("File does not exist, or there was a problem reading the file.")
End If
If LegacySupport Then
Send(wsf.FileData)
Else
Send(wsf)
End If
Catch ex As Exception
SharedMethods.RaiseError(Me, ex.Message)
End Try
End Sub
''' <summary>
''' Gets the next object from the buffer, removing it from the buffer.
''' </summary>
''' <returns>
''' A Deserialized object or if it can't be deserialized the byte array.
''' </returns>
Public Function [Get]() As Object
Dim byt() As Byte = _asSock.GetData()
Return ObjectPacker.GetObject(byt)
End Function
''' <summary>
''' Gets the next object from the buffer, leaving it ing the buffer.
''' </summary>
''' <returns>
''' A Deserialized object or if it can't be deserialized the byte array.
''' </returns>
Public Function Peek() As Object
Dim byt() As Byte = _asSock.PeekData()
Return ObjectPacker.GetObject(byt)
End Function
#End Region
Public Class ObjectPacker
''' <summary>
''' Serializes an object to a byte array.
''' </summary>
''' <param name="obj">The object to be serialized.</param>
Public Shared Function GetBytes(ByVal obj As Object) As Byte()
Dim bytArrayTester(0) As Byte
If obj.GetType Is bytArrayTester.GetType Then
Return CType(obj, Byte())
End If
Dim ret() As Byte = Nothing
Dim blnRetAfterClose As Boolean = False
Dim ms As New MemoryStream
Dim bf As New BinaryFormatter
Try
bf.Serialize(ms, obj)
Catch ex As Exception
blnRetAfterClose = True
Finally
ms.Close()
End Try
If blnRetAfterClose Then Return ret
ret = ms.ToArray()
Return ret
End Function
''' <summary>
''' Deserializes an object from a byte array.
''' </summary>
''' <param name="byt">The byte array from which to obtain the object.</param>
Public Shared Function GetObject(ByVal byt() As Byte) As Object
Dim ms As New MemoryStream(byt, False)
Dim bf As New BinaryFormatter
Dim x As Object
Try
x = bf.Deserialize(ms)
Catch ex As Exception
x = byt
Finally
ms.Close()
End Try
Return x
End Function
End Class
End Class
[code]Imports System.Net
Imports System.Net.Sockets
Imports System.Threading
Imports System.IO
Imports System.Xml
Imports System.Xml.Serialization
Imports System.Runtime.Serialization.Formatters.Binary
Imports Service.Winsock_Code.WinsockReceiveProgressEventArgs
Imports System.ComponentModel
Public Class Winsock_Code
Public Class AsyncSocket
Private myParent As IWinsock ' The parent Winsock - allows access to properties and event raising
Private mSock1 As Socket ' for IPv4 (listening) or both IPv4 and IPv6 (connections)
Private mSock2 As Socket ' for IPv6 (listening) only
Private byteBuffer() As Byte ' Stores the incoming bytes waiting to be processed
Private incBufferSize As Integer = 8192 ' The buffer size of the socket
Private _buff As ByteBufferCol ' Temporary byte buffer - used while an object is being assembled
Private _closing As Boolean = False ' Prevents the Close() method from being run while it already running
Private qBuffer As Queue ' The Buffer Queue - where objects wait to be picked up by the Get() method
Private phProcessor As PacketHeader ' The PacketHeader processor - looks for and reads the packet header added to the byte array
Private sBufferMutex As New Mutex
Private sendBuffer As Deque ' The Sending Buffer Queue - where objects wait to be sent.
Private thdSendLoop As Thread ' Used to send everything in the sendBuffer
Private lckSendLoop As Object ' Used for a syncronized lock on the SendLoop thread
Private bIsSending As Boolean ' Used internally to tell if a sending loop is in progress.
Public Sub New(ByRef parent As IWinsock)
Try
myParent = parent
phProcessor = New PacketHeader
qBuffer = New Queue
_buff = New ByteBufferCol
sendBuffer = New Deque
lckSendLoop = New Object
ReDim byteBuffer(incBufferSize)
Catch ex As Exception
SharedMethods.RaiseError(myParent, "Unable to initialize the AsyncSocket.")
End Try
End Sub
''' <summary>
''' Gets a value containing the remote IP address.
''' </summary>
Protected Friend ReadOnly Property RemoteIP() As String
Get
Dim rEP As System.Net.IPEndPoint = CType(mSock1.RemoteEndPoint, System.Net.IPEndPoint)
Return rEP.Address.ToString()
End Get
End Property
''' <summary>
''' Gets a value containing the remote port number.
''' </summary>
Protected Friend ReadOnly Property RemotePort() As Integer
Get
Dim rEP As System.Net.IPEndPoint = CType(mSock1.RemoteEndPoint, System.Net.IPEndPoint)
Return rEP.Port
End Get
End Property
''' <summary>
''' Gets a value containing the local port number.
''' </summary>
Protected Friend ReadOnly Property LocalPort() As Integer
Get
Dim lEP As System.Net.IPEndPoint = CType(mSock1.LocalEndPoint, IPEndPoint)
Return lEP.Port
End Get
End Property
Protected Friend ReadOnly Property BufferCount() As Integer
Get
Return qBuffer.Count
End Get
End Property
Protected Friend Property BufferSize() As Integer
Get
Return incBufferSize
End Get
Set(ByVal value As Integer)
incBufferSize = value
End Set
End Property
Protected Friend ReadOnly Property UnderlyingStream() As Net.Sockets.NetworkStream
Get
If Not mSock1 Is Nothing Then Return New Net.Sockets.NetworkStream(mSock1, IO.FileAccess.ReadWrite, False)
Return Nothing
End Get
End Property
#Region " Public Methods "
''' <summary>
''' Accepts an incoming connection and starts the data listener.
''' </summary>
''' <param name="client">The client to accept.</param>
Public Function Accept(ByVal client As Socket) As Boolean
Try
If myParent.State <> WinsockStates.Closed Then
Throw New Exception("Cannot accept a connection while the State is not closed.")
End If
mSock1 = client
Receive()
myParent.ChangeState(WinsockStates.Connected)
Dim e As New WinsockConnectedEventArgs(CType(mSock1.RemoteEndPoint, System.Net.IPEndPoint).Address.ToString, CType(mSock1.RemoteEndPoint, System.Net.IPEndPoint).Port)
myParent.OnConnected(e)
Return True
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message)
Return False
End Try
End Function
''' <summary>
''' Closes the socket if its already open or listening.
''' </summary>
Public Sub Close()
Try
' If we are already closing then exit the subroutine
If _closing Then Exit Sub
' Set the closing flag so that this doesn't get run more than once
' at a time.
_closing = True
' If we are already closed - exit the subroutine
If myParent.State = WinsockStates.Closed Then _closing = False : Exit Sub
Dim bAllowDisconnect As Boolean = False
' Close the Socket(s) as necessary
Select Case myParent.State
Case WinsockStates.Connected
' Change the state to Closing
myParent.ChangeState(WinsockStates.Closing)
If Not (mSock1 Is Nothing) Then mSock1.Close()
' Allow disconnect event to raise
bAllowDisconnect = True
Case WinsockStates.Listening
' Change the state to Closing
myParent.ChangeState(WinsockStates.Closing)
If Not (mSock1 Is Nothing) Then mSock1.Close()
If Not (mSock2 Is Nothing) Then mSock2.Close()
' Do not allow Disconnect event - we weren't connected to anything
' only listening.
bAllowDisconnect = False
End Select
' Change state to Closed
myParent.ChangeState(WinsockStates.Closed)
' Raise the Disconnected event - if allowed to
If bAllowDisconnect Then myParent.OnDisconnected()
_closing = False
Catch ex As Exception
_closing = False
If Not (ex.InnerException Is Nothing) Then
SharedMethods.RaiseError(myParent, ex.Message, ex.InnerException.Message)
Else
SharedMethods.RaiseError(myParent, ex.Message)
End If
End Try
End Sub
''' <summary>
''' Starts Listening for incoming connections. For UDP sockets it starts listening for incoming data.
''' </summary>
''' <param name="port">The port to start listening on.</param>
''' <param name="max_pending">The maximum length of the pending connections queue.</param>
Public Sub Listen(ByVal port As Integer, ByVal max_pending As Integer)
Try
If myParent.Protocol = WinsockProtocol.Tcp Then
Dim blnChangePort As Boolean = False
' Start listening on IPv4 - if available
If Socket.SupportsIPv4 Then
mSock1 = New Socket(AddressFamily.InterNetwork, SocketType.Stream, ProtocolType.Tcp)
Dim ipLocal As New IPEndPoint(IPAddress.Any, port)
mSock1.Bind(ipLocal)
mSock1.Listen(max_pending)
mSock1.BeginAccept(New AsyncCallback(AddressOf ListenCallback), mSock1)
End If
' if port was 0 find port used for IPv4 and use it for IPv6
If port = 0 Then
Dim lEP As System.Net.IPEndPoint = CType(mSock1.LocalEndPoint, IPEndPoint)
port = lEP.Port
blnChangePort = True
End If
' Start listening on IPv6 - if available
If Socket.SupportsIPv6 Then
mSock2 = New Socket(AddressFamily.InterNetworkV6, SocketType.Stream, ProtocolType.Tcp)
Dim ipLocal As New IPEndPoint(IPAddress.IPv6Any, port)
mSock2.Bind(ipLocal)
mSock2.Listen(max_pending)
mSock2.BeginAccept(New AsyncCallback(AddressOf ListenCallback), mSock2)
End If
If blnChangePort Then
myParent.ChangeLocalPort(port)
End If
' Change state to Listening
myParent.ChangeState(WinsockStates.Listening)
ElseIf myParent.Protocol = WinsockProtocol.Udp Then
If port <= 0 Then
Throw New ArgumentException("While port 0 works for getting random port for UPD, there is no way for the server operator to know the port used until a completed send/receive method call is used which means the port is known already.", "port")
End If
' Start data listening on IPv4 - if available
If Socket.SupportsIPv4 Then
mSock1 = New Socket(AddressFamily.InterNetwork, SocketType.Dgram, ProtocolType.Udp)
Dim ipLocal As New IPEndPoint(IPAddress.Any, port)
Dim ipeSender As New IPEndPoint(IPAddress.Any, 0)
Dim xe As New UdpReceiveState
xe.SendingSocket = mSock1
xe.ReceivingEndpoint = ipeSender
mSock1.Bind(ipLocal)
mSock1.BeginReceiveFrom(byteBuffer, 0, incBufferSize, SocketFlags.None, xe.ReceivingEndpoint, New AsyncCallback(AddressOf ReceiveCallbackUDP), xe)
End If
' Start data listening on IPv6 - if available
If Socket.SupportsIPv6 Then
mSock2 = New Socket(AddressFamily.InterNetworkV6, SocketType.Dgram, ProtocolType.Udp)
Dim ipLocal As New IPEndPoint(IPAddress.IPv6Any, port)
Dim ipeSender As New IPEndPoint(IPAddress.IPv6Any, 0)
Dim xe As New UdpReceiveState
xe.SendingSocket = mSock2
xe.ReceivingEndpoint = ipeSender
mSock2.Bind(ipLocal)
mSock2.BeginReceiveFrom(byteBuffer, 0, incBufferSize, SocketFlags.None, xe.ReceivingEndpoint, New AsyncCallback(AddressOf ReceiveCallbackUDP), xe)
End If
myParent.ChangeState(WinsockStates.Listening)
End If
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
''' <summary>
''' Starts Listening for incoming connections on the specified IP address. For UDP sockets it starts listening for incoming data.
''' </summary>
''' <param name="port">The port to start listening on.</param>
''' <param name="max_pending">The maximum length of the pending connections queue.</param>
''' <param name="ip">The IP address on which to listen.</param>
Public Sub Listen(ByVal port As Integer, ByVal max_pending As Integer, ByVal ip As IPAddress)
Try
' IP contains information on type (IPv4 vs. IPv6) so we can just work with that
If myParent.Protocol = WinsockProtocol.Tcp Then
mSock1 = New Socket(ip.AddressFamily, SocketType.Stream, ProtocolType.Tcp)
Dim ipLocal As New IPEndPoint(ip, port)
mSock1.Bind(ipLocal)
mSock1.Listen(max_pending)
mSock1.BeginAccept(New AsyncCallback(AddressOf ListenCallback), mSock1)
If port = 0 Then
Dim lEP As System.Net.IPEndPoint = CType(mSock1.LocalEndPoint, IPEndPoint)
myParent.ChangeLocalPort(lEP.Port)
End If
ElseIf myParent.Protocol = WinsockProtocol.Udp Then
If port <= 0 Then
Throw New ArgumentException("While port 0 works for getting random port for UPD, there is no way for the server operator to know the port used until a completed send/receive method call is used which means the port is known already.", "port")
End If
mSock1 = New Socket(ip.AddressFamily, SocketType.Dgram, ProtocolType.Udp)
Dim ipLocal As New IPEndPoint(ip, port)
Dim ipeSender As New IPEndPoint(ip, 0)
Dim xe As New UdpReceiveState
xe.SendingSocket = mSock1
xe.ReceivingEndpoint = ipeSender
mSock1.Bind(ipLocal)
mSock1.BeginReceiveFrom(byteBuffer, 0, incBufferSize, SocketFlags.None, xe.ReceivingEndpoint, New AsyncCallback(AddressOf ReceiveCallbackUDP), xe)
End If
myParent.ChangeState(WinsockStates.Listening)
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
''' <summary>
''' Gets the first object in the buffer without removing it.
''' </summary>
Public Function PeekData() As Byte()
If qBuffer.Count = 0 Then Return Nothing
Return DirectCast(qBuffer.Peek(), Byte())
End Function
''' <summary>
''' Gets and removes the first object in the buffer.
''' </summary>
Public Function GetData() As Byte()
If qBuffer.Count = 0 Then Return Nothing
Return DirectCast(qBuffer.Dequeue, Byte())
End Function
''' <summary>
''' Attemps to connect to a remote computer.
''' </summary>
''' <param name="remoteHostOrIp">The remote host or IP address of the remote computer.</param>
''' <param name="remote_port">The port number on which to connect to the remote computer.</param>
Public Sub Connect(ByVal remoteHostOrIp As String, ByVal remote_port As Integer)
Try
If myParent.State <> WinsockStates.Closed Then
Throw New Exception("Cannot connect to a remote host when the Winsock State is not closed.")
End If
myParent.ChangeState(WinsockStates.ResolvingHost)
Dim resolvedIP As IPAddress = Nothing
Try
resolvedIP = IPAddress.Parse(remoteHostOrIp) ' no TryParse - Bah!
If resolvedIP Is Nothing Then Throw New Exception("Failed to get.")
myParent.ChangeState(WinsockStates.HostResolved)
Connect(resolvedIP, remote_port)
Catch ex As Exception
'Dns.BeginGetHostEntry(remoteHostOrIp, New AsyncCallback(AddressOf DoConnectCallback), remote_port)
Dns.BeginResolve(remoteHostOrIp, New AsyncCallback(AddressOf DoConnectCallback), remote_port)
End Try
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
''' <summary>
''' Attempts to connect to a remote computer.
''' </summary>
''' <param name="remIP">The IP address of the remote computer.</param>
''' <param name="port">The port number on which to connect to the remote computer.</param>
Public Sub Connect(ByVal remIP As IPAddress, ByVal port As Integer)
Try
Dim remEP As New IPEndPoint(remIP, port)
If myParent.State <> WinsockStates.HostResolved Then Exit Sub
mSock1 = New Socket(remIP.AddressFamily, SocketType.Stream, ProtocolType.Tcp)
myParent.ChangeState(WinsockStates.Connecting)
mSock1.BeginConnect(remEP, New AsyncCallback(AddressOf ConnectCallback), mSock1)
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message, ex.ToString)
End Try
End Sub
''' <summary>
''' Sends data to the remote computer.
''' </summary>
''' <param name="byt">The byte array of data to send.</param>
Public Sub Send(ByVal byt() As Byte)
Try
' If it's going out UDP, get the location it's going to
Dim remEP As IPEndPoint = Nothing
If myParent.Protocol = WinsockProtocol.Udp Then
Dim ihe As IPHostEntry = Dns.Resolve(myParent.RemoteHost)
remEP = New IPEndPoint(ihe.AddressList(0), myParent.RemotePort)
End If
' LegacySupport doesn't need a header, so if it's NOT active we can add one
If Not myParent.LegacySupport Then
' LegacySupport INACTIVE - add a packet header, the other end knows how to decode it
phProcessor.AddHeader(byt)
End If
' Create the data object and add it to the queue
Dim sqData As New SendQueueData(remEP, byt)
' We must lock access to the send buffer to prevent simultaneous access
' from multiple threads
SyncLock sendBuffer.SyncRoot
sendBuffer.Enqueue(sqData)
End SyncLock
' Start the sending process - if the process isn't already started.
SyncLock lckSendLoop
If Not bIsSending Then
thdSendLoop = New Thread(AddressOf DoSend)
thdSendLoop.Start()
End If
End SyncLock
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
#End Region
#Region " Callback Methods "
''' <summary>
''' The callback for the listener - only used for a TCP listener.
''' </summary>
''' <remarks>This routine starts again when finished making it loop to continuously receive connections.</remarks>
Private Sub ListenCallback(ByVal ar As IAsyncResult)
Try
' Get the socket doing the listening, if it's not there
' we can't continue.
Dim listener As Socket = Nothing
Try
listener = DirectCast(ar.AsyncState, Socket)
Catch ex As Exception
listener = Nothing
End Try
If listener Is Nothing Then
Throw New Exception("Listener object no longer exists.")
End If
' End the Accept that was started
Dim client As Socket = listener.EndAccept(ar)
' Raise ConnectionRequest event
Dim e As New WinsockConnectionRequestEventArgs(client)
myParent.OnConnectionRequest(e)
' If the Winsock is no longer in the listening state
' close and exit gracefully.
If myParent.State <> WinsockStates.Listening Then
listener.Close()
Exit Sub
End If
' start listening again
listener.BeginAccept(New AsyncCallback(AddressOf ListenCallback), listener)
Catch exO As ObjectDisposedException
' Close was called, destroying the object - exit without
' displaying an error.
Exit Try
Catch exS As SocketException
' There was a problem with the connection
' If the SocketError is not Success, then close and
' show an error message
SharedMethods.RaiseError(myParent, exS.Message, "", exS.ErrorCode)
Catch ex As Exception
' General execption - show an error message.
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
''' <summary>
''' The callback method for the Receive method (UDP only) - used when there is incoming data.
''' </summary>
Private Sub ReceiveCallbackUDP(ByVal ar As IAsyncResult)
Try
' Get actively receiving socket and IPEndPoint
Dim xe As UdpReceiveState = CType(ar.AsyncState, UdpReceiveState)
Dim cb_UDP As Socket = CType(xe.SendingSocket, Socket)
' Get the size of the received data
Dim iSize As Integer = cb_UDP.EndReceiveFrom(ar, xe.ReceivingEndpoint)
Dim remEP As IPEndPoint = DirectCast(xe.ReceivingEndpoint, IPEndPoint)
If iSize < 1 Then
If _buff.Count > 0 Then _buff.Clear()
Close()
Exit Sub
End If
' Process the receieved data
ProcessIncoming(byteBuffer, iSize, remEP.Address.ToString, remEP.Port)
' Clear and resize the buffer
ReDim byteBuffer(incBufferSize)
' Restart data listener
Dim ipeSender As IPEndPoint
If remEP.AddressFamily = AddressFamily.InterNetwork Then
ipeSender = New IPEndPoint(IPAddress.Any, 0)
Else
ipeSender = New IPEndPoint(IPAddress.IPv6Any, 0)
End If
xe.ReceivingEndpoint = ipeSender
cb_UDP.BeginReceiveFrom(byteBuffer, 0, incBufferSize, SocketFlags.None, xe.ReceivingEndpoint, New AsyncCallback(AddressOf ReceiveCallbackUDP), xe)
Catch exO As ObjectDisposedException
' Close was called, destroying the object - exit without
' displaying an error.
Exit Try
Catch exS As SocketException
' There was a problem with the connection
' If the SocketError is not Success, then close and
' show an error message
SharedMethods.RaiseError(myParent, exS.Message, "", exS.ErrorCode)
Catch ex As Exception
' General execption - show an error message.
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
''' <summary>
''' The callback for the Connect method - used on the client to start looking for data.
''' </summary>
Private Sub ConnectCallback(ByVal ar As IAsyncResult)
Try
If myParent.State <> WinsockStates.Connecting Then Exit Sub
Dim sock As Socket = CType(ar.AsyncState, Socket)
sock.EndConnect(ar)
If myParent.State <> WinsockStates.Connecting Then sock.Close() : Exit Sub
' start listening for data
Receive()
' Finished - raise events...
myParent.ChangeState(WinsockStates.Connected)
' Raise the Connected event
Dim ipE As IPEndPoint = DirectCast(sock.RemoteEndPoint, IPEndPoint)
Dim e As New WinsockConnectedEventArgs(ipE.Address.ToString, ipE.Port)
myParent.OnConnected(e)
Catch exS As SocketException
Close()
SharedMethods.RaiseError(myParent, exS.Message, "", exS.ErrorCode)
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
''' <summary>
''' The callback method for the Receive method (TCP only) - used when there is incoming data.
''' </summary>
Private Sub ReceiveCallback(ByVal ar As IAsyncResult)
Try
' Get the size of the incoming data while ending the receive
Dim iSize As Integer = mSock1.EndReceive(ar)
If iSize < 1 Then
' no size identified - connection closed
If _buff.Count > 0 Then _buff.Clear()
Close()
Exit Sub
End If
' Get the remote IP address
Dim ipE As IPEndPoint = CType(mSock1.RemoteEndPoint, IPEndPoint)
' Process the incoming data (also raises DataArrival)
ProcessIncoming(byteBuffer, iSize, ipE.Address.ToString, ipE.Port)
ReDim byteBuffer(incBufferSize)
' Start listening for data again
mSock1.BeginReceive(byteBuffer, 0, incBufferSize, SocketFlags.None, New AsyncCallback(AddressOf ReceiveCallback), Nothing)
Catch exS As SocketException
Close()
SharedMethods.RaiseError(myParent, exS.Message, "", exS.ErrorCode)
Catch exO As ObjectDisposedException
Exit Try
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
''' <summary>
''' The callback method for the Send method (TCP only) - loops if not all the data was sent.
''' </summary>
Private Sub SendCallback(ByVal ar As IAsyncResult)
Try
' Retrieve the AsyncState
Dim ssState As SendState = DirectCast(ar.AsyncState, SendState)
' Update the total sent - while ending this send call
ssState.TotalSent += ssState.SendingSocket.EndSend(ar)
' Build the event args for the event that will be raised
Dim e As New WinsockSendEventArgs(RemoteIP, ssState.TotalSent, ssState.Length)
If ssState.SendCompleted Then
' Object finished sending - raise the SendComplete event
myParent.OnSendComplete(e)
' Check for more items in the buffer - if so run the send again
' we can't run DoSend from within the SyncLock, or we'll run into
' a deadlock
Dim blnRunAgain As Boolean = False
SyncLock sendBuffer.SyncRoot
If sendBuffer.Count > 0 Then
blnRunAgain = True
End If
End SyncLock
If blnRunAgain Then DoSend()
Else
' Raise SendProgress event
myParent.OnSendProgress(e)
' Object still has more data in the buffer, get the next part and send it
Dim byt() As Byte
SyncLock sendBuffer.SyncRoot
Dim sqData As SendQueueData = sendBuffer.Dequeue()
byt = sqData.Data
If byt.GetUpperBound(0) > incBufferSize Then
Dim tmpByt() As Byte = SharedMethods.ShrinkArray(byt, incBufferSize + 1)
sqData.Data = DirectCast(byt.Clone(), Byte())
sendBuffer.Push(sqData)
byt = DirectCast(tmpByt.Clone(), Byte())
End If
End SyncLock
ssState.SendingSocket.BeginSend(byt, 0, byt.Length, SocketFlags.None, New AsyncCallback(AddressOf SendCallback), ssState)
End If
Catch exS As SocketException
Close()
SharedMethods.RaiseError(myParent, exS.Message, "", exS.ErrorCode)
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
''' <summary>
''' The callback method for the Send method (UDP only) - loops if not all the data was sent.
''' </summary>
Private Sub SendToCallback(ByVal ar As IAsyncResult)
Try
' Retrieve the AsyncState
Dim ssState As SendState = DirectCast(ar.AsyncState, SendState)
' Update the total sent - while ending this send call
ssState.TotalSent += ssState.SendingSocket.EndSendTo(ar)
' Build the event args for the event that will be raised
Dim e As New WinsockSendEventArgs(ssState.SendToAddress.Address.ToString(), ssState.TotalSent, ssState.Length)
If ssState.SendCompleted Then
' Object finished sending - raise the SendComplete event
myParent.OnSendComplete(e)
' Check for more items in the buffer - if so run the send again
' we can't run DoSend from within the SyncLock, or we'll run into
' a deadlock
Dim blnRunAgain As Boolean = False
SyncLock sendBuffer.SyncRoot
If sendBuffer.Count > 0 Then
blnRunAgain = True
End If
End SyncLock
If blnRunAgain Then DoSend()
Else
' Raise the SendProgress event
myParent.OnSendProgress(e)
' Object still has more data in the buffer, get it and send it.
Dim byt() As Byte
SyncLock sendBuffer.SyncRoot
Dim sqData As SendQueueData = sendBuffer.Dequeue()
byt = sqData.Data
If byt.GetUpperBound(0) > incBufferSize Then
Dim tmpByt() As Byte = SharedMethods.ShrinkArray(byt, incBufferSize + 1)
sqData.Data = DirectCast(byt.Clone(), Byte())
sendBuffer.Push(sqData)
byt = DirectCast(tmpByt.Clone(), Byte())
End If
End SyncLock
ssState.SendingSocket.BeginSendTo(byt, 0, byt.Length, SocketFlags.None, ssState.SendToAddress, New AsyncCallback(AddressOf SendToCallback), ssState)
End If
Catch exS As SocketException
Close()
SharedMethods.RaiseError(myParent, exS.Message, "", exS.ErrorCode)
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
''' <summary>
''' The callback method for resolving the address given - starts the socket on connecting.
''' </summary>
Public Sub DoConnectCallback(ByVal ar As IAsyncResult)
Try
Dim port As Integer = DirectCast(ar.AsyncState, Integer)
Dim resolved As IPHostEntry
Try
resolved = Dns.EndResolve(ar)
Catch ex As SocketException
resolved = Nothing
End Try
If resolved Is Nothing OrElse resolved.AddressList.Length = 0 Then
Dim name As String = CStr(IIf(Not (resolved Is Nothing), """" & resolved.HostName & """ ", ""))
Throw New Exception("Hostname " & name & "could not be resolved.")
End If
myParent.ChangeState(WinsockStates.HostResolved)
Connect(resolved.AddressList(0), port)
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
#End Region
#Region " Private Methods "
''' <summary>
''' Processes raw data that was received from the socket and places it into the appropriate buffer.
''' </summary>
''' <param name="byt">The raw byte buffer containing the data received from the socket.</param>
''' <param name="iSize">The size of the data received from the socket (reported from the EndReceive).</param>
''' <param name="source_ip">The IP address the data came from, used for event raising.</param>
''' <param name="source_port">The Port the data arrived on, used for event raising.</param>
Private Sub ProcessIncoming(ByVal byt() As Byte, ByVal iSize As Integer, ByVal source_ip As String, ByVal source_port As Integer)
'check if we are using LegacySupport
If myParent.LegacySupport Then
' legacy support is active just output the data to the buffer
' if we actually received some data
If iSize > 0 Then
' yes we received some data - resize the array
ResizeArray(byt, iSize)
' add the byte array to the buffer queue
qBuffer.Enqueue(byt)
' raise the DataArrival event
Dim e As New WinsockDataArrivalEventArgs(iSize, source_ip, source_port)
myParent.OnDataArrival(e)
End If
Else
' legacy support is inactive
' if total size is <= 0 no data came in - exit
If iSize <= 0 Then Exit Sub
' reduce the size of the array to the reported size (fixes trailling zeros)
ResizeArray(byt, iSize)
' Do we have a packet header?
If Not phProcessor.Completed And _buff.Count > 1 Then
' no packet header and already have more than enough data for a header
' most likely no header is coming - throw an error to use LegacySupport
phProcessor.Reset()
Throw New Exception("Unable to determine size of incoming packet. It's possible you may need to use Legacy Support.")
ElseIf Not phProcessor.Completed Then
phProcessor.ProcessHeader(byt, _buff)
If Not phProcessor.Completed Then
ProcessIncoming(byt, byt.Length, source_ip, source_port)
End If
End If
' Packet Header obtained... Process data, raise data arrival when all is received
If _buff.Count = 0 AndAlso byt.Length >= phProcessor.Size Then
' everything is located in the byte array
If byt.GetUpperBound(0) > phProcessor.Size Then
' everything and MORE in byte array
' remove our data, and run process again on the rest
Dim tmp() As Byte = SharedMethods.ShrinkArray(byt, phProcessor.Size)
' add the data to the queue
qBuffer.Enqueue(tmp)
' reset the packet header processor
phProcessor.Reset()
' raise the received progress event
Dim eR As New WinsockReceiveProgressEventArgs(source_ip, tmp.Length, phProcessor.Size)
myParent.OnReceiveProgress(eR)
' raise the DataArrival event
Dim e As New WinsockDataArrivalEventArgs(tmp.Length, source_ip, source_port)
myParent.OnDataArrival(e)
' process the extra data
ProcessIncoming(byt, byt.Length, source_ip, source_port)
Else
' add everything to the queue
qBuffer.Enqueue(byt)
' raise the received progress event
Dim eR As New WinsockReceiveProgressEventArgs(source_ip, byt.Length, phProcessor.Size)
myParent.OnReceiveProgress(eR)
' reset the packet header processor
phProcessor.Reset()
' raise the DataArrival event
Dim e As New WinsockDataArrivalEventArgs(byt.Length, source_ip, source_port)
myParent.OnDataArrival(e)
End If
ElseIf _buff.Count > 0 AndAlso _buff.Combine().Length + byt.Length >= phProcessor.Size Then
' if you include the temp buffer, we have all the data
' get everything
_buff.Add(byt)
Dim tmp() As Byte = _buff.Combine()
' clear the temp buffer
_buff.Clear()
If tmp.GetUpperBound(0) > phProcessor.Size Then
' tmp contains more than what we need
' remove what wee need, and then run the process again on the rest
Dim t2() As Byte = SharedMethods.ShrinkArray(tmp, phProcessor.Size)
' add the data to the queue
qBuffer.Enqueue(t2)
' raise the received progress event
Dim eR As New WinsockReceiveProgressEventArgs(source_ip, tmp.Length, phProcessor.Size)
myParent.OnReceiveProgress(eR)
' reset the packet header processor
phProcessor.Reset()
' raise the DataArrival event
Dim e As New WinsockDataArrivalEventArgs(tmp.Length, source_ip, source_port)
myParent.OnDataArrival(e)
' process the extra data
ProcessIncoming(tmp, tmp.Length, source_ip, source_port)
Else
' tmp contains only what we need - add it to the queue
qBuffer.Enqueue(tmp)
' raise the received progress event
Dim eR As New WinsockReceiveProgressEventArgs(source_ip, tmp.Length, phProcessor.Size)
myParent.OnReceiveProgress(eR)
' reset the packet header processor
phProcessor.Reset()
' raise the DataArrival event
Dim e As New WinsockDataArrivalEventArgs(tmp.Length, source_ip, source_port)
myParent.OnDataArrival(e)
End If
Else
_buff.Add(byt)
' raise the received progress event
Dim eR As New WinsockReceiveProgressEventArgs(source_ip, _buff.Combine.Length, phProcessor.Size)
myParent.OnReceiveProgress(eR)
End If
End If
End Sub
''' <summary>
''' Resizes an array to the desired length - preserving the data at the begining of the array.
''' </summary>
''' <param name="byt">The array to be resized.</param>
''' <param name="iSize">The size to resize the array to.</param>
Private Sub ResizeArray(ByRef byt() As Byte, ByVal iSize As Integer)
If iSize - 1 < byt.GetUpperBound(0) Then
ReDim Preserve byt(iSize - 1)
End If
End Sub
''' <summary>
''' Starts listening for incoming packets on the socket.
''' </summary>
''' <remarks>The is private because, the user should never have to call this.</remarks>
Private Sub Receive()
Try
mSock1.BeginReceive(byteBuffer, 0, incBufferSize, SocketFlags.None, New AsyncCallback(AddressOf ReceiveCallback), Nothing)
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
''' <summary>
''' Starts the sending of an object in the send buffer.
''' </summary>
Private Sub DoSend()
Try
' Retrieve the bytes to send
Dim byt() As Byte
Dim fullSize As Integer
Dim remEP As IPEndPoint
SyncLock sendBuffer.SyncRoot
If sendBuffer.Count = 0 Then Exit Sub
Dim sqData As SendQueueData = sendBuffer.Dequeue()
If sqData Is Nothing Then
Throw New Exception("Buffer count was greater than zero, yet a nothing was retrieved. Something broke.")
End If
remEP = sqData.IPAddress
byt = sqData.Data()
fullSize = byt.GetUpperBound(0)
If byt.GetUpperBound(0) > incBufferSize Then
Dim tmpByt() As Byte = SharedMethods.ShrinkArray(byt, incBufferSize + 1)
sqData.Data = DirectCast(byt.Clone(), Byte())
sendBuffer.Push(sqData)
byt = DirectCast(tmpByt.Clone(), Byte())
End If
End SyncLock
' Send according to the appropriate protocol
If myParent.Protocol = WinsockProtocol.Tcp Then
Dim ssState As SendState = SendState.Build(fullSize, mSock1, incBufferSize)
mSock1.BeginSend(byt, 0, byt.Length, SocketFlags.None, New AsyncCallback(AddressOf SendCallback), ssState)
ElseIf myParent.Protocol = WinsockProtocol.Udp Then
Dim tmpSock As New Socket(remEP.AddressFamily, SocketType.Dgram, ProtocolType.Udp)
Dim ssState As SendState = SendState.Build(fullSize, tmpSock, incBufferSize)
ssState.SendToAddress = remEP
tmpSock.BeginSendTo(byt, 0, byt.Length, SocketFlags.None, remEP, New AsyncCallback(AddressOf SendToCallback), ssState)
End If
Catch ex As Exception
SharedMethods.RaiseError(myParent, ex.Message)
End Try
End Sub
#End Region
#Region " Private Classes "
''' <summary>
''' A class that decodes and stores the packet header information.
''' </summary>
Public Class PacketHeader
Private _delimiter As Byte
Private _hasDelim As Boolean
Private _size As Integer
Public Sub New()
_size = -1
