mirror of https://github.com/grpc/grpc.git
The C based gRPC (C++, Python, Ruby, Objective-C, PHP, C#)
https://grpc.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
77 lines
2.3 KiB
77 lines
2.3 KiB
#!/usr/bin/env python3 |
|
# Copyright 2015 gRPC authors. |
|
# |
|
# Licensed under the Apache License, Version 2.0 (the "License"); |
|
# you may not use this file except in compliance with the License. |
|
# You may obtain a copy of the License at |
|
# |
|
# http://www.apache.org/licenses/LICENSE-2.0 |
|
# |
|
# Unless required by applicable law or agreed to in writing, software |
|
# distributed under the License is distributed on an "AS IS" BASIS, |
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
# See the License for the specific language governing permissions and |
|
# limitations under the License. |
|
"""Server for httpcli_test""" |
|
|
|
import argparse |
|
from http.server import BaseHTTPRequestHandler |
|
from http.server import HTTPServer |
|
import os |
|
import ssl |
|
import sys |
|
|
|
_PEM = os.path.abspath( |
|
os.path.join( |
|
os.path.dirname(sys.argv[0]), |
|
"../../..", |
|
"src/core/tsi/test_creds/server1.pem", |
|
) |
|
) |
|
_KEY = os.path.abspath( |
|
os.path.join( |
|
os.path.dirname(sys.argv[0]), |
|
"../../..", |
|
"src/core/tsi/test_creds/server1.key", |
|
) |
|
) |
|
print(_PEM) |
|
open(_PEM).close() |
|
|
|
argp = argparse.ArgumentParser(description="Server for httpcli_test") |
|
argp.add_argument("-p", "--port", default=10080, type=int) |
|
argp.add_argument("-s", "--ssl", default=False, action="store_true") |
|
args = argp.parse_args() |
|
|
|
print("server running on port %d" % args.port) |
|
|
|
|
|
class Handler(BaseHTTPRequestHandler): |
|
def good(self): |
|
self.send_response(200) |
|
self.send_header("Content-Type", "text/html") |
|
self.end_headers() |
|
self.wfile.write( |
|
"<html><head><title>Hello world!</title></head>".encode("ascii") |
|
) |
|
self.wfile.write( |
|
"<body><p>This is a test</p></body></html>".encode("ascii") |
|
) |
|
|
|
def do_GET(self): |
|
if self.path == "/get": |
|
self.good() |
|
|
|
def do_POST(self): |
|
content_len = self.headers.get("content-length") |
|
content = self.rfile.read(int(content_len)).decode("ascii") |
|
if self.path == "/post" and content == "hello": |
|
self.good() |
|
|
|
|
|
httpd = HTTPServer(("localhost", args.port), Handler) |
|
if args.ssl: |
|
ctx = ssl.SSLContext() |
|
ctx.load_cert_chain(certfile=_PEM, keyfile=_KEY) |
|
httpd.socket = ctx.wrap_socket(httpd.socket, server_side=True) |
|
httpd.serve_forever()
|
|
|