Add XMLRPC::RackServer

herwinw · herwinw · commit c3f00ceb7ec3 · 2025-04-24T10:26:25.000+02:00
This is a simple Rack middleware that acts as an XMLRPC server. This can be used
in any Rack compatible webserver.
diff --git a/Gemfile b/Gemfile
@@ -6,6 +6,8 @@ group :development do
   gem "bundler"
   gem "libxml-ruby", platforms: [:ruby, :jruby]
   gem "nokogiri", platforms: [:ruby, :jruby]
+  gem "rack"
+  gem "rack-test"
   gem "rake"
   gem "test-unit"
 end
diff --git a/lib/xmlrpc/server.rb b/lib/xmlrpc/server.rb
@@ -532,6 +532,49 @@ def http_write(body, status, header)
 end
 
 
+# Implements a XML-RPC server, which works with Rack
+class RackServer < BasicServer
+
+  # This method processes a XML-RPC method call and sends the answer
+  # back to the client.
+  def call(env)
+    length = env['CONTENT_LENGTH'].to_i
+
+    return http_error(405, "Method Not Allowed") unless env['REQUEST_METHOD'] == "POST"
+    return http_error(400, "Bad Request")        unless parse_content_type(env['CONTENT_TYPE']).first == "text/xml"
+    return http_error(411, "Length Required")    unless length > 0
+
+    req = Rack::Request.new(env)
+    data = req.body.read(length)
+
+    return http_error(400, "Bad Request")        if data.nil? or data.bytesize != length
+
+    [200, { "Content-type" => "text/xml; charset=utf-8" }, [process(data)]]
+  end
+
+
+  private
+
+  def http_error(status, message)
+    err = "#{status} #{message}"
+    msg = <<-"MSGEND"
+      <html>
+        <head>
+          <title>#{err}</title>
+        </head>
+        <body>
+          <h1>#{err}</h1>
+          <p>Unexpected error occurred while processing XML-RPC request!</p>
+        </body>
+      </html>
+    MSGEND
+
+    [status, { "Content-Type" => "text/html" }, [msg]]
+  end
+
+end
+
+
 class WEBrickServlet < BasicServer; end # forward declaration
 
 # Implements a standalone XML-RPC server. The method XMLRPC::Server#serve is
diff --git a/test/test_rack_server.rb b/test/test_rack_server.rb
@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+require "test/unit"
+require "rack/test"
+require "xmlrpc/server"
+require 'xmlrpc/create'
+require 'xmlrpc/parser'
+
+class TestRack < Test::Unit::TestCase
+  include Rack::Test::Methods
+
+  def app
+    s = XMLRPC::RackServer.new
+
+    s.add_handler("test.add") do |a,b|
+      a + b
+    end
+
+    s.add_handler("test.div") do |a,b|
+      if b == 0
+        raise XMLRPC::FaultException.new(1, "division by zero")
+      else
+        a / b
+      end
+    end
+
+    s.set_default_handler do |name, *args|
+      raise XMLRPC::FaultException.new(-99, "Method #{name} missing" +
+            " or wrong number of parameters!")
+    end
+
+    s.add_introspection
+
+    s
+  end
+
+  def test_successful_call
+    assert_equal([true, 9], call("test.add", 4, 5))
+  end
+
+  def test_fault_exception
+    assert_equal([false, XMLRPC::FaultException.new(1, "division by zero")], call("test.div", 1, 0))
+  end
+
+  def test_introspection
+    methods = ["test.add", "test.div", "system.listMethods", "system.methodSignature", "system.methodHelp"]
+    assert_equal([true, methods], call("system.listMethods"))
+  end
+
+  def test_missing_handler
+    assert_equal([false, XMLRPC::FaultException.new(-99, "Method test.nonexisting missing or wrong number of parameters!")], call("test.nonexisting"))
+  end
+
+  def test_wrong_number_of_arguments
+    assert_equal([false, XMLRPC::FaultException.new(-99, "Method test.add missing or wrong number of parameters!")], call("test.add", 1, 2, 3))
+  end
+
+  def test_multibyte_characters
+    assert_equal([true, "あいうえおかきくけこ"], call('test.add', "あいうえお", "かきくけこ"))
+  end
+
+  def call(methodname, *args)
+    create = XMLRPC::Create.new(XMLRPC::Config.default_writer.new)
+    parser = XMLRPC::Config.default_parser.new
+
+    request = create.methodCall(methodname, *args)
+    post("/", request, 'CONTENT_TYPE' => 'text/xml')
+    assert(last_response.ok?)
+    parser.parseMethodResponse(last_response.body)
+  end
+end
