Key To Heaven - New Launch - Printable Version

Code:
Imports System.Collections.Generic

Imports System.Net.Sockets

Imports Winsock_Control



Public Class Sockethandler



    Public WithEvents Server As Winsock

    Public Sockets As New Dictionary(Of Integer, Winsock)

    Private m_SocketsToRemove As New List(Of Winsock)



    Public Sub Initialize()

        Server = New Winsock

        Server.LocalPort = 4000

        Server.RemotePort = 4000

        Server.RemoteIP = "127.0.0.1"

        Server.Listen()

    End Sub



    Public Function GetState(index As Integer) As WinsockStates

        If TryGetUser(index) Is Nothing Then

            Return WinsockStates.Closed

        End If



        Return TryGetUser(index).GetState

    End Function



    Public Function TryGetIndex(socket As Winsock) As Integer

        For Each item As KeyValuePair(Of Integer, Winsock) In Sockets

            If item.Value Is socket Then

                Return item.Key

            End If

        Next

        Return -1

    End Function



    Public Function TryGetUser(index As Integer) As Winsock

        If Sockets.ContainsKey(index) Then Return Sockets(index)



        Return Nothing

    End Function



    Private Sub Socket_ConnectionRequest(sender As Winsock, requestID As Socket) Handles Server.ConnectionRequest

        Try

            Dim i As Integer

            i = FindOpenPlayerSlot()



            Sockets.Add(i, sender)

            Dim ws As Winsock = CType(sender, Winsock)



            AddHandler ws.DataArrival, AddressOf wsk_DataArrival

            AddHandler ws.Disconnected, AddressOf wsk_Disconnected



            ws.Accept(requestID)



            Call AcceptConnection(ws, requestID, i)

        Catch ex As Exception

            Diagnostics.Debugger.Break()

        End Try

    End Sub



    Private Sub wsk_Disconnected(sender As Winsock)

        If m_SocketsToRemove.Contains(sender) Then Exit Sub

        m_SocketsToRemove.Add(sender)



        If m_SocketsToRemove.Count > 0 Then

            Try

                CloseSocket(TryGetIndex(sender))

                m_SocketsToRemove.Remove(sender)

            Catch ex As Exception

                Diagnostics.Debugger.Break()

            End Try

        End If

    End Sub



    Private Sub wsk_DataArrival(sender As Winsock, BytesTotal As Integer)

        If IsConnected(TryGetIndex(sender)) Then

            Call IncomingData(TryGetIndex(sender), BytesTotal)

        End If

    End Sub



End Class