1  #!/usr/bin/env python3
  2  # Copyright (c) 2014-2021 The Bitcoin Core developers
  3  # Distributed under the MIT software license, see the accompanying
  4  # file COPYING or http://www.opensource.org/licenses/mit-license.php.
  5  """Test the RPC HTTP basics."""
  6  
  7  from test_framework.test_framework import BitcoinTestFramework
  8  from test_framework.util import assert_equal, str_to_b64str
  9  
 10  import http.client
 11  import urllib.parse
 12  
 13  class HTTPBasicsTest (BitcoinTestFramework):
 14      def set_test_params(self):
 15          self.num_nodes = 3
 16          self.supports_cli = False
 17  
 18      def setup_network(self):
 19          self.setup_nodes()
 20  
 21      def run_test(self):
 22  
 23          #################################################
 24          # lowlevel check for http persistent connection #
 25          #################################################
 26          url = urllib.parse.urlparse(self.nodes[0].url)
 27          authpair = f'{url.username}:{url.password}'
 28          headers = {"Authorization": f"Basic {str_to_b64str(authpair)}"}
 29  
 30          conn = http.client.HTTPConnection(url.hostname, url.port)
 31          conn.connect()
 32          conn.request('POST', '/', '{"method": "getbestblockhash"}', headers)
 33          out1 = conn.getresponse().read()
 34          assert b'"error":null' in out1
 35          assert conn.sock is not None  #according to http/1.1 connection must still be open!
 36  
 37          #send 2nd request without closing connection
 38          conn.request('POST', '/', '{"method": "getchaintips"}', headers)
 39          out1 = conn.getresponse().read()
 40          assert b'"error":null' in out1  #must also response with a correct json-rpc message
 41          assert conn.sock is not None  #according to http/1.1 connection must still be open!
 42          conn.close()
 43  
 44          #same should be if we add keep-alive because this should be the std. behaviour
 45          headers = {"Authorization": f"Basic {str_to_b64str(authpair)}", "Connection": "keep-alive"}
 46  
 47          conn = http.client.HTTPConnection(url.hostname, url.port)
 48          conn.connect()
 49          conn.request('POST', '/', '{"method": "getbestblockhash"}', headers)
 50          out1 = conn.getresponse().read()
 51          assert b'"error":null' in out1
 52          assert conn.sock is not None  #according to http/1.1 connection must still be open!
 53  
 54          #send 2nd request without closing connection
 55          conn.request('POST', '/', '{"method": "getchaintips"}', headers)
 56          out1 = conn.getresponse().read()
 57          assert b'"error":null' in out1  #must also response with a correct json-rpc message
 58          assert conn.sock is not None  #according to http/1.1 connection must still be open!
 59          conn.close()
 60  
 61          #now do the same with "Connection: close"
 62          headers = {"Authorization": f"Basic {str_to_b64str(authpair)}", "Connection":"close"}
 63  
 64          conn = http.client.HTTPConnection(url.hostname, url.port)
 65          conn.connect()
 66          conn.request('POST', '/', '{"method": "getbestblockhash"}', headers)
 67          out1 = conn.getresponse().read()
 68          assert b'"error":null' in out1
 69          assert conn.sock is None  #now the connection must be closed after the response
 70  
 71          #node1 (2nd node) is running with disabled keep-alive option
 72          urlNode1 = urllib.parse.urlparse(self.nodes[1].url)
 73          authpair = f'{urlNode1.username}:{urlNode1.password}'
 74          headers = {"Authorization": f"Basic {str_to_b64str(authpair)}"}
 75  
 76          conn = http.client.HTTPConnection(urlNode1.hostname, urlNode1.port)
 77          conn.connect()
 78          conn.request('POST', '/', '{"method": "getbestblockhash"}', headers)
 79          out1 = conn.getresponse().read()
 80          assert b'"error":null' in out1
 81  
 82          #node2 (third node) is running with standard keep-alive parameters which means keep-alive is on
 83          urlNode2 = urllib.parse.urlparse(self.nodes[2].url)
 84          authpair = f'{urlNode2.username}:{urlNode2.password}'
 85          headers = {"Authorization": f"Basic {str_to_b64str(authpair)}"}
 86  
 87          conn = http.client.HTTPConnection(urlNode2.hostname, urlNode2.port)
 88          conn.connect()
 89          conn.request('POST', '/', '{"method": "getbestblockhash"}', headers)
 90          out1 = conn.getresponse().read()
 91          assert b'"error":null' in out1
 92          assert conn.sock is not None  #connection must be closed because bitcoind should use keep-alive by default
 93  
 94          # Check excessive request size
 95          conn = http.client.HTTPConnection(urlNode2.hostname, urlNode2.port)
 96          conn.connect()
 97          conn.request('GET', f'/{"x"*1000}', '', headers)
 98          out1 = conn.getresponse()
 99          assert_equal(out1.status, http.client.NOT_FOUND)
100  
101          conn = http.client.HTTPConnection(urlNode2.hostname, urlNode2.port)
102          conn.connect()
103          conn.request('GET', f'/{"x"*10000}', '', headers)
104          out1 = conn.getresponse()
105          assert_equal(out1.status, http.client.BAD_REQUEST)
106  
107  
108  if __name__ == '__main__':
109      HTTPBasicsTest ().main ()