def handle(self):
"""Handle multiple requests if necessary."""
self.close_connection = True
self.stream = None # for compatibility
# if request version not set check preface first
preface = self.rfile.peek(PREFACE_SIZE)
if preface[0:PREFACE_SIZE] == PREFACE_CODE: # check code
preface = self.rfile.read(PREFACE_SIZE) # read preface
print(preface)
self.request_version = 'HTTP/2.0'
# keep alive connection
self.close_connection = False
# send server preface
setting_frame = SettingFrame(is_ack=True)
settings_bin = setting_frame.get_frame_bin()
self.wfile.write(settings_bin)
self.http2_connection = Connection(self)
else:
self.request_version = ''
if self.request_version == 'HTTP/2.0':
try:
while True:
if self.parse_http2_request():
# clear
self.handle_one_http2_request()
self.headers = []
# self.request_version = 'HTTP/2.0' always
self.requestline = ''
self.command = ''
self.response_stream = None
except Exception as e:
if isinstance(e, HTTP2Error):
print('connection closed - with error : ' + repr(e))
print('error')
print(traceback.format_exc())
print(e)
else:
print('error')
print(traceback.format_exc())
print(e)
raise e
self.close_connection = True
return
else:
self.handle_one_request()
while not self.close_connection:
self.handle_one_request()
class BaseHTTP2RequestHandler(BaseHTTPRequestHandler):
server_version = "BaseHTTP2/" + __version__
def request_worker(self):
pass
def parse_http2_request(self):
self.raw_requestdata = self.connection.read(9)
frame_header = Frame.parse_header(self.raw_requestdata)
frm_len, frm_type, frm_flag, frm_id = frame_header
while frm_len + 9 > len(self.raw_requestdata):
self.raw_requestdata += self.connection.read(frm_len - len(self.raw_requestdata) + 9) # read left data
stream = self.http2_connection.get_stream(frm_id)
stream.receive_frame(frame_header, self.raw_requestdata)
if stream.is_wait_for_res:
print('end stream id : ', stream.id, ' frm_id ', frm_id)
self.headers = stream._client_headers
# self.request_version = 'HTTP/2.0' always
self.requestline = stream.method + ' ' + stream.path + ' HTTP/2.0' # virtual request line
self.path = stream.path
self.command = stream.method
self.response_stream = stream
self.stream = stream
return True # handle one request
return False
def handle(self):
"""Handle multiple requests if necessary."""
self.close_connection = True
self.stream = None # for compatibility
# if request version not set check preface first
preface = self.rfile.peek(PREFACE_SIZE)
if preface[0:PREFACE_SIZE] == PREFACE_CODE: # check code
preface = self.rfile.read(PREFACE_SIZE) # read preface
print(preface)
self.request_version = 'HTTP/2.0'
# keep alive connection
self.close_connection = False
# send server preface
setting_frame = SettingFrame(is_ack=True)
settings_bin = setting_frame.get_frame_bin()
self.wfile.write(settings_bin)
self.http2_connection = Connection(self)
else:
self.request_version = ''
if self.request_version == 'HTTP/2.0':
try:
while True:
if self.parse_http2_request():
# clear
self.handle_one_http2_request()
self.headers = []
# self.request_version = 'HTTP/2.0' always
self.requestline = ''
self.command = ''
self.response_stream = None
except Exception as e:
if isinstance(e, HTTP2Error):
print('connection closed - with error : ' + repr(e))
print('error')
print(traceback.format_exc())
print(e)
else:
print('error')
print(traceback.format_exc())
print(e)
raise e
self.close_connection = True
return
else:
self.handle_one_request()
while not self.close_connection:
self.handle_one_request()
def handle_one_http2_request(self):
print('create_stream')
# self.response_stream = self.http2_connection.create_stream()
self._headers_buffer = []
# parse request
mname = 'do_' + self.command
if not hasattr(self, mname):
self.send_error(501, "Unsupported method (%r)" % self.command)
return
method = getattr(self, mname)
method()
def send_response_only(self, code, message=None):
if self.request_version <= 'HTTP/1.1':
BaseHTTPRequestHandler.send_response_only(self, code, message)
elif self.request_version == 'HTTP/2.0':
self.send_header(':status', str(code))
self.send_header(':scheme', 'https') # TODO : fix it later
def send_header(self, keyword, value):
if self.request_version <= 'HTTP/1.1':
BaseHTTPRequestHandler.send_header(self, keyword, value)
elif self.request_version == 'HTTP/2.0':
self._headers_buffer.append((keyword.lower(), str(value)))
def end_headers(self):
if self.request_version <= 'HTTP/1.1':
BaseHTTPRequestHandler.end_headers(self)
elif self.request_version == 'HTTP/2.0':
self.flush_headers()
def flush_headers(self):
if self.request_version <= 'HTTP/1.1':
BaseHTTPRequestHandler.flush_headers(self)
elif self.request_version == 'HTTP/2.0':
self.response_stream.send_header(self._headers_buffer)
self._headers_buffer = []
def push(self, stream=None, req_headers=[]):
if self.request_version == 'HTTP/2.0':
if stream is not None:
return self.stream.promise(promise_headers=req_headers)
else:
return self.http2_connection.promise(req_headers)
return None
def send_data(self, data):
if self.request_version <= 'HTTP/1.1':
self.wfile.write(data)
elif self.request_version == 'HTTP/2.0':
self.response_stream.send_data(data, end_stream=True)
def flush(self):
if self.request_version == 'HTTP/2.0':
print('end_data')
# self.response_stream.send_data(bytearray(0), end_stream=True) # TODO : fix it to RST_STREAM frame
self.wfile.flush()
