jh2_http2_ssl_socket_get_request_with_event_loop.py

1import socket
2import ssl
3import jh2.connection
4import jh2.events
5
6SERVER_NAME = 'google.com'
7SERVER_PORT = 443
8
9# Set up the SSL context for HTTP/2
10ssl_conf = ssl.create_default_context()
11ssl_conf.set_alpn_protocols(['h2'])
12
13# Establish a network connection
14sock = socket.create_connection((SERVER_NAME, SERVER_PORT))
15sock = ssl_conf.wrap_socket(sock, server_hostname=SERVER_NAME)
16
17# Create the HTTP/2 connection object
18conn = jh2.connection.H2Connection()
19conn.initiate_connection()
20sock.sendall(conn.data_to_send())
21
22# Send a request
23headers = [
24    (':method', 'GET'),
25    (':path', '/'),
26    (':authority', SERVER_NAME),
27    (':scheme', 'https'),
28]
29conn.send_headers(1, headers, end_stream=True)
30sock.sendall(conn.data_to_send())
31
32# Basic event loop to read the response
33response_received = False
34while not response_received:
35    data = sock.recv(65535)
36    if not data:
37        break
38    
39    events = conn.receive_data(data)
40    for event in events:
41        if isinstance(event, jh2.events.ResponseReceived):
42            print(f"Response headers: {event.headers}")
43        elif isinstance(event, jh2.events.DataReceived):
44            print(f"Response body: {event.data}")
45        elif isinstance(event, jh2.events.StreamEnded):
46            response_received = True
47
48sock.close()