1#!/usr/bin/env ruby 2 3# 4# Licensed to the Apache Software Foundation (ASF) under one 5# or more contributor license agreements. See the NOTICE file 6# distributed with this work for additional information 7# regarding copyright ownership. The ASF licenses this file 8# to you under the Apache License, Version 2.0 (the 9# "License"); you may not use this file except in compliance 10# with the License. You may obtain a copy of the License at 11# 12# http://www.apache.org/licenses/LICENSE-2.0 13# 14# Unless required by applicable law or agreed to in writing, 15# software distributed under the License is distributed on an 16# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY 17# KIND, either express or implied. See the License for the 18# specific language governing permissions and limitations 19# under the License. 20# 21 22$:.push File.dirname(__FILE__) + '/..' 23 24require 'test_helper' 25require 'thrift' 26require 'thrift_test' 27require 'thrift_test_types' 28 29class SimpleHandler 30 [:testVoid, :testString, :testBool, :testByte, :testI32, :testI64, :testDouble, :testBinary, 31 :testStruct, :testMap, :testStringMap, :testSet, :testList, :testNest, :testEnum, :testTypedef, 32 :testEnum, :testTypedef, :testMultiException].each do |meth| 33 34 define_method(meth) do |thing| 35 p meth 36 p thing 37 thing 38 end 39 40 end 41 42 def testVoid() 43 end 44 45 def testInsanity(thing) 46 return { 47 1 => { 48 2 => thing, 49 3 => thing 50 }, 51 2 => { 52 6 => Thrift::Test::Insanity::new() 53 } 54 } 55 end 56 57 def testMapMap(thing) 58 return { 59 -4 => { 60 -4 => -4, 61 -3 => -3, 62 -2 => -2, 63 -1 => -1, 64 }, 65 4 => { 66 4 => 4, 67 3 => 3, 68 2 => 2, 69 1 => 1, 70 } 71 } 72 end 73 74 def testMulti(arg0, arg1, arg2, arg3, arg4, arg5) 75 return Thrift::Test::Xtruct.new({ 76 'string_thing' => 'Hello2', 77 'byte_thing' => arg0, 78 'i32_thing' => arg1, 79 'i64_thing' => arg2, 80 }) 81 end 82 83 def testException(thing) 84 if thing == "Xception" 85 raise Thrift::Test::Xception, :errorCode => 1001, :message => thing 86 elsif thing == "TException" 87 raise Thrift::Exception, :message => thing 88 else 89 # no-op 90 end 91 end 92 93 def testMultiException(arg0, arg1) 94 if arg0 == "Xception2" 95 raise Thrift::Test::Xception2, :errorCode => 2002, :struct_thing => ::Thrift::Test::Xtruct.new({ :string_thing => 'This is an Xception2' }) 96 elsif arg0 == "Xception" 97 raise Thrift::Test::Xception, :errorCode => 1001, :message => 'This is an Xception' 98 else 99 return ::Thrift::Test::Xtruct.new({'string_thing' => arg1}) 100 end 101 end 102 103 def testOneway(arg0) 104 sleep(arg0) 105 end 106 107end 108 109domain_socket = nil 110port = 9090 111protocol = "binary" 112@protocolFactory = nil 113ssl = false 114transport = "buffered" 115@transportFactory = nil 116 117ARGV.each do|a| 118 if a == "--help" 119 puts "Allowed options:" 120 puts "\t -h [ --help ] \t produce help message" 121 puts "\t--domain-socket arg (=) \t Unix domain socket path" 122 puts "\t--port arg (=9090) \t Port number to listen \t not valid with domain-socket" 123 puts "\t--protocol arg (=binary) \t protocol: accel, binary, compact, json" 124 puts "\t--ssl \t use ssl \t not valid with domain-socket" 125 puts "\t--transport arg (=buffered) transport: buffered, framed, http" 126 exit 127 elsif a.start_with?("--domain-socket") 128 domain_socket = a.split("=")[1] 129 elsif a.start_with?("--protocol") 130 protocol = a.split("=")[1] 131 elsif a == "--ssl" 132 ssl = true 133 elsif a.start_with?("--transport") 134 transport = a.split("=")[1] 135 elsif a.start_with?("--port") 136 port = a.split("=")[1].to_i 137 end 138end 139 140if protocol == "binary" || protocol.to_s.strip.empty? 141 @protocolFactory = Thrift::BinaryProtocolFactory.new 142elsif protocol == "compact" 143 @protocolFactory = Thrift::CompactProtocolFactory.new 144elsif protocol == "json" 145 @protocolFactory = Thrift::JsonProtocolFactory.new 146elsif protocol == "accel" 147 @protocolFactory = Thrift::BinaryProtocolAcceleratedFactory.new 148else 149 raise 'Unknown protocol type' 150end 151 152if transport == "buffered" || transport.to_s.strip.empty? 153 @transportFactory = Thrift::BufferedTransportFactory.new 154elsif transport == "framed" 155 @transportFactory = Thrift::FramedTransportFactory.new 156else 157 raise 'Unknown transport type' 158end 159 160@handler = SimpleHandler.new 161@processor = Thrift::Test::ThriftTest::Processor.new(@handler) 162@transport = nil 163if domain_socket.to_s.strip.empty? 164 if ssl 165 # the working directory for ruby crosstest is test/rb/gen-rb 166 keysDir = File.join(File.dirname(File.dirname(Dir.pwd)), "keys") 167 ctx = OpenSSL::SSL::SSLContext.new 168 ctx.ca_file = File.join(keysDir, "CA.pem") 169 ctx.cert = OpenSSL::X509::Certificate.new(File.open(File.join(keysDir, "server.crt"))) 170 ctx.cert_store = OpenSSL::X509::Store.new 171 ctx.cert_store.add_file(File.join(keysDir, 'client.pem')) 172 ctx.key = OpenSSL::PKey::RSA.new(File.open(File.join(keysDir, "server.key"))) 173 ctx.options = OpenSSL::SSL::OP_NO_SSLv2 | OpenSSL::SSL::OP_NO_SSLv3 174 ctx.ssl_version = :SSLv23 175 ctx.verify_mode = OpenSSL::SSL::VERIFY_PEER 176 @transport = Thrift::SSLServerSocket.new(nil, port, ctx) 177 else 178 @transport = Thrift::ServerSocket.new(port) 179 end 180else 181 @transport = Thrift::UNIXServerSocket.new(domain_socket) 182end 183 184@server = Thrift::ThreadedServer.new(@processor, @transport, @transportFactory, @protocolFactory) 185 186puts "Starting TestServer #{@server.to_s}" 187@server.serve 188puts "done." 189