Imports System.IO
Imports System.Net
Imports System.Net.Sockets

Namespace Query
	Class RCONQuery
		Private qSocket As Socket
		Private address As IPAddress
		Private _port As Integer = 0
		Private _password As String = Nothing

		Private results As String() = New String(49) {}
		Private _count As Integer = 0

		Public Sub New(IP As String, port As Integer, password As String)
			qSocket = New Socket(AddressFamily.InterNetwork, SocketType.Dgram, ProtocolType.Udp)

			qSocket.SendTimeout = 5000
			qSocket.ReceiveTimeout = 5000

			Try
				address = Dns.GetHostAddresses(IP)(0)


			Catch
			End Try

			_port = port
			_password = password
		End Sub

		Public Function Send(command As String) As Boolean
			Try
				Dim endpoint As New IPEndPoint(address, _port)

				Using stream As New MemoryStream()
					Using writer As New BinaryWriter(stream)
						writer.Write("SAMP".ToCharArray())

						Dim SplitIP As String() = address.ToString().Split("."C)

						writer.Write(Convert.ToByte(Convert.ToInt32(SplitIP(0))))
						writer.Write(Convert.ToByte(Convert.ToInt32(SplitIP(1))))
						writer.Write(Convert.ToByte(Convert.ToInt32(SplitIP(2))))
						writer.Write(Convert.ToByte(Convert.ToInt32(SplitIP(3))))

						writer.Write(CUShort(_port))

						writer.Write("x"C)

						writer.Write(CUShort(_password.Length))
						writer.Write(_password.ToCharArray())

						writer.Write(CUShort(command.Length))
						writer.Write(command.ToCharArray())
					End Using

					If qSocket.SendTo(stream.ToArray(), endpoint) > 0 Then
						Return True
					End If
				End Using

			Catch
				Return False
			End Try

			Return False
		End Function

		Public Function Rceive() As Integer
			Try
				For i As Integer = 0 To results.GetLength(0) - 1
					results.SetValue(Nothing, i)
				Next

				_count = 0

				Dim endpoint As EndPoint = New IPEndPoint(address, _port)

				Dim rBuffer As Byte() = New Byte(499) {}

				Dim count As Integer = qSocket.ReceiveFrom(rBuffer, endpoint)

				Using stream As New MemoryStream(rBuffer)
					Using reader As New BinaryReader(stream)
						If stream.Length <= 11 Then
							Return _count
						End If

						reader.ReadBytes(11)
						Dim len As Short

						Try
							While (InlineAssignHelper(len, reader.ReadInt16())) <> 0
								results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = New String(reader.ReadChars(Convert.ToInt32(len)))
							End While

						Catch
							Return _count
						End Try
					End Using
				End Using

			Catch
				Return _count
			End Try

			Return _count
		End Function

		Public Function Store(count As Integer) As String()
			Dim rString As String() = New String(count - 1) {}

			Dim i As Integer = 0
			While i < count AndAlso i < _count
				rString(i) = results(i)
				i += 1
			End While

			_count = 0

			Return rString
		End Function
		Private Shared Function InlineAssignHelper(Of T)(ByRef target As T, value As T) As T
			target = value
			Return value
		End Function
	End Class

	Class Query
		Private qSocket As Socket
		Private address As IPAddress
		Private _port As Integer = 0

		Private results As String()
		Private _count As Integer = 0

		Private timestamp As DateTime() = New DateTime(1) {}

		Public Sub New(IP As String, port As Integer)
			qSocket = New Socket(AddressFamily.InterNetwork, SocketType.Dgram, ProtocolType.Udp)

			qSocket.SendTimeout = 5000
			qSocket.ReceiveTimeout = 5000

			Try
				address = Dns.GetHostAddresses(IP)(0)


			Catch
			End Try

			_port = port
		End Sub

		Public Function Send(opcode As Char) As Boolean
			Try
				Dim endpoint As EndPoint = New IPEndPoint(address, _port)

				Using stream As New MemoryStream()
					Using writer As New BinaryWriter(stream)
						writer.Write("SAMP".ToCharArray())

						Dim SplitIP As String() = address.ToString().Split("."C)

						writer.Write(Convert.ToByte(Convert.ToInt32(SplitIP(0))))
						writer.Write(Convert.ToByte(Convert.ToInt32(SplitIP(1))))
						writer.Write(Convert.ToByte(Convert.ToInt32(SplitIP(2))))
						writer.Write(Convert.ToByte(Convert.ToInt32(SplitIP(3))))

						writer.Write(CUShort(_port))

						writer.Write(opcode)

						If opcode = "p"C Then
							writer.Write("8493".ToCharArray())
						End If

						timestamp(0) = DateTime.Now
					End Using

					If qSocket.SendTo(stream.ToArray(), endpoint) > 0 Then
						Return True
					End If
				End Using

			Catch
				Return False
			End Try

			Return False
		End Function

		Public Function Receive() As Integer
			Try
				_count = 0

				Dim endpoint As EndPoint = New IPEndPoint(address, _port)

				Dim rBuffer As Byte() = New Byte(3401) {}
				qSocket.ReceiveFrom(rBuffer, endpoint)

				timestamp(1) = DateTime.Now

				Using stream As New MemoryStream(rBuffer)
					Using reader As New BinaryReader(stream)
						If stream.Length <= 10 Then
							Return _count
						End If

						reader.ReadBytes(10)

						Select Case reader.ReadChar()
							Case "i"C
								' Information
								If True Then
									results = New String(5) {}

									results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = Convert.ToString(reader.ReadByte())

									results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = Convert.ToString(reader.ReadInt16())

									results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = Convert.ToString(reader.ReadInt16())

									Dim hostnamelen As Integer = reader.ReadInt32()
									results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = New String(reader.ReadChars(hostnamelen))

									Dim gamemodelen As Integer = reader.ReadInt32()
									results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = New String(reader.ReadChars(gamemodelen))

									Dim mapnamelen As Integer = reader.ReadInt32()
									results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = New String(reader.ReadChars(mapnamelen))

									Return _count
								End If

							Case "r"C
								' Rules
								If True Then
									Dim rulecount As Integer = reader.ReadInt16()

									results = New String(rulecount * 2 - 1) {}

									For i As Integer = 0 To rulecount - 1
										Dim rulelen As Integer = reader.ReadByte()
										results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = New String(reader.ReadChars(rulelen))

										Dim valuelen As Integer = reader.ReadByte()
										results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = New String(reader.ReadChars(valuelen))
									Next

									Return _count
								End If

							Case "c"C
								' Client list
								If True Then
									Dim playercount As Integer = reader.ReadInt16()

									results = New String(playercount * 2 - 1) {}

									For i As Integer = 0 To playercount - 1
										Dim namelen As Integer = reader.ReadByte()
										results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = New String(reader.ReadChars(namelen))

										results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = Convert.ToString(reader.ReadInt32())
									Next

									Return _count
								End If

							Case "d"C
								' Detailed player information
								If True Then
									Dim playercount As Integer = reader.ReadInt16()

									results = New String(playercount * 4 - 1) {}

									For i As Integer = 0 To playercount - 1
										results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = Convert.ToString(reader.ReadByte())

										Dim namelen As Integer = reader.ReadByte()
										results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = New String(reader.ReadChars(namelen))

										results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = Convert.ToString(reader.ReadInt32())
										results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = Convert.ToString(reader.ReadInt32())
									Next

									Return _count
								End If

							Case "p"C
								' Ping
								If True Then
									results = New String(0) {}

									results(System.Math.Max(System.Threading.Interlocked.Increment(_count),_count - 1)) = timestamp(1).Subtract(timestamp(0)).Milliseconds.ToString()

									Return _count
								End If
							Case Else

								Return _count
						End Select
					End Using
				End Using

			Catch
				Return _count
			End Try
		End Function

		Public Function Store(count As Integer) As String()
			Dim rString As String() = New String(count - 1) {}

			Dim i As Integer = 0
			While i < count AndAlso i < _count
				rString(i) = results(i)
				i += 1
			End While

			_count = 0

			Return rString
		End Function
	End Class
End Namespace