27e68365fc79baaf012b8881617d9b2b8c48c2df
[WebKit-https.git] / LayoutTests / http / tests / resources / portabilityLayer.py
1 # Copyright (C) 2021 Apple Inc. All rights reserved.
2 #
3 # Redistribution and use in source and binary forms, with or without
4 # modification, are permitted provided that the following conditions
5 # are met:
6 # 1.  Redistributions of source code must retain the above copyright
7 #     notice, this list of conditions and the following disclaimer.
8 # 2.  Redistributions in binary form must reproduce the above copyright
9 #     notice, this list of conditions and the following disclaimer in the
10 #     documentation and/or other materials provided with the distribution.
11 #
12 # THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS "AS IS" AND
13 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
14 # WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
15 # DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS BE LIABLE FOR
16 # ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
17 # DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
18 # SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
19 # CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
20 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
21 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
22
23 import cgi
24 import os
25 from urllib.parse import parse_qs
26
27
28 def get_cookies():
29     cookies = {}
30     if 'HTTP_COOKIE' in os.environ:
31         header_cookies = os.environ['HTTP_COOKIE']
32         header_cookies = header_cookies.split('; ')
33
34         for cookie in header_cookies:
35             cookie = cookie.split('=')
36             cookies[cookie[0]] = cookie[1]
37
38     return cookies
39
40
41 def get_request():
42     request = {}
43     request_method = os.environ.get('REQUEST_METHOD', '')
44     if request_method == 'POST':
45         form = cgi.FieldStorage()
46         for key in form.keys():
47             request.update({key: form.getvalue(key)})
48     else:
49         query = parse_qs(os.environ.get('QUERY_STRING', ''), keep_blank_values=True)
50         for key in query.keys():
51             request.update({key: query[key][0]})
52
53     request.update(get_cookies())
54
55     return request
56
57
58 def get_count(file):
59     if not os.path.isfile(file):
60         with open(file, 'w') as open_file:
61             open_file.write('0')
62             return '0'
63
64     with open(file, 'r') as open_file:
65         return open_file.read()
66
67
68 def get_state(file, default='Uninitialized'):
69     if not os.path.isfile(file):
70         return default
71     with open(file, 'r') as file:
72         return file.read()
73
74
75 def set_state(state, file):
76     with open(file, 'w') as open_file:
77         open_file.write(state)
78     return state
79
80
81 def step_state(file):
82     state = get_count(file)
83     with open(file, 'w') as open_file:
84         open_file.write(f'{int(state) + 1}')
85     return state