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