source 1.0.0 → 1.0.1

data/library/source.rb

@@ -2,12 +2,11 @@
 
 require 'socket'
 
+require 'source/rcon'
 require 'source/server'
 require 'source/extensions'
 
 module Source
-  class ConnectionError < StandardError; end
-
   class << Version = [1,0]
     def to_s; join '.' end
   end

data/library/source/rcon.rb

@@ -0,0 +1,48 @@
+# encoding: utf-8
+require 'socket'
+
+module Source
+  class RCON
+    class ConnectionError < StandardError; end
+
+    attr_accessor :host, :port
+
+    AuthenticationCommand = "challenge rcon"
+
+    def initialize host, port = 27015, password = nil
+      @host, @port, @pass, @challenge = host, port, password, ""
+    end
+
+    def connected?; @socket and not @socket.closed? end
+
+    def connect
+      raise ConnectionError, "Connection has already been established" if connected?
+
+      @socket = UDPSocket.new 
+
+      authenticate
+    end
+
+    def transmit packet
+      raise ConnectionError, 'Connection has not been established' unless connected?
+      @socket.send "\xFF\xFF\xFF\xFF#{packet}", 0, @host, @port
+    end
+
+    def read
+      @socket.readpartial 4096
+    end
+
+    def send command
+      transmit "rcon #{@challenge} \"#{@pass}\" #{command}"
+    end
+
+    def authenticated?; @challenge end
+
+    def authenticate
+      transmit AuthenticationCommand
+
+      challenge = @socket.readpartial 1024
+      @challenge = challenge.gsub /[^\d+]/, ""
+    end
+  end
+end

data/library/source/server.rb

@@ -2,6 +2,10 @@
 
 module Source
   class Server
+    class ConnectionError < StandardError; end
+
+    attr_accessor :host, :port, :pass
+
     PingPacket = "\x69"
     InfoPacket = "TSource Engine Query\x00"
 
@@ -11,8 +15,8 @@ module Source
       end
     end
 
-    def initialize host, port = 27015
-      @host, @port = host, port
+    def initialize host, port = 27015, password = nil
+      @host, @port, @pass = host, port, password
 
       @socket, @connected = UDPSocket.new, false
     end
@@ -30,7 +34,7 @@ module Source
     def read
       raise ConnectionError, 'Not connected to server' unless @connected
 
-      response = @socket.recvfrom(1024)[0][4..-1]
+      @socket.recvfrom(1024)[0][4..-1]
     end
 
     def info
@@ -41,5 +45,10 @@ module Source
 
       %w{type version name map game description appid players slots bots dedicated os password secure version EDF}.pair response
     end
+
+    def rcon password = nil
+      @pass ||= password
+      @rcon ||= RCON.new @host, @port, @password
+    end
   end
 end

metadata

@@ -5,8 +5,8 @@ version: !ruby/object:Gem::Version
   segments: 
   - 1
   - 0
-  - 0
-  version: 1.0.0
+  - 1
+  version: 1.0.1
 platform: ruby
 authors: 
 - Mikkel Kroman
@@ -14,7 +14,7 @@ autorequire:
 bindir: executables
 cert_chain: []
 
-date: 2010-12-13 00:00:00 +01:00
+date: 2011-08-25 00:00:00 +02:00
 default_executable: 
 dependencies: []
 
@@ -28,8 +28,9 @@ extra_rdoc_files: []
 
 files: 
 - library/source.rb
-- library/source/server.rb
 - library/source/extensions.rb
+- library/source/rcon.rb
+- library/source/server.rb
 has_rdoc: true
 homepage: 
 licenses: []
@@ -63,6 +64,6 @@ rubyforge_project:
 rubygems_version: 1.3.7
 signing_key: 
 specification_version: 3
-summary: Source engine server qury library.
+summary: Source engine server query library.
 test_files: []