1 #!/usr/bin/env python2.7 2 # Copyright 2015 gRPC authors. 3 # 4 # Licensed under the Apache License, Version 2.0 (the "License"); 5 # you may not use this file except in compliance with the License. 6 # You may obtain a copy of the License at 7 # 8 # http://www.apache.org/licenses/LICENSE-2.0 9 # 10 # Unless required by applicable law or agreed to in writing, software 11 # distributed under the License is distributed on an "AS IS" BASIS, 12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 # See the License for the specific language governing permissions and 14 # limitations under the License. 15 16 """Server for httpcli_test""" 17 18 import argparse 19 import BaseHTTPServer 20 import os 21 import ssl 22 import sys 23 24 _PEM = os.path.abspath(os.path.join(os.path.dirname(sys.argv[0]), '../../..', 'src/core/tsi/test_creds/server1.pem')) 25 _KEY = os.path.abspath(os.path.join(os.path.dirname(sys.argv[0]), '../../..', 'src/core/tsi/test_creds/server1.key')) 26 print _PEM 27 open(_PEM).close() 28 29 argp = argparse.ArgumentParser(description='Server for httpcli_test') 30 argp.add_argument('-p', '--port', default=10080, type=int) 31 argp.add_argument('-s', '--ssl', default=False, action='store_true') 32 args = argp.parse_args() 33 34 print 'server running on port %d' % args.port 35 36 class Handler(BaseHTTPServer.BaseHTTPRequestHandler): 37 def good(self): 38 self.send_response(200) 39 self.send_header('Content-Type', 'text/html') 40 self.end_headers() 41 self.wfile.write('<html><head><title>Hello world!</title></head>') 42 self.wfile.write('<body><p>This is a test</p></body></html>') 43 44 def do_GET(self): 45 if self.path == '/get': 46 self.good() 47 48 def do_POST(self): 49 content = self.rfile.read(int(self.headers.getheader('content-length'))) 50 if self.path == '/post' and content == 'hello': 51 self.good() 52 53 httpd = BaseHTTPServer.HTTPServer(('localhost', args.port), Handler) 54 if args.ssl: 55 httpd.socket = ssl.wrap_socket(httpd.socket, certfile=_PEM, keyfile=_KEY, server_side=True) 56 httpd.serve_forever() 57
