forked from STI/Astroport.ONE
460 lines
6.8 KiB
Python
460 lines
6.8 KiB
Python
|
from typing import Union
|
||
|
|
||
|
"""
|
||
|
_url.py
|
||
|
websocket - WebSocket client library for Python
|
||
|
|
||
|
Copyright 2023 engn33r
|
||
|
|
||
|
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.
|
||
|
"""
|
||
|
__all__ = ["NoLock", "validate_utf8", "extract_err_message", "extract_error_code"]
|
||
|
|
||
|
|
||
|
class NoLock:
|
||
|
def __enter__(self) -> None:
|
||
|
pass
|
||
|
|
||
|
def __exit__(self, exc_type, exc_value, traceback) -> None:
|
||
|
pass
|
||
|
|
||
|
|
||
|
try:
|
||
|
# If wsaccel is available we use compiled routines to validate UTF-8
|
||
|
# strings.
|
||
|
from wsaccel.utf8validator import Utf8Validator
|
||
|
|
||
|
def _validate_utf8(utfbytes: Union[str, bytes]) -> bool:
|
||
|
result: bool = Utf8Validator().validate(utfbytes)[0]
|
||
|
return result
|
||
|
|
||
|
except ImportError:
|
||
|
# UTF-8 validator
|
||
|
# python implementation of http://bjoern.hoehrmann.de/utf-8/decoder/dfa/
|
||
|
|
||
|
_UTF8_ACCEPT = 0
|
||
|
_UTF8_REJECT = 12
|
||
|
|
||
|
_UTF8D = [
|
||
|
# The first part of the table maps bytes to character classes that
|
||
|
# to reduce the size of the transition table and create bitmasks.
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
0,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
1,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
9,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
7,
|
||
|
8,
|
||
|
8,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
2,
|
||
|
10,
|
||
|
3,
|
||
|
3,
|
||
|
3,
|
||
|
3,
|
||
|
3,
|
||
|
3,
|
||
|
3,
|
||
|
3,
|
||
|
3,
|
||
|
3,
|
||
|
3,
|
||
|
3,
|
||
|
4,
|
||
|
3,
|
||
|
3,
|
||
|
11,
|
||
|
6,
|
||
|
6,
|
||
|
6,
|
||
|
5,
|
||
|
8,
|
||
|
8,
|
||
|
8,
|
||
|
8,
|
||
|
8,
|
||
|
8,
|
||
|
8,
|
||
|
8,
|
||
|
8,
|
||
|
8,
|
||
|
8,
|
||
|
# The second part is a transition table that maps a combination
|
||
|
# of a state of the automaton and a character class to a state.
|
||
|
0,
|
||
|
12,
|
||
|
24,
|
||
|
36,
|
||
|
60,
|
||
|
96,
|
||
|
84,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
48,
|
||
|
72,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
0,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
0,
|
||
|
12,
|
||
|
0,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
24,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
24,
|
||
|
12,
|
||
|
24,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
24,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
24,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
24,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
36,
|
||
|
12,
|
||
|
36,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
36,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
36,
|
||
|
12,
|
||
|
36,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
36,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
12,
|
||
|
]
|
||
|
|
||
|
def _decode(state: int, codep: int, ch: int) -> tuple:
|
||
|
tp = _UTF8D[ch]
|
||
|
|
||
|
codep = (
|
||
|
(ch & 0x3F) | (codep << 6) if (state != _UTF8_ACCEPT) else (0xFF >> tp) & ch
|
||
|
)
|
||
|
state = _UTF8D[256 + state + tp]
|
||
|
|
||
|
return state, codep
|
||
|
|
||
|
def _validate_utf8(utfbytes: Union[str, bytes]) -> bool:
|
||
|
state = _UTF8_ACCEPT
|
||
|
codep = 0
|
||
|
for i in utfbytes:
|
||
|
state, codep = _decode(state, codep, int(i))
|
||
|
if state == _UTF8_REJECT:
|
||
|
return False
|
||
|
|
||
|
return True
|
||
|
|
||
|
|
||
|
def validate_utf8(utfbytes: Union[str, bytes]) -> bool:
|
||
|
"""
|
||
|
validate utf8 byte string.
|
||
|
utfbytes: utf byte string to check.
|
||
|
return value: if valid utf8 string, return true. Otherwise, return false.
|
||
|
"""
|
||
|
return _validate_utf8(utfbytes)
|
||
|
|
||
|
|
||
|
def extract_err_message(exception: Exception) -> Union[str, None]:
|
||
|
if exception.args:
|
||
|
exception_message: str = exception.args[0]
|
||
|
return exception_message
|
||
|
else:
|
||
|
return None
|
||
|
|
||
|
|
||
|
def extract_error_code(exception: Exception) -> Union[int, None]:
|
||
|
if exception.args and len(exception.args) > 1:
|
||
|
return exception.args[0] if isinstance(exception.args[0], int) else None
|