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 18import argparse 19import BaseHTTPServer 20import os 21import ssl 22import 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')) 26print _PEM 27open(_PEM).close() 28 29argp = argparse.ArgumentParser(description='Server for httpcli_test') 30argp.add_argument('-p', '--port', default=10080, type=int) 31argp.add_argument('-s', '--ssl', default=False, action='store_true') 32args = argp.parse_args() 33 34print 'server running on port %d' % args.port 35 36class 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 53httpd = BaseHTTPServer.HTTPServer(('localhost', args.port), Handler) 54if args.ssl: 55 httpd.socket = ssl.wrap_socket(httpd.socket, certfile=_PEM, keyfile=_KEY, server_side=True) 56httpd.serve_forever() 57