Show More
@@ -0,0 +1,54 b'' | |||
|
1 | [build-system] | |
|
2 | requires = ["setuptools>=61.0.0", "wheel"] | |
|
3 | build-backend = "setuptools.build_meta" | |
|
4 | ||
|
5 | [project] | |
|
6 | name = "rhodecode-vcsserver" | |
|
7 | description = "Version Control System Server for RhodeCode" | |
|
8 | authors = [ | |
|
9 | {name = "RhodeCode GmbH", email = "support@rhodecode.com"}, | |
|
10 | ] | |
|
11 | ||
|
12 | license = {text = "GPL V3"} | |
|
13 | requires-python = ">=3.10" | |
|
14 | dynamic = ["version", "readme", "dependencies", "optional-dependencies"] | |
|
15 | classifiers = [ | |
|
16 | 'Development Status :: 6 - Mature', | |
|
17 | 'Intended Audience :: Developers', | |
|
18 | 'Operating System :: OS Independent', | |
|
19 | 'Topic :: Software Development :: Version Control', | |
|
20 | 'License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)', | |
|
21 | 'Programming Language :: Python :: 3.10', | |
|
22 | ] | |
|
23 | ||
|
24 | [project.entry-points."paste.app_factory"] | |
|
25 | main = "vcsserver.http_main:main" | |
|
26 | ||
|
27 | ||
|
28 | [tool.setuptools] | |
|
29 | packages = ["vcsserver"] | |
|
30 | ||
|
31 | [tool.setuptools.dynamic] | |
|
32 | readme = {file = ["README.rst"], content-type = "text/rst"} | |
|
33 | version = {file = "vcsserver/VERSION"} | |
|
34 | dependencies = {file = ["requirements.txt"]} | |
|
35 | optional-dependencies.tests = {file = ["requirements_test.txt"]} | |
|
36 | ||
|
37 | [tool.ruff] | |
|
38 | select = [ | |
|
39 | # Pyflakes | |
|
40 | "F", | |
|
41 | # Pycodestyle | |
|
42 | "E", | |
|
43 | "W", | |
|
44 | # isort | |
|
45 | "I001" | |
|
46 | ] | |
|
47 | ignore = [ | |
|
48 | "E501", # line too long, handled by black | |
|
49 | ] | |
|
50 | # Same as Black. | |
|
51 | line-length = 120 | |
|
52 | ||
|
53 | [tool.ruff.isort] | |
|
54 | known-first-party = ["vcsserver"] |
@@ -0,0 +1,27 b'' | |||
|
1 | # Copyright (C) 2010-2023 RhodeCode GmbH | |
|
2 | # | |
|
3 | # This program is free software: you can redistribute it and/or modify | |
|
4 | # it under the terms of the GNU Affero General Public License, version 3 | |
|
5 | # (only), as published by the Free Software Foundation. | |
|
6 | # | |
|
7 | # This program is distributed in the hope that it will be useful, | |
|
8 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
|
9 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
|
10 | # GNU General Public License for more details. | |
|
11 | # | |
|
12 | # You should have received a copy of the GNU Affero General Public License | |
|
13 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
|
14 | # | |
|
15 | # This program is dual-licensed. If you wish to learn more about the | |
|
16 | # RhodeCode Enterprise Edition, including its added features, Support services, | |
|
17 | # and proprietary license terms, please see https://rhodecode.com/licenses/ | |
|
18 | ||
|
19 | HOOK_REPO_SIZE = 'changegroup.repo_size' | |
|
20 | ||
|
21 | # HG | |
|
22 | HOOK_PRE_PULL = 'preoutgoing.pre_pull' | |
|
23 | HOOK_PULL = 'outgoing.pull_logger' | |
|
24 | HOOK_PRE_PUSH = 'prechangegroup.pre_push' | |
|
25 | HOOK_PRETX_PUSH = 'pretxnchangegroup.pre_push' | |
|
26 | HOOK_PUSH = 'changegroup.push_logger' | |
|
27 | HOOK_PUSH_KEY = 'pushkey.key_push' |
@@ -0,0 +1,53 b'' | |||
|
1 | # Copyright (C) 2010-2023 RhodeCode GmbH | |
|
2 | # | |
|
3 | # This program is free software: you can redistribute it and/or modify | |
|
4 | # it under the terms of the GNU Affero General Public License, version 3 | |
|
5 | # (only), as published by the Free Software Foundation. | |
|
6 | # | |
|
7 | # This program is distributed in the hope that it will be useful, | |
|
8 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
|
9 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
|
10 | # GNU General Public License for more details. | |
|
11 | # | |
|
12 | # You should have received a copy of the GNU Affero General Public License | |
|
13 | # along with this program. If not, see <http://www.gnu.org/licenses/>. | |
|
14 | # | |
|
15 | # This program is dual-licensed. If you wish to learn more about the | |
|
16 | # RhodeCode Enterprise Edition, including its added features, Support services, | |
|
17 | # and proprietary license terms, please see https://rhodecode.com/licenses/ | |
|
18 | ||
|
19 | import sys | |
|
20 | import logging | |
|
21 | ||
|
22 | ||
|
23 | BLACK, RED, GREEN, YELLOW, BLUE, MAGENTA, CYAN, WHITE = list(range(30, 38)) | |
|
24 | ||
|
25 | # Sequences | |
|
26 | RESET_SEQ = "\033[0m" | |
|
27 | COLOR_SEQ = "\033[0;%dm" | |
|
28 | BOLD_SEQ = "\033[1m" | |
|
29 | ||
|
30 | COLORS = { | |
|
31 | 'CRITICAL': MAGENTA, | |
|
32 | 'ERROR': RED, | |
|
33 | 'WARNING': CYAN, | |
|
34 | 'INFO': GREEN, | |
|
35 | 'DEBUG': BLUE, | |
|
36 | 'SQL': YELLOW | |
|
37 | } | |
|
38 | ||
|
39 | ||
|
40 | class ColorFormatter(logging.Formatter): | |
|
41 | ||
|
42 | def format(self, record): | |
|
43 | """ | |
|
44 | Change record's levelname to use with COLORS enum | |
|
45 | """ | |
|
46 | def_record = super().format(record) | |
|
47 | ||
|
48 | levelname = record.levelname | |
|
49 | start = COLOR_SEQ % (COLORS[levelname]) | |
|
50 | end = RESET_SEQ | |
|
51 | ||
|
52 | colored_record = ''.join([start, def_record, end]) | |
|
53 | return colored_record |
@@ -0,0 +1,87 b'' | |||
|
1 | # RhodeCode VCSServer provides access to different vcs backends via network. | |
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | # | |
|
4 | # This program is free software; you can redistribute it and/or modify | |
|
5 | # it under the terms of the GNU General Public License as published by | |
|
6 | # the Free Software Foundation; either version 3 of the License, or | |
|
7 | # (at your option) any later version. | |
|
8 | # | |
|
9 | # This program is distributed in the hope that it will be useful, | |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
|
12 | # GNU General Public License for more details. | |
|
13 | # | |
|
14 | # You should have received a copy of the GNU General Public License | |
|
15 | # along with this program; if not, write to the Free Software Foundation, | |
|
16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
|
17 | ||
|
18 | import logging | |
|
19 | import os | |
|
20 | import diskcache | |
|
21 | from diskcache import RLock | |
|
22 | ||
|
23 | log = logging.getLogger(__name__) | |
|
24 | ||
|
25 | cache_meta = None | |
|
26 | ||
|
27 | ||
|
28 | class ReentrantLock(RLock): | |
|
29 | def __enter__(self): | |
|
30 | reentrant_lock_key = self._key | |
|
31 | ||
|
32 | log.debug('Acquire ReentrantLock(key=%s) for archive cache generation...', reentrant_lock_key) | |
|
33 | #self.acquire() | |
|
34 | log.debug('Lock for key=%s acquired', reentrant_lock_key) | |
|
35 | ||
|
36 | def __exit__(self, *exc_info): | |
|
37 | #self.release() | |
|
38 | pass | |
|
39 | ||
|
40 | ||
|
41 | def get_archival_config(config): | |
|
42 | ||
|
43 | final_config = { | |
|
44 | 'archive_cache.eviction_policy': 'least-frequently-used' | |
|
45 | } | |
|
46 | ||
|
47 | for k, v in config.items(): | |
|
48 | if k.startswith('archive_cache'): | |
|
49 | final_config[k] = v | |
|
50 | ||
|
51 | return final_config | |
|
52 | ||
|
53 | ||
|
54 | def get_archival_cache_store(config): | |
|
55 | ||
|
56 | global cache_meta | |
|
57 | if cache_meta is not None: | |
|
58 | return cache_meta | |
|
59 | ||
|
60 | config = get_archival_config(config) | |
|
61 | ||
|
62 | archive_cache_dir = config['archive_cache.store_dir'] | |
|
63 | archive_cache_size_gb = config['archive_cache.cache_size_gb'] | |
|
64 | archive_cache_shards = config['archive_cache.cache_shards'] | |
|
65 | archive_cache_eviction_policy = config['archive_cache.eviction_policy'] | |
|
66 | ||
|
67 | log.debug('Initializing archival cache instance under %s', archive_cache_dir) | |
|
68 | ||
|
69 | # check if it's ok to write, and re-create the archive cache | |
|
70 | if not os.path.isdir(archive_cache_dir): | |
|
71 | os.makedirs(archive_cache_dir, exist_ok=True) | |
|
72 | ||
|
73 | d_cache = diskcache.FanoutCache( | |
|
74 | archive_cache_dir, shards=archive_cache_shards, | |
|
75 | cull_limit=0, # manual eviction required | |
|
76 | size_limit=archive_cache_size_gb * 1024 * 1024 * 1024, | |
|
77 | eviction_policy=archive_cache_eviction_policy, | |
|
78 | timeout=30 | |
|
79 | ) | |
|
80 | cache_meta = d_cache | |
|
81 | return cache_meta | |
|
82 | ||
|
83 | ||
|
84 | def includeme(config): | |
|
85 | # init our cache at start, for vcsserver we don't init at runtime | |
|
86 | # because our cache config is sent via wire on make archive call, this call just lazy-enables the client | |
|
87 | return |
@@ -0,0 +1,160 b'' | |||
|
1 | # RhodeCode VCSServer provides access to different vcs backends via network. | |
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | # | |
|
4 | # This program is free software; you can redistribute it and/or modify | |
|
5 | # it under the terms of the GNU General Public License as published by | |
|
6 | # the Free Software Foundation; either version 3 of the License, or | |
|
7 | # (at your option) any later version. | |
|
8 | # | |
|
9 | # This program is distributed in the hope that it will be useful, | |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
|
12 | # GNU General Public License for more details. | |
|
13 | # | |
|
14 | # You should have received a copy of the GNU General Public License | |
|
15 | # along with this program; if not, write to the Free Software Foundation, | |
|
16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
|
17 | ||
|
18 | import os | |
|
19 | import tempfile | |
|
20 | ||
|
21 | from svn import client | |
|
22 | from svn import core | |
|
23 | from svn import ra | |
|
24 | ||
|
25 | from mercurial import error | |
|
26 | ||
|
27 | from vcsserver.str_utils import safe_bytes | |
|
28 | ||
|
29 | core.svn_config_ensure(None) | |
|
30 | svn_config = core.svn_config_get_config(None) | |
|
31 | ||
|
32 | ||
|
33 | class RaCallbacks(ra.Callbacks): | |
|
34 | @staticmethod | |
|
35 | def open_tmp_file(pool): # pragma: no cover | |
|
36 | (fd, fn) = tempfile.mkstemp() | |
|
37 | os.close(fd) | |
|
38 | return fn | |
|
39 | ||
|
40 | @staticmethod | |
|
41 | def get_client_string(pool): | |
|
42 | return b'RhodeCode-subversion-url-checker' | |
|
43 | ||
|
44 | ||
|
45 | class SubversionException(Exception): | |
|
46 | pass | |
|
47 | ||
|
48 | ||
|
49 | class SubversionConnectionException(SubversionException): | |
|
50 | """Exception raised when a generic error occurs when connecting to a repository.""" | |
|
51 | ||
|
52 | ||
|
53 | def normalize_url(url): | |
|
54 | if not url: | |
|
55 | return url | |
|
56 | if url.startswith(b'svn+http://') or url.startswith(b'svn+https://'): | |
|
57 | url = url[4:] | |
|
58 | url = url.rstrip(b'/') | |
|
59 | return url | |
|
60 | ||
|
61 | ||
|
62 | def _create_auth_baton(pool): | |
|
63 | """Create a Subversion authentication baton. """ | |
|
64 | # Give the client context baton a suite of authentication | |
|
65 | # providers.h | |
|
66 | platform_specific = [ | |
|
67 | 'svn_auth_get_gnome_keyring_simple_provider', | |
|
68 | 'svn_auth_get_gnome_keyring_ssl_client_cert_pw_provider', | |
|
69 | 'svn_auth_get_keychain_simple_provider', | |
|
70 | 'svn_auth_get_keychain_ssl_client_cert_pw_provider', | |
|
71 | 'svn_auth_get_kwallet_simple_provider', | |
|
72 | 'svn_auth_get_kwallet_ssl_client_cert_pw_provider', | |
|
73 | 'svn_auth_get_ssl_client_cert_file_provider', | |
|
74 | 'svn_auth_get_windows_simple_provider', | |
|
75 | 'svn_auth_get_windows_ssl_server_trust_provider', | |
|
76 | ] | |
|
77 | ||
|
78 | providers = [] | |
|
79 | ||
|
80 | for p in platform_specific: | |
|
81 | if getattr(core, p, None) is not None: | |
|
82 | try: | |
|
83 | providers.append(getattr(core, p)()) | |
|
84 | except RuntimeError: | |
|
85 | pass | |
|
86 | ||
|
87 | providers += [ | |
|
88 | client.get_simple_provider(), | |
|
89 | client.get_username_provider(), | |
|
90 | client.get_ssl_client_cert_file_provider(), | |
|
91 | client.get_ssl_client_cert_pw_file_provider(), | |
|
92 | client.get_ssl_server_trust_file_provider(), | |
|
93 | ] | |
|
94 | ||
|
95 | return core.svn_auth_open(providers, pool) | |
|
96 | ||
|
97 | ||
|
98 | class SubversionRepo(object): | |
|
99 | """Wrapper for a Subversion repository. | |
|
100 | ||
|
101 | It uses the SWIG Python bindings, see above for requirements. | |
|
102 | """ | |
|
103 | def __init__(self, svn_url: bytes = b'', username: bytes = b'', password: bytes = b''): | |
|
104 | ||
|
105 | self.username = username | |
|
106 | self.password = password | |
|
107 | self.svn_url = core.svn_path_canonicalize(svn_url) | |
|
108 | ||
|
109 | self.auth_baton_pool = core.Pool() | |
|
110 | self.auth_baton = _create_auth_baton(self.auth_baton_pool) | |
|
111 | # self.init_ra_and_client() assumes that a pool already exists | |
|
112 | self.pool = core.Pool() | |
|
113 | ||
|
114 | self.ra = self.init_ra_and_client() | |
|
115 | self.uuid = ra.get_uuid(self.ra, self.pool) | |
|
116 | ||
|
117 | def init_ra_and_client(self): | |
|
118 | """Initializes the RA and client layers, because sometimes getting | |
|
119 | unified diffs runs the remote server out of open files. | |
|
120 | """ | |
|
121 | ||
|
122 | if self.username: | |
|
123 | core.svn_auth_set_parameter(self.auth_baton, | |
|
124 | core.SVN_AUTH_PARAM_DEFAULT_USERNAME, | |
|
125 | self.username) | |
|
126 | if self.password: | |
|
127 | core.svn_auth_set_parameter(self.auth_baton, | |
|
128 | core.SVN_AUTH_PARAM_DEFAULT_PASSWORD, | |
|
129 | self.password) | |
|
130 | ||
|
131 | callbacks = RaCallbacks() | |
|
132 | callbacks.auth_baton = self.auth_baton | |
|
133 | ||
|
134 | try: | |
|
135 | return ra.open2(self.svn_url, callbacks, svn_config, self.pool) | |
|
136 | except SubversionException as e: | |
|
137 | # e.child contains a detailed error messages | |
|
138 | msglist = [] | |
|
139 | svn_exc = e | |
|
140 | while svn_exc: | |
|
141 | if svn_exc.args[0]: | |
|
142 | msglist.append(svn_exc.args[0]) | |
|
143 | svn_exc = svn_exc.child | |
|
144 | msg = '\n'.join(msglist) | |
|
145 | raise SubversionConnectionException(msg) | |
|
146 | ||
|
147 | ||
|
148 | class svnremoterepo(object): | |
|
149 | """ the dumb wrapper for actual Subversion repositories """ | |
|
150 | ||
|
151 | def __init__(self, username: bytes = b'', password: bytes = b'', svn_url: bytes = b''): | |
|
152 | self.username = username or b'' | |
|
153 | self.password = password or b'' | |
|
154 | self.path = normalize_url(svn_url) | |
|
155 | ||
|
156 | def svn(self): | |
|
157 | try: | |
|
158 | return SubversionRepo(self.path, self.username, self.password) | |
|
159 | except SubversionConnectionException as e: | |
|
160 | raise error.Abort(safe_bytes(e)) |
@@ -0,0 +1,17 b'' | |||
|
1 | # RhodeCode VCSServer provides access to different vcs backends via network. | |
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | # | |
|
4 | # This program is free software; you can redistribute it and/or modify | |
|
5 | # it under the terms of the GNU General Public License as published by | |
|
6 | # the Free Software Foundation; either version 3 of the License, or | |
|
7 | # (at your option) any later version. | |
|
8 | # | |
|
9 | # This program is distributed in the hope that it will be useful, | |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
|
12 | # GNU General Public License for more details. | |
|
13 | # | |
|
14 | # You should have received a copy of the GNU General Public License | |
|
15 | # along with this program; if not, write to the Free Software Foundation, | |
|
16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
|
17 |
@@ -0,0 +1,133 b'' | |||
|
1 | # RhodeCode VCSServer provides access to different vcs backends via network. | |
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | # | |
|
4 | # This program is free software; you can redistribute it and/or modify | |
|
5 | # it under the terms of the GNU General Public License as published by | |
|
6 | # the Free Software Foundation; either version 3 of the License, or | |
|
7 | # (at your option) any later version. | |
|
8 | # | |
|
9 | # This program is distributed in the hope that it will be useful, | |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
|
12 | # GNU General Public License for more details. | |
|
13 | # | |
|
14 | # You should have received a copy of the GNU General Public License | |
|
15 | # along with this program; if not, write to the Free Software Foundation, | |
|
16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
|
17 | ||
|
18 | import typing | |
|
19 | import base64 | |
|
20 | import logging | |
|
21 | ||
|
22 | ||
|
23 | log = logging.getLogger(__name__) | |
|
24 | ||
|
25 | ||
|
26 | def safe_int(val, default=None) -> int: | |
|
27 | """ | |
|
28 | Returns int() of val if val is not convertable to int use default | |
|
29 | instead | |
|
30 | ||
|
31 | :param val: | |
|
32 | :param default: | |
|
33 | """ | |
|
34 | ||
|
35 | try: | |
|
36 | val = int(val) | |
|
37 | except (ValueError, TypeError): | |
|
38 | val = default | |
|
39 | ||
|
40 | return val | |
|
41 | ||
|
42 | ||
|
43 | def base64_to_str(text) -> str: | |
|
44 | return safe_str(base64.encodebytes(safe_bytes(text))).strip() | |
|
45 | ||
|
46 | ||
|
47 | def get_default_encodings() -> list[str]: | |
|
48 | return ['utf8'] | |
|
49 | ||
|
50 | ||
|
51 | def safe_str(str_, to_encoding=None) -> str: | |
|
52 | """ | |
|
53 | safe str function. Does few trick to turn unicode_ into string | |
|
54 | ||
|
55 | :param str_: str to encode | |
|
56 | :param to_encoding: encode to this type UTF8 default | |
|
57 | """ | |
|
58 | if isinstance(str_, str): | |
|
59 | return str_ | |
|
60 | ||
|
61 | # if it's bytes cast to str | |
|
62 | if not isinstance(str_, bytes): | |
|
63 | return str(str_) | |
|
64 | ||
|
65 | to_encoding = to_encoding or get_default_encodings() | |
|
66 | if not isinstance(to_encoding, (list, tuple)): | |
|
67 | to_encoding = [to_encoding] | |
|
68 | ||
|
69 | for enc in to_encoding: | |
|
70 | try: | |
|
71 | return str(str_, enc) | |
|
72 | except UnicodeDecodeError: | |
|
73 | pass | |
|
74 | ||
|
75 | return str(str_, to_encoding[0], 'replace') | |
|
76 | ||
|
77 | ||
|
78 | def safe_bytes(str_, from_encoding=None) -> bytes: | |
|
79 | """ | |
|
80 | safe bytes function. Does few trick to turn str_ into bytes string: | |
|
81 | ||
|
82 | :param str_: string to decode | |
|
83 | :param from_encoding: encode from this type UTF8 default | |
|
84 | """ | |
|
85 | if isinstance(str_, bytes): | |
|
86 | return str_ | |
|
87 | ||
|
88 | if not isinstance(str_, str): | |
|
89 | raise ValueError(f'safe_bytes cannot convert other types than str: got: {type(str_)}') | |
|
90 | ||
|
91 | from_encoding = from_encoding or get_default_encodings() | |
|
92 | if not isinstance(from_encoding, (list, tuple)): | |
|
93 | from_encoding = [from_encoding] | |
|
94 | ||
|
95 | for enc in from_encoding: | |
|
96 | try: | |
|
97 | return str_.encode(enc) | |
|
98 | except UnicodeDecodeError: | |
|
99 | pass | |
|
100 | ||
|
101 | return str_.encode(from_encoding[0], 'replace') | |
|
102 | ||
|
103 | ||
|
104 | def ascii_bytes(str_, allow_bytes=False) -> bytes: | |
|
105 | """ | |
|
106 | Simple conversion from str to bytes, with assumption that str_ is pure ASCII. | |
|
107 | Fails with UnicodeError on invalid input. | |
|
108 | This should be used where encoding and "safe" ambiguity should be avoided. | |
|
109 | Where strings already have been encoded in other ways but still are unicode | |
|
110 | string - for example to hex, base64, json, urlencoding, or are known to be | |
|
111 | identifiers. | |
|
112 | """ | |
|
113 | if allow_bytes and isinstance(str_, bytes): | |
|
114 | return str_ | |
|
115 | ||
|
116 | if not isinstance(str_, str): | |
|
117 | raise ValueError(f'ascii_bytes cannot convert other types than str: got: {type(str_)}') | |
|
118 | return str_.encode('ascii') | |
|
119 | ||
|
120 | ||
|
121 | def ascii_str(str_) -> str: | |
|
122 | """ | |
|
123 | Simple conversion from bytes to str, with assumption that str_ is pure ASCII. | |
|
124 | Fails with UnicodeError on invalid input. | |
|
125 | This should be used where encoding and "safe" ambiguity should be avoided. | |
|
126 | Where strings are encoded but also in other ways are known to be ASCII, and | |
|
127 | where a unicode string is wanted without caring about encoding. For example | |
|
128 | to hex, base64, urlencoding, or are known to be identifiers. | |
|
129 | """ | |
|
130 | ||
|
131 | if not isinstance(str_, bytes): | |
|
132 | raise ValueError(f'ascii_str cannot convert other types than bytes: got: {type(str_)}') | |
|
133 | return str_.decode('ascii') |
@@ -0,0 +1,53 b'' | |||
|
1 | # RhodeCode VCSServer provides access to different vcs backends via network. | |
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | # | |
|
4 | # This program is free software; you can redistribute it and/or modify | |
|
5 | # it under the terms of the GNU General Public License as published by | |
|
6 | # the Free Software Foundation; either version 3 of the License, or | |
|
7 | # (at your option) any later version. | |
|
8 | # | |
|
9 | # This program is distributed in the hope that it will be useful, | |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
|
12 | # GNU General Public License for more details. | |
|
13 | # | |
|
14 | # You should have received a copy of the GNU General Public License | |
|
15 | # along with this program; if not, write to the Free Software Foundation, | |
|
16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
|
17 | ||
|
18 | import pytest | |
|
19 | from vcsserver.str_utils import ascii_bytes, ascii_str | |
|
20 | ||
|
21 | ||
|
22 | @pytest.mark.parametrize('given, expected', [ | |
|
23 | ('a', b'a'), | |
|
24 | ('a', b'a'), | |
|
25 | ]) | |
|
26 | def test_ascii_bytes(given, expected): | |
|
27 | assert ascii_bytes(given) == expected | |
|
28 | ||
|
29 | ||
|
30 | @pytest.mark.parametrize('given', [ | |
|
31 | 'å', | |
|
32 | 'å'.encode('utf8') | |
|
33 | ]) | |
|
34 | def test_ascii_bytes_raises(given): | |
|
35 | with pytest.raises(ValueError): | |
|
36 | ascii_bytes(given) | |
|
37 | ||
|
38 | ||
|
39 | @pytest.mark.parametrize('given, expected', [ | |
|
40 | (b'a', 'a'), | |
|
41 | ]) | |
|
42 | def test_ascii_str(given, expected): | |
|
43 | assert ascii_str(given) == expected | |
|
44 | ||
|
45 | ||
|
46 | @pytest.mark.parametrize('given', [ | |
|
47 | 'a', | |
|
48 | 'å'.encode('utf8'), | |
|
49 | 'å' | |
|
50 | ]) | |
|
51 | def test_ascii_str_raises(given): | |
|
52 | with pytest.raises(ValueError): | |
|
53 | ascii_str(given) |
@@ -0,0 +1,67 b'' | |||
|
1 | # RhodeCode VCSServer provides access to different vcs backends via network. | |
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | # | |
|
4 | # This program is free software; you can redistribute it and/or modify | |
|
5 | # it under the terms of the GNU General Public License as published by | |
|
6 | # the Free Software Foundation; either version 3 of the License, or | |
|
7 | # (at your option) any later version. | |
|
8 | # | |
|
9 | # This program is distributed in the hope that it will be useful, | |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
|
12 | # GNU General Public License for more details. | |
|
13 | # | |
|
14 | # You should have received a copy of the GNU General Public License | |
|
15 | # along with this program; if not, write to the Free Software Foundation, | |
|
16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
|
17 | ||
|
18 | ||
|
19 | import logging | |
|
20 | ||
|
21 | log = logging.getLogger(__name__) | |
|
22 | ||
|
23 | ||
|
24 | def str2bool(str_): | |
|
25 | """ | |
|
26 | returns True/False value from given string, it tries to translate the | |
|
27 | string into boolean | |
|
28 | ||
|
29 | :param str_: string value to translate into boolean | |
|
30 | :rtype: boolean | |
|
31 | :returns: boolean from given string | |
|
32 | """ | |
|
33 | if str_ is None: | |
|
34 | return False | |
|
35 | if str_ in (True, False): | |
|
36 | return str_ | |
|
37 | str_ = str(str_).strip().lower() | |
|
38 | return str_ in ('t', 'true', 'y', 'yes', 'on', '1') | |
|
39 | ||
|
40 | ||
|
41 | def aslist(obj, sep=None, strip=True) -> list: | |
|
42 | """ | |
|
43 | Returns given string separated by sep as list | |
|
44 | ||
|
45 | :param obj: | |
|
46 | :param sep: | |
|
47 | :param strip: | |
|
48 | """ | |
|
49 | if isinstance(obj, str): | |
|
50 | if obj in ['', ""]: | |
|
51 | return [] | |
|
52 | ||
|
53 | lst = obj.split(sep) | |
|
54 | if strip: | |
|
55 | lst = [v.strip() for v in lst] | |
|
56 | return lst | |
|
57 | elif isinstance(obj, (list, tuple)): | |
|
58 | return obj | |
|
59 | elif obj is None: | |
|
60 | return [] | |
|
61 | else: | |
|
62 | return [obj] | |
|
63 | ||
|
64 | ||
|
65 | def assert_bytes(input_type, expected_types=(bytes,)): | |
|
66 | if not isinstance(input_type, expected_types): | |
|
67 | raise ValueError(f'input_types should be one of {expected_types} got {type(input_type)} instead') |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | [bumpversion] |
|
2 |
current_version = |
|
|
2 | current_version = 5.0.0 | |
|
3 | 3 | message = release: Bump version {current_version} to {new_version} |
|
4 | 4 | |
|
5 | 5 | [bumpversion:file:vcsserver/VERSION] |
@@ -1,4 +1,5 b'' | |||
|
1 | 1 | syntax: glob |
|
2 | ||
|
2 | 3 | *.orig |
|
3 | 4 | *.pyc |
|
4 | 5 | *.swp |
@@ -19,8 +20,11 b' syntax: regexp' | |||
|
19 | 20 | ^\.pydevproject$ |
|
20 | 21 | ^\.coverage$ |
|
21 | 22 | ^\.cache.*$ |
|
23 | ^\.venv.*$ | |
|
24 | ^\.ruff_cache.*$ | |
|
22 | 25 | ^\.rhodecode$ |
|
23 | 26 | |
|
27 | ||
|
24 | 28 | ^.dev |
|
25 | 29 | ^build/ |
|
26 | 30 | ^coverage\.xml$ |
@@ -1,45 +1,111 b'' | |||
|
1 | .DEFAULT_GOAL := help | |
|
1 | # required for pushd to work.. | |
|
2 | #SHELL = /bin/bash | |
|
3 | ||
|
2 | 4 |
|
|
3 | 5 | # set by: PATH_TO_OUTDATED_PACKAGES=/some/path/outdated_packages.py |
|
4 | 6 | OUTDATED_PACKAGES = ${PATH_TO_OUTDATED_PACKAGES} |
|
5 | 7 | |
|
6 | 8 | .PHONY: clean |
|
7 | clean: ## full clean | |
|
9 | ## Cleanup compiled and cache py files | |
|
10 | clean: | |
|
8 | 11 | make test-clean |
|
9 | 12 | find . -type f \( -iname '*.c' -o -iname '*.pyc' -o -iname '*.so' -o -iname '*.orig' \) -exec rm '{}' ';' |
|
10 | 13 | |
|
11 | 14 | |
|
12 | 15 | .PHONY: test |
|
13 |
|
|
|
16 | ## run test-clean and tests | |
|
17 | test: | |
|
14 | 18 | make test-clean |
|
15 | 19 | make test-only |
|
16 | 20 | |
|
17 | 21 | |
|
18 | .PHONY:test-clean | |
|
19 |
|
|
|
22 | .PHONY: test-clean | |
|
23 | ## run test-clean and tests | |
|
24 | test-clean: | |
|
20 | 25 | rm -rf coverage.xml htmlcov junit.xml pylint.log result |
|
21 | 26 | find . -type d -name "__pycache__" -prune -exec rm -rf '{}' ';' |
|
22 | 27 | find . -type f \( -iname '.coverage.*' \) -exec rm '{}' ';' |
|
23 | 28 | |
|
24 | 29 | |
|
25 | 30 | .PHONY: test-only |
|
26 | test-only: ## run tests | |
|
31 | ## Run tests only without cleanup | |
|
32 | test-only: | |
|
27 | 33 | PYTHONHASHSEED=random \ |
|
28 | 34 | py.test -x -vv -r xw -p no:sugar \ |
|
29 |
|
|
|
30 | vcsserver | |
|
31 | ||
|
32 | ||
|
33 | .PHONY: generate-pkgs | |
|
34 | generate-pkgs: ## generate new python packages | |
|
35 | nix-shell pkgs/shell-generate.nix --command "pip2nix generate --licenses" | |
|
35 | --cov-report=term-missing --cov-report=html \ | |
|
36 | --cov=vcsserver vcsserver | |
|
36 | 37 | |
|
37 | 38 | |
|
38 | 39 | .PHONY: pip-packages |
|
39 |
|
|
|
40 | ## Show outdated packages | |
|
41 | pip-packages: | |
|
40 | 42 | python ${OUTDATED_PACKAGES} |
|
41 | 43 | |
|
42 | 44 | |
|
43 |
.PHONY: |
|
|
44 | help: | |
|
45 | @grep -E '^[a-zA-Z_-]+:.*?## .*$$' $(MAKEFILE_LIST) | sort | awk 'BEGIN {FS = ":.*?## "}; {printf "\033[36m%-24s\033[0m %s\n", $$1, $$2}' | |
|
45 | .PHONY: build | |
|
46 | ## Build sdist/egg | |
|
47 | build: | |
|
48 | python -m build | |
|
49 | ||
|
50 | ||
|
51 | .PHONY: dev-env | |
|
52 | ## make dev-env based on the requirements files and install develop of packages | |
|
53 | dev-env: | |
|
54 | pip install build virtualenv | |
|
55 | pip wheel --wheel-dir=/home/rhodecode/.cache/pip/wheels -r requirements.txt -r requirements_test.txt -r requirements_debug.txt | |
|
56 | pip install --no-index --find-links=/home/rhodecode/.cache/pip/wheels -r requirements.txt -r requirements_test.txt -r requirements_debug.txt | |
|
57 | pip install -e . | |
|
58 | ||
|
59 | ||
|
60 | .PHONY: dev-srv | |
|
61 | ## run develop server instance | |
|
62 | dev-srv: | |
|
63 | pserve --reload .dev/dev.ini | |
|
64 | ||
|
65 | ||
|
66 | .PHONY: dev-srv-g | |
|
67 | ## run gunicorn multi process workers | |
|
68 | dev-srv-g: | |
|
69 | gunicorn --workers=4 --paste .dev/dev.ini --bind=0.0.0.0:10010 --worker-class=sync --threads=1 --config=configs/gunicorn_config.py --timeout=120 | |
|
70 | ||
|
71 | # Default command on calling make | |
|
72 | .DEFAULT_GOAL := show-help | |
|
73 | ||
|
74 | .PHONY: show-help | |
|
75 | show-help: | |
|
76 | @echo "$$(tput bold)Available rules:$$(tput sgr0)" | |
|
77 | @echo | |
|
78 | @sed -n -e "/^## / { \ | |
|
79 | h; \ | |
|
80 | s/.*//; \ | |
|
81 | :doc" \ | |
|
82 | -e "H; \ | |
|
83 | n; \ | |
|
84 | s/^## //; \ | |
|
85 | t doc" \ | |
|
86 | -e "s/:.*//; \ | |
|
87 | G; \ | |
|
88 | s/\\n## /---/; \ | |
|
89 | s/\\n/ /g; \ | |
|
90 | p; \ | |
|
91 | }" ${MAKEFILE_LIST} \ | |
|
92 | | LC_ALL='C' sort --ignore-case \ | |
|
93 | | awk -F '---' \ | |
|
94 | -v ncol=$$(tput cols) \ | |
|
95 | -v indent=19 \ | |
|
96 | -v col_on="$$(tput setaf 6)" \ | |
|
97 | -v col_off="$$(tput sgr0)" \ | |
|
98 | '{ \ | |
|
99 | printf "%s%*s%s ", col_on, -indent, $$1, col_off; \ | |
|
100 | n = split($$2, words, " "); \ | |
|
101 | line_length = ncol - indent; \ | |
|
102 | for (i = 1; i <= n; i++) { \ | |
|
103 | line_length -= length(words[i]) + 1; \ | |
|
104 | if (line_length <= 0) { \ | |
|
105 | line_length = ncol - indent - length(words[i]) - 1; \ | |
|
106 | printf "\n%*s ", -indent, " "; \ | |
|
107 | } \ | |
|
108 | printf "%s ", words[i]; \ | |
|
109 | } \ | |
|
110 | printf "\n"; \ | |
|
111 | }' |
@@ -1,4 +1,4 b'' | |||
|
1 | ## -*- coding: utf-8 -*- | |
|
1 | # | |
|
2 | 2 | |
|
3 | 3 | ; ################################# |
|
4 | 4 | ; RHODECODE VCSSERVER CONFIGURATION |
@@ -57,12 +57,9 b' def _get_process_rss(pid=None):' | |||
|
57 | 57 | |
|
58 | 58 | |
|
59 | 59 | def _get_config(ini_path): |
|
60 | import configparser | |
|
60 | 61 | |
|
61 | 62 | try: |
|
62 | import configparser | |
|
63 | except ImportError: | |
|
64 | import ConfigParser as configparser | |
|
65 | try: | |
|
66 | 63 | config = configparser.RawConfigParser() |
|
67 | 64 | config.read(ini_path) |
|
68 | 65 | return config |
@@ -336,7 +333,7 b' class RhodeCodeLogger(Logger):' | |||
|
336 | 333 | def now(self): |
|
337 | 334 | """ return date in RhodeCode Log format """ |
|
338 | 335 | now = time.time() |
|
339 |
msecs = int((now - |
|
|
336 | msecs = int((now - int(now)) * 1000) | |
|
340 | 337 | return time.strftime(self.datefmt, time.localtime(now)) + '.{0:03d}'.format(msecs) |
|
341 | 338 | |
|
342 | 339 | def atoms(self, resp, req, environ, request_time): |
@@ -1,4 +1,4 b'' | |||
|
1 | ## -*- coding: utf-8 -*- | |
|
1 | # | |
|
2 | 2 | |
|
3 | 3 | ; ################################# |
|
4 | 4 | ; RHODECODE VCSSERVER CONFIGURATION |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -39,7 +39,7 b' def repeat(request):' | |||
|
39 | 39 | @pytest.fixture(scope='session') |
|
40 | 40 | def vcsserver_port(request): |
|
41 | 41 | port = get_available_port() |
|
42 |
print('Using vcsserver port |
|
|
42 | print(f'Using vcsserver port {port}') | |
|
43 | 43 | return port |
|
44 | 44 | |
|
45 | 45 |
@@ -1,47 +1,56 b'' | |||
|
1 | ## dependencies | |
|
2 | ||
|
3 | # our custom configobj | |
|
4 | https://code.rhodecode.com/upstream/configobj/artifacts/download/0-012de99a-b1e1-4f64-a5c0-07a98a41b324.tar.gz?md5=6a513f51fe04b2c18cf84c1395a7c626#egg=configobj==5.0.6 | |
|
1 | # deps, generated via pipdeptree --exclude setuptools,wheel,pipdeptree,pip -f | tr '[:upper:]' '[:lower:]' | |
|
5 | 2 | |
|
6 | dogpile.cache==0.9.0 | |
|
7 | decorator==4.1.2 | |
|
8 | dulwich==0.13.0 | |
|
9 | hgsubversion==1.9.3 | |
|
10 | hg-evolve==9.1.0 | |
|
11 | mako==1.1.0 | |
|
12 | markupsafe==1.1.1 | |
|
13 | mercurial==5.1.1 | |
|
14 | msgpack-python==0.5.6 | |
|
15 | ||
|
16 | pastedeploy==2.1.0 | |
|
17 | pyramid==1.10.4 | |
|
18 | pyramid-mako==1.1.0 | |
|
19 | pygit2==0.28.2 | |
|
20 | ||
|
3 | async-timeout==4.0.2 | |
|
4 | atomicwrites==1.4.1 | |
|
5 | contextlib2==21.6.0 | |
|
6 | cov-core==1.15.0 | |
|
7 | coverage==7.2.3 | |
|
8 | diskcache==5.6.1 | |
|
9 | dogpile.cache==1.2.2 | |
|
10 | decorator==5.1.1 | |
|
11 | stevedore==5.0.0 | |
|
12 | pbr==5.11.1 | |
|
13 | dulwich==0.21.5 | |
|
14 | urllib3==1.26.14 | |
|
15 | gunicorn==21.0.1 | |
|
16 | packaging==23.1 | |
|
17 | hg-evolve==11.0.2 | |
|
18 | importlib-metadata==6.0.0 | |
|
19 | zipp==3.15.0 | |
|
20 | mercurial==6.3.3 | |
|
21 | mock==5.0.2 | |
|
22 | more-itertools==9.1.0 | |
|
23 | msgpack==1.0.5 | |
|
24 | orjson==3.9.2 | |
|
25 | psutil==5.9.5 | |
|
26 | py==1.11.0 | |
|
27 | pygit2==1.12.2 | |
|
28 | cffi==1.15.1 | |
|
29 | pycparser==2.21 | |
|
30 | pygments==2.15.1 | |
|
31 | pyparsing==3.0.9 | |
|
32 | pyramid==2.0.1 | |
|
33 | hupper==1.12 | |
|
34 | plaster==1.1.2 | |
|
35 | plaster-pastedeploy==1.0.1 | |
|
36 | pastedeploy==3.0.1 | |
|
37 | plaster==1.1.2 | |
|
38 | translationstring==1.4 | |
|
39 | venusian==3.0.0 | |
|
40 | webob==1.8.7 | |
|
41 | zope.deprecation==5.0.0 | |
|
42 | zope.interface==6.0.0 | |
|
43 | redis==4.6.0 | |
|
44 | async-timeout==4.0.2 | |
|
21 | 45 | repoze.lru==0.7 |
|
22 | redis==3.5.3 | |
|
23 | simplejson==3.16.0 | |
|
24 | subprocess32==3.5.4 | |
|
25 | subvertpy==0.10.1 | |
|
46 | scandir==1.10.0 | |
|
47 | setproctitle==1.3.2 | |
|
48 | subvertpy==0.11.0 | |
|
49 | wcwidth==0.2.6 | |
|
26 | 50 | |
|
27 | six==1.11.0 | |
|
28 | translationstring==1.3 | |
|
29 | webob==1.8.5 | |
|
30 | zope.deprecation==4.4.0 | |
|
31 | zope.interface==4.6.0 | |
|
32 | ||
|
33 | ## http servers | |
|
34 | gevent==1.5.0 | |
|
35 | greenlet==0.4.15 | |
|
36 | gunicorn==19.9.0 | |
|
37 | waitress==1.3.1 | |
|
38 | ||
|
39 | ## debug | |
|
40 | ipdb==0.13.2 | |
|
41 | ipython==5.1.0 | |
|
42 | 51 | |
|
43 | 52 | ## test related requirements |
|
44 | -r requirements_test.txt | |
|
53 | #-r requirements_test.txt | |
|
45 | 54 | |
|
46 | 55 | ## uncomment to add the debug libraries |
|
47 | 56 | #-r requirements_debug.txt |
@@ -1,8 +1,22 b'' | |||
|
1 | 1 | ## special libraries we could extend the requirements.txt file with to add some |
|
2 | ## custom libraries useful for debug and memory tracing | |
|
3 | ||
|
4 | ## uncomment inclusion of this file in requirements.txt run make generate-pkgs and nix-shell | |
|
2 | ## custom libraries usefull for debug and memory tracing | |
|
5 | 3 | |
|
6 | 4 | objgraph |
|
7 | 5 | memory-profiler |
|
8 | 6 | pympler |
|
7 | ||
|
8 | ## debug | |
|
9 | ipdb | |
|
10 | ipython | |
|
11 | rich | |
|
12 | ||
|
13 | # format | |
|
14 | flake8 | |
|
15 | ruff | |
|
16 | ||
|
17 | pipdeptree==2.7.1 | |
|
18 | invoke==2.0.0 | |
|
19 | bumpversion==0.6.0 | |
|
20 | bump2version==1.0.1 | |
|
21 | ||
|
22 | docutils-stubs |
@@ -1,16 +1,45 b'' | |||
|
1 | 1 | # test related requirements |
|
2 | pytest==4.6.9 | |
|
3 | py==1.8.1 | |
|
4 | pytest-cov==2.8.1 | |
|
5 | pytest-sugar==0.9.3 | |
|
6 | pytest-runner==5.2.0 | |
|
2 | ||
|
3 | cov-core==1.15.0 | |
|
4 | coverage==7.2.3 | |
|
5 | mock==5.0.2 | |
|
6 | py==1.11.0 | |
|
7 | pytest-cov==4.0.0 | |
|
8 | coverage==7.2.3 | |
|
9 | pytest==7.3.1 | |
|
10 | attrs==22.2.0 | |
|
11 | iniconfig==2.0.0 | |
|
12 | packaging==23.1 | |
|
13 | pluggy==1.0.0 | |
|
7 | 14 | pytest-profiling==1.7.0 |
|
8 | pytest-timeout==1.3.3 | |
|
9 | gprof2dot==2017.9.19 | |
|
15 | gprof2dot==2022.7.29 | |
|
16 | pytest==7.3.1 | |
|
17 | attrs==22.2.0 | |
|
18 | iniconfig==2.0.0 | |
|
19 | packaging==23.1 | |
|
20 | pluggy==1.0.0 | |
|
21 | six==1.16.0 | |
|
22 | pytest-runner==6.0.0 | |
|
23 | pytest-sugar==0.9.7 | |
|
24 | packaging==23.1 | |
|
25 | pytest==7.3.1 | |
|
26 | attrs==22.2.0 | |
|
27 | iniconfig==2.0.0 | |
|
28 | packaging==23.1 | |
|
29 | pluggy==1.0.0 | |
|
30 | termcolor==2.3.0 | |
|
31 | pytest-timeout==2.1.0 | |
|
32 | pytest==7.3.1 | |
|
33 | attrs==22.2.0 | |
|
34 | iniconfig==2.0.0 | |
|
35 | packaging==23.1 | |
|
36 | pluggy==1.0.0 | |
|
37 | webtest==3.0.0 | |
|
38 | beautifulsoup4==4.11.2 | |
|
39 | soupsieve==2.4 | |
|
40 | waitress==2.1.2 | |
|
41 | webob==1.8.7 | |
|
10 | 42 | |
|
11 | mock==3.0.5 | |
|
12 | cov-core==1.15.0 | |
|
13 | coverage==4.5.4 | |
|
14 | ||
|
15 | webtest==2.0.34 | |
|
16 | beautifulsoup4==4.6.3 | |
|
43 | # RhodeCode test-data | |
|
44 | rc_testdata @ https://code.rhodecode.com/upstream/rc-testdata-dist/raw/77378e9097f700b4c1b9391b56199fe63566b5c9/rc_testdata-0.11.0.tar.gz#egg=rc_testdata | |
|
45 | rc_testdata==0.11.0 |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -18,7 +18,7 b'' | |||
|
18 | 18 | import pkgutil |
|
19 | 19 | |
|
20 | 20 | |
|
21 | __version__ = pkgutil.get_data('vcsserver', 'VERSION').strip() | |
|
21 | __version__ = pkgutil.get_data('vcsserver', 'VERSION').strip().decode() | |
|
22 | 22 | |
|
23 | 23 | # link to config for pyramid |
|
24 | 24 | CONFIG = {} |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -16,13 +16,18 b'' | |||
|
16 | 16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
17 | 17 | import os |
|
18 | 18 | import sys |
|
19 | import tempfile | |
|
19 | 20 | import traceback |
|
20 | 21 | import logging |
|
21 | import urlparse | |
|
22 | import urllib.parse | |
|
23 | ||
|
24 | from vcsserver.lib.rc_cache.archive_cache import get_archival_cache_store | |
|
25 | from vcsserver.lib.rc_cache import region_meta | |
|
22 | 26 | |
|
23 | 27 | from vcsserver import exceptions |
|
24 | 28 | from vcsserver.exceptions import NoContentException |
|
25 |
from vcsserver.hgcompat import |
|
|
29 | from vcsserver.hgcompat import archival | |
|
30 | from vcsserver.str_utils import safe_bytes | |
|
26 | 31 | |
|
27 | 32 | log = logging.getLogger(__name__) |
|
28 | 33 | |
@@ -37,7 +42,7 b' class RepoFactory(object):' | |||
|
37 | 42 | repo_type = None |
|
38 | 43 | |
|
39 | 44 | def __init__(self): |
|
40 | pass | |
|
45 | self._cache_region = region_meta.dogpile_cache_regions['repo_object'] | |
|
41 | 46 | |
|
42 | 47 | def _create_config(self, path, config): |
|
43 | 48 | config = {} |
@@ -55,30 +60,33 b' def obfuscate_qs(query_string):' | |||
|
55 | 60 | return None |
|
56 | 61 | |
|
57 | 62 | parsed = [] |
|
58 | for k, v in urlparse.parse_qsl(query_string, keep_blank_values=True): | |
|
63 | for k, v in urllib.parse.parse_qsl(query_string, keep_blank_values=True): | |
|
59 | 64 | if k in ['auth_token', 'api_key']: |
|
60 | 65 | v = "*****" |
|
61 | 66 | parsed.append((k, v)) |
|
62 | 67 | |
|
63 | 68 | return '&'.join('{}{}'.format( |
|
64 |
k, '={}' |
|
|
69 | k, f'={v}' if v else '') for k, v in parsed) | |
|
65 | 70 | |
|
66 | 71 | |
|
67 | def raise_from_original(new_type): | |
|
72 | def raise_from_original(new_type, org_exc: Exception): | |
|
68 | 73 | """ |
|
69 | 74 | Raise a new exception type with original args and traceback. |
|
70 | 75 | """ |
|
76 | ||
|
71 | 77 | exc_type, exc_value, exc_traceback = sys.exc_info() |
|
72 | 78 | new_exc = new_type(*exc_value.args) |
|
79 | ||
|
73 | 80 | # store the original traceback into the new exc |
|
74 |
new_exc._org_exc_tb = traceback.format_ |
|
|
81 | new_exc._org_exc_tb = traceback.format_tb(exc_traceback) | |
|
75 | 82 | |
|
76 | 83 | try: |
|
77 |
raise new_exc |
|
|
84 | raise new_exc.with_traceback(exc_traceback) | |
|
78 | 85 | finally: |
|
79 | 86 | del exc_traceback |
|
80 | 87 | |
|
81 | 88 | |
|
89 | ||
|
82 | 90 | class ArchiveNode(object): |
|
83 | 91 | def __init__(self, path, mode, is_link, raw_bytes): |
|
84 | 92 | self.path = path |
@@ -87,34 +95,58 b' class ArchiveNode(object):' | |||
|
87 | 95 | self.raw_bytes = raw_bytes |
|
88 | 96 | |
|
89 | 97 | |
|
90 |
def archive_ |
|
|
91 |
|
|
|
98 | def store_archive_in_cache(node_walker, archive_key, kind, mtime, archive_at_path, archive_dir_name, | |
|
99 | commit_id, write_metadata=True, extra_metadata=None, cache_config=None): | |
|
92 | 100 | """ |
|
101 | Function that would store an generate archive and send it to a dedicated backend store | |
|
102 | In here we use diskcache | |
|
103 | ||
|
104 | :param node_walker: a generator returning nodes to add to archive | |
|
105 | :param archive_key: key used to store the path | |
|
106 | :param kind: archive kind | |
|
107 | :param mtime: time of creation | |
|
108 | :param archive_at_path: default '/' the path at archive was started. if this is not '/' it means it's a partial archive | |
|
109 | :param archive_dir_name: inside dir name when creating an archive | |
|
110 | :param commit_id: commit sha of revision archive was created at | |
|
111 | :param write_metadata: | |
|
112 | :param extra_metadata: | |
|
113 | :param cache_config: | |
|
114 | ||
|
93 | 115 | walker should be a file walker, for example: |
|
94 | def walker(): | |
|
116 | def node_walker(): | |
|
95 | 117 | for file_info in files: |
|
96 | 118 | yield ArchiveNode(fn, mode, is_link, ctx[fn].data) |
|
97 | 119 | """ |
|
98 | 120 | extra_metadata = extra_metadata or {} |
|
99 | 121 | |
|
122 | d_cache = get_archival_cache_store(config=cache_config) | |
|
123 | ||
|
124 | if archive_key in d_cache: | |
|
125 | with d_cache as d_cache_reader: | |
|
126 | reader, tag = d_cache_reader.get(archive_key, read=True, tag=True, retry=True) | |
|
127 | return reader.name | |
|
128 | ||
|
129 | archive_tmp_path = safe_bytes(tempfile.mkstemp()[1]) | |
|
130 | log.debug('Creating new temp archive in %s', archive_tmp_path) | |
|
131 | ||
|
100 | 132 | if kind == "tgz": |
|
101 |
archiver = archival.tarit(archive_ |
|
|
133 | archiver = archival.tarit(archive_tmp_path, mtime, b"gz") | |
|
102 | 134 | elif kind == "tbz2": |
|
103 |
archiver = archival.tarit(archive_ |
|
|
135 | archiver = archival.tarit(archive_tmp_path, mtime, b"bz2") | |
|
104 | 136 | elif kind == 'zip': |
|
105 |
archiver = archival.zipit(archive_ |
|
|
137 | archiver = archival.zipit(archive_tmp_path, mtime) | |
|
106 | 138 | else: |
|
107 | 139 | raise exceptions.ArchiveException()( |
|
108 |
'Remote does not support: " |
|
|
140 | f'Remote does not support: "{kind}" archive type.') | |
|
109 | 141 | |
|
110 | for f in walker(commit_id, archive_at_path): | |
|
111 | f_path = os.path.join(archive_dir_name, f.path.lstrip('/')) | |
|
142 | for f in node_walker(commit_id, archive_at_path): | |
|
143 | f_path = os.path.join(safe_bytes(archive_dir_name), safe_bytes(f.path).lstrip(b'/')) | |
|
112 | 144 | try: |
|
113 | 145 | archiver.addfile(f_path, f.mode, f.is_link, f.raw_bytes()) |
|
114 | 146 | except NoContentException: |
|
115 | 147 | # NOTE(marcink): this is a special case for SVN so we can create "empty" |
|
116 | 148 | # directories which arent supported by archiver |
|
117 | archiver.addfile(os.path.join(f_path, '.dir'), f.mode, f.is_link, '') | |
|
149 | archiver.addfile(os.path.join(f_path, b'.dir'), f.mode, f.is_link, b'') | |
|
118 | 150 | |
|
119 | 151 | if write_metadata: |
|
120 | 152 | metadata = dict([ |
@@ -123,8 +155,41 b' def archive_repo(walker, archive_dest_pa' | |||
|
123 | 155 | ]) |
|
124 | 156 | metadata.update(extra_metadata) |
|
125 | 157 | |
|
126 |
meta = [" |
|
|
127 | f_path = os.path.join(archive_dir_name, '.archival.txt') | |
|
128 | archiver.addfile(f_path, 0o644, False, '\n'.join(meta)) | |
|
158 | meta = [safe_bytes(f"{f_name}:{value}") for f_name, value in metadata.items()] | |
|
159 | f_path = os.path.join(safe_bytes(archive_dir_name), b'.archival.txt') | |
|
160 | archiver.addfile(f_path, 0o644, False, b'\n'.join(meta)) | |
|
161 | ||
|
162 | archiver.done() | |
|
163 | ||
|
164 | # ensure set & get are atomic | |
|
165 | with d_cache.transact(): | |
|
166 | ||
|
167 | with open(archive_tmp_path, 'rb') as archive_file: | |
|
168 | add_result = d_cache.set(archive_key, archive_file, read=True, tag='db-name', retry=True) | |
|
169 | if not add_result: | |
|
170 | log.error('Failed to store cache for key=%s', archive_key) | |
|
171 | ||
|
172 | os.remove(archive_tmp_path) | |
|
129 | 173 | |
|
130 | return archiver.done() | |
|
174 | reader, tag = d_cache.get(archive_key, read=True, tag=True, retry=True) | |
|
175 | if not reader: | |
|
176 | raise AssertionError(f'empty reader on key={archive_key} added={add_result}') | |
|
177 | ||
|
178 | return reader.name | |
|
179 | ||
|
180 | ||
|
181 | class BinaryEnvelope(object): | |
|
182 | def __init__(self, val): | |
|
183 | self.val = val | |
|
184 | ||
|
185 | ||
|
186 | class BytesEnvelope(bytes): | |
|
187 | def __new__(cls, content): | |
|
188 | if isinstance(content, bytes): | |
|
189 | return super().__new__(cls, content) | |
|
190 | else: | |
|
191 | raise TypeError('Content must be bytes.') | |
|
192 | ||
|
193 | ||
|
194 | class BinaryBytesEnvelope(BytesEnvelope): | |
|
195 | pass |
@@ -0,0 +1,1 b'' | |||
|
1 | # Copyright (C) 2014-2023 RhodeCode GmbH |
@@ -1,6 +1,4 b'' | |||
|
1 | # -*- coding: utf-8 -*- | |
|
2 | ||
|
3 | # Copyright (C) 2010-2020 RhodeCode GmbH | |
|
1 | # Copyright (C) 2010-2023 RhodeCode GmbH | |
|
4 | 2 | # |
|
5 | 3 | # This program is free software: you can redistribute it and/or modify |
|
6 | 4 | # it under the terms of the GNU Affero General Public License, version 3 |
@@ -25,6 +23,9 b' import functools' | |||
|
25 | 23 | import logging |
|
26 | 24 | import tempfile |
|
27 | 25 | import logging.config |
|
26 | ||
|
27 | from vcsserver.type_utils import str2bool, aslist | |
|
28 | ||
|
28 | 29 | log = logging.getLogger(__name__) |
|
29 | 30 | |
|
30 | 31 | # skip keys, that are set here, so we don't double process those |
@@ -33,47 +34,6 b' set_keys = {' | |||
|
33 | 34 | } |
|
34 | 35 | |
|
35 | 36 | |
|
36 | def str2bool(_str): | |
|
37 | """ | |
|
38 | returns True/False value from given string, it tries to translate the | |
|
39 | string into boolean | |
|
40 | ||
|
41 | :param _str: string value to translate into boolean | |
|
42 | :rtype: boolean | |
|
43 | :returns: boolean from given string | |
|
44 | """ | |
|
45 | if _str is None: | |
|
46 | return False | |
|
47 | if _str in (True, False): | |
|
48 | return _str | |
|
49 | _str = str(_str).strip().lower() | |
|
50 | return _str in ('t', 'true', 'y', 'yes', 'on', '1') | |
|
51 | ||
|
52 | ||
|
53 | def aslist(obj, sep=None, strip=True): | |
|
54 | """ | |
|
55 | Returns given string separated by sep as list | |
|
56 | ||
|
57 | :param obj: | |
|
58 | :param sep: | |
|
59 | :param strip: | |
|
60 | """ | |
|
61 | if isinstance(obj, (basestring,)): | |
|
62 | if obj in ['', ""]: | |
|
63 | return [] | |
|
64 | ||
|
65 | lst = obj.split(sep) | |
|
66 | if strip: | |
|
67 | lst = [v.strip() for v in lst] | |
|
68 | return lst | |
|
69 | elif isinstance(obj, (list, tuple)): | |
|
70 | return obj | |
|
71 | elif obj is None: | |
|
72 | return [] | |
|
73 | else: | |
|
74 | return [obj] | |
|
75 | ||
|
76 | ||
|
77 | 37 | class SettingsMaker(object): |
|
78 | 38 | |
|
79 | 39 | def __init__(self, app_settings): |
@@ -81,8 +41,9 b' class SettingsMaker(object):' | |||
|
81 | 41 | |
|
82 | 42 | @classmethod |
|
83 | 43 | def _bool_func(cls, input_val): |
|
84 |
if isinstance(input_val, |
|
|
85 | input_val = input_val.encode('utf8') | |
|
44 | if isinstance(input_val, bytes): | |
|
45 | # decode to str | |
|
46 | input_val = input_val.decode('utf8') | |
|
86 | 47 | return str2bool(input_val) |
|
87 | 48 | |
|
88 | 49 | @classmethod |
@@ -108,10 +69,10 b' class SettingsMaker(object):' | |||
|
108 | 69 | |
|
109 | 70 | # ensure we have our dir created |
|
110 | 71 | if not os.path.isdir(input_val) and ensure_dir: |
|
111 | os.makedirs(input_val, mode=mode) | |
|
72 | os.makedirs(input_val, mode=mode, exist_ok=True) | |
|
112 | 73 | |
|
113 | 74 | if not os.path.isdir(input_val): |
|
114 |
raise Exception('Dir at {} does not exist' |
|
|
75 | raise Exception(f'Dir at {input_val} does not exist') | |
|
115 | 76 | return input_val |
|
116 | 77 | |
|
117 | 78 | @classmethod |
@@ -163,7 +124,7 b' class SettingsMaker(object):' | |||
|
163 | 124 | 'file does not exist.... specify path using logging.logging_conf_file= config setting. ', logging_conf) |
|
164 | 125 | return |
|
165 | 126 | |
|
166 |
with open(logging_conf, 'r |
|
|
127 | with open(logging_conf, 'rt') as f: | |
|
167 | 128 | ini_template = textwrap.dedent(f.read()) |
|
168 | 129 | ini_template = string.Template(ini_template).safe_substitute( |
|
169 | 130 | RC_LOGGING_LEVEL=os.environ.get('RC_LOGGING_LEVEL', '') or level, |
@@ -1,3 +1,5 b'' | |||
|
1 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
2 | ||
|
1 | 3 | """ |
|
2 | 4 | Provides a stub implementation for VCS operations. |
|
3 | 5 |
@@ -1,3 +1,5 b'' | |||
|
1 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
2 | ||
|
1 | 3 | """ |
|
2 | 4 | Implementation of :class:`EchoApp`. |
|
3 | 5 | |
@@ -23,7 +25,7 b' class EchoApp(object):' | |||
|
23 | 25 | status = '200 OK' |
|
24 | 26 | headers = [('Content-Type', 'text/plain')] |
|
25 | 27 | start_response(status, headers) |
|
26 | return ["ECHO"] | |
|
28 | return [b"ECHO"] | |
|
27 | 29 | |
|
28 | 30 | |
|
29 | 31 | class EchoAppStream(object): |
@@ -41,8 +43,8 b' class EchoAppStream(object):' | |||
|
41 | 43 | start_response(status, headers) |
|
42 | 44 | |
|
43 | 45 | def generator(): |
|
44 |
for _ in |
|
|
45 | yield "ECHO" | |
|
46 | for _ in range(1000000): | |
|
47 | yield b"ECHO_STREAM" | |
|
46 | 48 | return generator() |
|
47 | 49 | |
|
48 | 50 |
@@ -1,3 +1,5 b'' | |||
|
1 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
2 | ||
|
1 | 3 | """ |
|
2 | 4 | Provides the same API as :mod:`remote_wsgi`. |
|
3 | 5 | |
@@ -42,4 +44,4 b' def _assert_valid_config(config):' | |||
|
42 | 44 | config = config.copy() |
|
43 | 45 | |
|
44 | 46 | # This is what git needs from config at this stage |
|
45 | config.pop('git_update_server_info') | |
|
47 | config.pop(b'git_update_server_info') |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -109,7 +109,7 b' class HTTPRepoLocked(HTTPLocked):' | |||
|
109 | 109 | def __init__(self, title, status_code=None, **kwargs): |
|
110 | 110 | self.code = status_code or HTTPLocked.code |
|
111 | 111 | self.title = title |
|
112 |
super( |
|
|
112 | super().__init__(**kwargs) | |
|
113 | 113 | |
|
114 | 114 | |
|
115 | 115 | class HTTPRepoBranchProtected(HTTPForbidden): |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -16,4 +16,4 b'' | |||
|
16 | 16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
17 | 17 | |
|
18 | 18 | |
|
19 | from app import create_app | |
|
19 | from .app import create_app |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -19,16 +19,16 b' import re' | |||
|
19 | 19 | import logging |
|
20 | 20 | from wsgiref.util import FileWrapper |
|
21 | 21 | |
|
22 | import simplejson as json | |
|
23 | 22 | from pyramid.config import Configurator |
|
24 | 23 | from pyramid.response import Response, FileIter |
|
25 | 24 | from pyramid.httpexceptions import ( |
|
26 | 25 | HTTPBadRequest, HTTPNotImplemented, HTTPNotFound, HTTPForbidden, |
|
27 | 26 | HTTPUnprocessableEntity) |
|
28 | 27 | |
|
28 | from vcsserver.lib.rc_json import json | |
|
29 | 29 | from vcsserver.git_lfs.lib import OidHandler, LFSOidStore |
|
30 | 30 | from vcsserver.git_lfs.utils import safe_result, get_cython_compat_decorator |
|
31 | from vcsserver.utils import safe_int | |
|
31 | from vcsserver.str_utils import safe_int | |
|
32 | 32 | |
|
33 | 33 | log = logging.getLogger(__name__) |
|
34 | 34 | |
@@ -116,7 +116,7 b' def lfs_objects_batch(request):' | |||
|
116 | 116 | |
|
117 | 117 | obj_data = {'oid': oid} |
|
118 | 118 | |
|
119 |
obj_href = request.route_url('lfs_objects_oid', repo=repo, oid=oid, |
|
|
119 | obj_href = request.route_url('lfs_objects_oid', repo=repo, oid=oid, | |
|
120 | 120 | _scheme=http_scheme) |
|
121 | 121 | obj_verify_href = request.route_url('lfs_objects_verify', repo=repo, |
|
122 | 122 | _scheme=http_scheme) |
@@ -212,7 +212,7 b' def lfs_objects_verify(request):' | |||
|
212 | 212 | |
|
213 | 213 | store_size = store.size_oid() |
|
214 | 214 | if store_size != size: |
|
215 |
msg = 'requested file size mismatch store size: |
|
|
215 | msg = 'requested file size mismatch store size:{} requested:{}'.format( | |
|
216 | 216 | store_size, size) |
|
217 | 217 | return write_response_error( |
|
218 | 218 | HTTPUnprocessableEntity, msg) |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -51,7 +51,7 b' class OidHandler(object):' | |||
|
51 | 51 | |
|
52 | 52 | if not store.has_oid(): |
|
53 | 53 | # error reply back to client that something is wrong with dl |
|
54 |
err_msg = 'object: { |
|
|
54 | err_msg = f'object: {store.oid} does not exist in store' | |
|
55 | 55 | has_errors = OrderedDict( |
|
56 | 56 | error=OrderedDict( |
|
57 | 57 | code=404, |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -18,8 +18,9 b'' | |||
|
18 | 18 | import os |
|
19 | 19 | import pytest |
|
20 | 20 | from webtest.app import TestApp as WebObTestApp |
|
21 | import simplejson as json | |
|
22 | 21 | |
|
22 | from vcsserver.lib.rc_json import json | |
|
23 | from vcsserver.str_utils import safe_bytes | |
|
23 | 24 | from vcsserver.git_lfs.app import create_app |
|
24 | 25 | |
|
25 | 26 | |
@@ -54,19 +55,19 b' class TestLFSApplication(object):' | |||
|
54 | 55 | def test_app_deprecated_endpoint(self, git_lfs_app): |
|
55 | 56 | response = git_lfs_app.post('/repo/info/lfs/objects', status=501) |
|
56 | 57 | assert response.status_code == 501 |
|
57 |
assert json.loads(response.text) == { |
|
|
58 | assert json.loads(response.text) == {'message': 'LFS: v1 api not supported'} | |
|
58 | 59 | |
|
59 | 60 | def test_app_lock_verify_api_not_available(self, git_lfs_app): |
|
60 | 61 | response = git_lfs_app.post('/repo/info/lfs/locks/verify', status=501) |
|
61 | 62 | assert response.status_code == 501 |
|
62 | 63 | assert json.loads(response.text) == { |
|
63 |
|
|
|
64 | 'message': 'GIT LFS locking api not supported'} | |
|
64 | 65 | |
|
65 | 66 | def test_app_lock_api_not_available(self, git_lfs_app): |
|
66 | 67 | response = git_lfs_app.post('/repo/info/lfs/locks', status=501) |
|
67 | 68 | assert response.status_code == 501 |
|
68 | 69 | assert json.loads(response.text) == { |
|
69 |
|
|
|
70 | 'message': 'GIT LFS locking api not supported'} | |
|
70 | 71 | |
|
71 | 72 | def test_app_batch_api_missing_auth(self, git_lfs_app): |
|
72 | 73 | git_lfs_app.post_json( |
@@ -77,14 +78,14 b' class TestLFSApplication(object):' | |||
|
77 | 78 | '/repo/info/lfs/objects/batch', params={}, status=400, |
|
78 | 79 | extra_environ=http_auth) |
|
79 | 80 | assert json.loads(response.text) == { |
|
80 |
|
|
|
81 | 'message': 'unsupported operation mode: `None`'} | |
|
81 | 82 | |
|
82 | 83 | def test_app_batch_api_missing_objects(self, git_lfs_app, http_auth): |
|
83 | 84 | response = git_lfs_app.post_json( |
|
84 | 85 | '/repo/info/lfs/objects/batch', params={'operation': 'download'}, |
|
85 | 86 | status=400, extra_environ=http_auth) |
|
86 | 87 | assert json.loads(response.text) == { |
|
87 |
|
|
|
88 | 'message': 'missing objects data'} | |
|
88 | 89 | |
|
89 | 90 | def test_app_batch_api_unsupported_data_in_objects( |
|
90 | 91 | self, git_lfs_app, http_auth): |
@@ -94,7 +95,7 b' class TestLFSApplication(object):' | |||
|
94 | 95 | '/repo/info/lfs/objects/batch', params=params, status=400, |
|
95 | 96 | extra_environ=http_auth) |
|
96 | 97 | assert json.loads(response.text) == { |
|
97 |
|
|
|
98 | 'message': 'unsupported data in objects'} | |
|
98 | 99 | |
|
99 | 100 | def test_app_batch_api_download_missing_object( |
|
100 | 101 | self, git_lfs_app, http_auth): |
@@ -105,12 +106,12 b' class TestLFSApplication(object):' | |||
|
105 | 106 | extra_environ=http_auth) |
|
106 | 107 | |
|
107 | 108 | expected_objects = [ |
|
108 |
{ |
|
|
109 |
|
|
|
110 |
|
|
|
111 |
|
|
|
112 |
|
|
|
113 |
|
|
|
109 | {'authenticated': True, | |
|
110 | 'errors': {'error': { | |
|
111 | 'code': 404, | |
|
112 | 'message': 'object: 123 does not exist in store'}}, | |
|
113 | 'oid': '123', | |
|
114 | 'size': '1024'} | |
|
114 | 115 | ] |
|
115 | 116 | assert json.loads(response.text) == { |
|
116 | 117 | 'objects': expected_objects, 'transfer': 'basic'} |
@@ -121,7 +122,7 b' class TestLFSApplication(object):' | |||
|
121 | 122 | if not os.path.isdir(os.path.dirname(oid_path)): |
|
122 | 123 | os.makedirs(os.path.dirname(oid_path)) |
|
123 | 124 | with open(oid_path, 'wb') as f: |
|
124 | f.write('OID_CONTENT') | |
|
125 | f.write(safe_bytes('OID_CONTENT')) | |
|
125 | 126 | |
|
126 | 127 | params = {'operation': 'download', |
|
127 | 128 | 'objects': [{'oid': oid, 'size': '1024'}]} |
@@ -130,14 +131,14 b' class TestLFSApplication(object):' | |||
|
130 | 131 | extra_environ=http_auth) |
|
131 | 132 | |
|
132 | 133 | expected_objects = [ |
|
133 |
{ |
|
|
134 |
|
|
|
135 |
|
|
|
136 |
|
|
|
137 |
|
|
|
134 | {'authenticated': True, | |
|
135 | 'actions': { | |
|
136 | 'download': { | |
|
137 | 'header': {'Authorization': 'Basic XXXXX'}, | |
|
138 | 'href': 'http://localhost/repo/info/lfs/objects/456'}, | |
|
138 | 139 | }, |
|
139 |
|
|
|
140 |
|
|
|
140 | 'oid': '456', | |
|
141 | 'size': '1024'} | |
|
141 | 142 | ] |
|
142 | 143 | assert json.loads(response.text) == { |
|
143 | 144 | 'objects': expected_objects, 'transfer': 'basic'} |
@@ -149,18 +150,18 b' class TestLFSApplication(object):' | |||
|
149 | 150 | '/repo/info/lfs/objects/batch', params=params, |
|
150 | 151 | extra_environ=http_auth) |
|
151 | 152 | expected_objects = [ |
|
152 |
{ |
|
|
153 |
|
|
|
154 |
|
|
|
155 |
|
|
|
156 |
|
|
|
157 |
|
|
|
158 |
|
|
|
159 |
|
|
|
160 |
|
|
|
153 | {'authenticated': True, | |
|
154 | 'actions': { | |
|
155 | 'upload': { | |
|
156 | 'header': {'Authorization': 'Basic XXXXX', | |
|
157 | 'Transfer-Encoding': 'chunked'}, | |
|
158 | 'href': 'http://localhost/repo/info/lfs/objects/123'}, | |
|
159 | 'verify': { | |
|
160 | 'header': {'Authorization': 'Basic XXXXX'}, | |
|
161 | 'href': 'http://localhost/repo/info/lfs/verify'} | |
|
161 | 162 | }, |
|
162 |
|
|
|
163 |
|
|
|
163 | 'oid': '123', | |
|
164 | 'size': '1024'} | |
|
164 | 165 | ] |
|
165 | 166 | assert json.loads(response.text) == { |
|
166 | 167 | 'objects': expected_objects, 'transfer': 'basic'} |
@@ -172,18 +173,18 b' class TestLFSApplication(object):' | |||
|
172 | 173 | '/repo/info/lfs/objects/batch', params=params, |
|
173 | 174 | extra_environ=http_auth) |
|
174 | 175 | expected_objects = [ |
|
175 |
{ |
|
|
176 |
|
|
|
177 |
|
|
|
178 |
|
|
|
179 |
|
|
|
180 |
|
|
|
181 |
|
|
|
182 |
|
|
|
183 |
|
|
|
176 | {'authenticated': True, | |
|
177 | 'actions': { | |
|
178 | 'upload': { | |
|
179 | 'header': {'Authorization': 'Basic XXXXX', | |
|
180 | 'Transfer-Encoding': 'chunked'}, | |
|
181 | 'href': 'https://localhost/repo/info/lfs/objects/123'}, | |
|
182 | 'verify': { | |
|
183 | 'header': {'Authorization': 'Basic XXXXX'}, | |
|
184 | 'href': 'https://localhost/repo/info/lfs/verify'} | |
|
184 | 185 | }, |
|
185 |
|
|
|
186 |
|
|
|
186 | 'oid': '123', | |
|
187 | 'size': '1024'} | |
|
187 | 188 | ] |
|
188 | 189 | assert json.loads(response.text) == { |
|
189 | 190 | 'objects': expected_objects, 'transfer': 'basic'} |
@@ -195,7 +196,7 b' class TestLFSApplication(object):' | |||
|
195 | 196 | status=400) |
|
196 | 197 | |
|
197 | 198 | assert json.loads(response.text) == { |
|
198 |
|
|
|
199 | 'message': 'missing oid and size in request data'} | |
|
199 | 200 | |
|
200 | 201 | def test_app_verify_api_missing_obj(self, git_lfs_app): |
|
201 | 202 | params = {'oid': 'missing', 'size': '1024'} |
@@ -204,7 +205,7 b' class TestLFSApplication(object):' | |||
|
204 | 205 | status=404) |
|
205 | 206 | |
|
206 | 207 | assert json.loads(response.text) == { |
|
207 |
|
|
|
208 | 'message': 'oid `missing` does not exists in store'} | |
|
208 | 209 | |
|
209 | 210 | def test_app_verify_api_size_mismatch(self, git_lfs_app): |
|
210 | 211 | oid = 'existing' |
@@ -212,15 +213,15 b' class TestLFSApplication(object):' | |||
|
212 | 213 | if not os.path.isdir(os.path.dirname(oid_path)): |
|
213 | 214 | os.makedirs(os.path.dirname(oid_path)) |
|
214 | 215 | with open(oid_path, 'wb') as f: |
|
215 | f.write('OID_CONTENT') | |
|
216 | f.write(safe_bytes('OID_CONTENT')) | |
|
216 | 217 | |
|
217 | 218 | params = {'oid': oid, 'size': '1024'} |
|
218 | 219 | response = git_lfs_app.post_json( |
|
219 | 220 | '/repo/info/lfs/verify', params=params, status=422) |
|
220 | 221 | |
|
221 | 222 | assert json.loads(response.text) == { |
|
222 |
|
|
|
223 |
|
|
|
223 | 'message': 'requested file size mismatch ' | |
|
224 | 'store size:11 requested:1024'} | |
|
224 | 225 | |
|
225 | 226 | def test_app_verify_api(self, git_lfs_app): |
|
226 | 227 | oid = 'existing' |
@@ -228,14 +229,14 b' class TestLFSApplication(object):' | |||
|
228 | 229 | if not os.path.isdir(os.path.dirname(oid_path)): |
|
229 | 230 | os.makedirs(os.path.dirname(oid_path)) |
|
230 | 231 | with open(oid_path, 'wb') as f: |
|
231 | f.write('OID_CONTENT') | |
|
232 | f.write(safe_bytes('OID_CONTENT')) | |
|
232 | 233 | |
|
233 | 234 | params = {'oid': oid, 'size': 11} |
|
234 | 235 | response = git_lfs_app.post_json( |
|
235 | 236 | '/repo/info/lfs/verify', params=params) |
|
236 | 237 | |
|
237 | 238 | assert json.loads(response.text) == { |
|
238 |
|
|
|
239 | 'message': {'size': 'ok', 'in_store': 'ok'}} | |
|
239 | 240 | |
|
240 | 241 | def test_app_download_api_oid_not_existing(self, git_lfs_app): |
|
241 | 242 | oid = 'missing' |
@@ -244,7 +245,7 b' class TestLFSApplication(object):' | |||
|
244 | 245 | '/repo/info/lfs/objects/{oid}'.format(oid=oid), status=404) |
|
245 | 246 | |
|
246 | 247 | assert json.loads(response.text) == { |
|
247 |
|
|
|
248 | 'message': 'requested file with oid `missing` not found in store'} | |
|
248 | 249 | |
|
249 | 250 | def test_app_download_api(self, git_lfs_app): |
|
250 | 251 | oid = 'existing' |
@@ -252,7 +253,7 b' class TestLFSApplication(object):' | |||
|
252 | 253 | if not os.path.isdir(os.path.dirname(oid_path)): |
|
253 | 254 | os.makedirs(os.path.dirname(oid_path)) |
|
254 | 255 | with open(oid_path, 'wb') as f: |
|
255 | f.write('OID_CONTENT') | |
|
256 | f.write(safe_bytes('OID_CONTENT')) | |
|
256 | 257 | |
|
257 | 258 | response = git_lfs_app.get( |
|
258 | 259 | '/repo/info/lfs/objects/{oid}'.format(oid=oid)) |
@@ -264,7 +265,7 b' class TestLFSApplication(object):' | |||
|
264 | 265 | response = git_lfs_app.put( |
|
265 | 266 | '/repo/info/lfs/objects/{oid}'.format(oid=oid), params='CONTENT') |
|
266 | 267 | |
|
267 |
assert json.loads(response.text) == { |
|
|
268 | assert json.loads(response.text) == {'upload': 'ok'} | |
|
268 | 269 | |
|
269 | 270 | # verify that we actually wrote that OID |
|
270 | 271 | oid_path = os.path.join(git_lfs_app._store, oid) |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -17,6 +17,7 b'' | |||
|
17 | 17 | |
|
18 | 18 | import os |
|
19 | 19 | import pytest |
|
20 | from vcsserver.str_utils import safe_bytes | |
|
20 | 21 | from vcsserver.git_lfs.lib import OidHandler, LFSOidStore |
|
21 | 22 | |
|
22 | 23 | |
@@ -70,7 +71,7 b' class TestOidHandler(object):' | |||
|
70 | 71 | os.makedirs(os.path.dirname(store.oid_path)) |
|
71 | 72 | |
|
72 | 73 | with open(store.oid_path, 'wb') as f: |
|
73 | f.write('CONTENT') | |
|
74 | f.write(safe_bytes('CONTENT')) | |
|
74 | 75 | |
|
75 | 76 | response, has_errors = oid_handler.exec_operation('download') |
|
76 | 77 | |
@@ -86,7 +87,7 b' class TestOidHandler(object):' | |||
|
86 | 87 | os.makedirs(os.path.dirname(store.oid_path)) |
|
87 | 88 | |
|
88 | 89 | with open(store.oid_path, 'wb') as f: |
|
89 | f.write('CONTENT') | |
|
90 | f.write(safe_bytes('CONTENT')) | |
|
90 | 91 | oid_handler.obj_size = 7 |
|
91 | 92 | response, has_errors = oid_handler.exec_operation('upload') |
|
92 | 93 | assert has_errors is None |
@@ -98,7 +99,7 b' class TestOidHandler(object):' | |||
|
98 | 99 | os.makedirs(os.path.dirname(store.oid_path)) |
|
99 | 100 | |
|
100 | 101 | with open(store.oid_path, 'wb') as f: |
|
101 | f.write('CONTENT') | |
|
102 | f.write(safe_bytes('CONTENT')) | |
|
102 | 103 | |
|
103 | 104 | oid_handler.obj_size = 10240 |
|
104 | 105 | response, has_errors = oid_handler.exec_operation('upload') |
@@ -127,7 +128,7 b' class TestLFSStore(object):' | |||
|
127 | 128 | |
|
128 | 129 | engine = lfs_store.get_engine(mode='wb') |
|
129 | 130 | with engine as f: |
|
130 | f.write('CONTENT') | |
|
131 | f.write(safe_bytes('CONTENT')) | |
|
131 | 132 | |
|
132 | 133 | assert os.path.isfile(oid_location) |
|
133 | 134 | |
@@ -136,6 +137,6 b' class TestLFSStore(object):' | |||
|
136 | 137 | assert lfs_store.has_oid() is False |
|
137 | 138 | engine = lfs_store.get_engine(mode='wb') |
|
138 | 139 | with engine as f: |
|
139 | f.write('CONTENT') | |
|
140 | f.write(safe_bytes('CONTENT')) | |
|
140 | 141 | |
|
141 | assert lfs_store.has_oid() is True No newline at end of file | |
|
142 | assert lfs_store.has_oid() is True |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -21,8 +21,11 b' Mercurial libs compatibility' | |||
|
21 | 21 | |
|
22 | 22 | import mercurial |
|
23 | 23 | from mercurial import demandimport |
|
24 | ||
|
24 | 25 | # patch demandimport, due to bug in mercurial when it always triggers |
|
25 | 26 | # demandimport.enable() |
|
27 | from vcsserver.str_utils import safe_bytes | |
|
28 | ||
|
26 | 29 | demandimport.enable = lambda *args, **kwargs: 1 |
|
27 | 30 | |
|
28 | 31 | from mercurial import ui |
@@ -39,7 +42,8 b' from mercurial import subrepo' | |||
|
39 | 42 | from mercurial import subrepoutil |
|
40 | 43 | from mercurial import tags as hg_tag |
|
41 | 44 | from mercurial import util as hgutil |
|
42 |
from mercurial.commands import clone, |
|
|
45 | from mercurial.commands import clone, pull | |
|
46 | from mercurial.node import nullid | |
|
43 | 47 | from mercurial.context import memctx, memfilectx |
|
44 | 48 | from mercurial.error import ( |
|
45 | 49 | LookupError, RepoError, RepoLookupError, Abort, InterventionRequired, |
@@ -53,7 +57,7 b' from mercurial.encoding import tolocal' | |||
|
53 | 57 | from mercurial.discovery import findcommonoutgoing |
|
54 | 58 | from mercurial.hg import peer |
|
55 | 59 | from mercurial.httppeer import makepeer |
|
56 | from mercurial.util import url as hg_url | |
|
60 | from mercurial.utils.urlutil import url as hg_url | |
|
57 | 61 | from mercurial.scmutil import revrange, revsymbol |
|
58 | 62 | from mercurial.node import nullrev |
|
59 | 63 | from mercurial import exchange |
@@ -63,8 +67,13 b' from hgext import largefiles' | |||
|
63 | 67 | # infinit looping when given invalid resources |
|
64 | 68 | from mercurial.url import httpbasicauthhandler, httpdigestauthhandler |
|
65 | 69 | |
|
70 | # hg strip is in core now | |
|
71 | from mercurial import strip as hgext_strip | |
|
72 | ||
|
66 | 73 | |
|
67 | 74 | def get_ctx(repo, ref): |
|
75 | if not isinstance(ref, int): | |
|
76 | ref = safe_bytes(ref) | |
|
68 | 77 | try: |
|
69 | 78 | ctx = repo[ref] |
|
70 | 79 | except (ProgrammingError, TypeError): |
@@ -73,7 +82,7 b' def get_ctx(repo, ref):' | |||
|
73 | 82 | ctx = revsymbol(repo, ref) |
|
74 | 83 | except (LookupError, RepoLookupError): |
|
75 | 84 | # Similar case as above but only for refs that are not numeric |
|
76 |
if isinstance(ref, |
|
|
85 | if isinstance(ref, int): | |
|
77 | 86 | raise |
|
78 | 87 | ctx = revsymbol(repo, ref) |
|
79 | 88 | return ctx |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -62,7 +62,7 b' def _dynamic_capabilities_wrapper(lfprot' | |||
|
62 | 62 | |
|
63 | 63 | def patch_subrepo_type_mapping(): |
|
64 | 64 | from collections import defaultdict |
|
65 | from hgcompat import subrepo, subrepoutil | |
|
65 | from .hgcompat import subrepo, subrepoutil | |
|
66 | 66 | from vcsserver.exceptions import SubrepoMergeException |
|
67 | 67 | |
|
68 | 68 | class NoOpSubrepo(subrepo.abstractsubrepo): |
@@ -1,7 +1,5 b'' | |||
|
1 | # -*- coding: utf-8 -*- | |
|
2 | ||
|
3 | 1 |
|
|
4 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
5 | 3 | # |
|
6 | 4 | # This program is free software; you can redistribute it and/or modify |
|
7 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -25,6 +23,7 b' import logging' | |||
|
25 | 23 | import pkg_resources |
|
26 | 24 | |
|
27 | 25 | import vcsserver |
|
26 | from vcsserver.str_utils import safe_bytes | |
|
28 | 27 | |
|
29 | 28 | log = logging.getLogger(__name__) |
|
30 | 29 | |
@@ -49,7 +48,7 b' def install_git_hooks(repo_path, bare, e' | |||
|
49 | 48 | hooks_path = get_git_hooks_path(repo_path, bare) |
|
50 | 49 | |
|
51 | 50 | if not os.path.isdir(hooks_path): |
|
52 | os.makedirs(hooks_path, mode=0o777) | |
|
51 | os.makedirs(hooks_path, mode=0o777, exist_ok=True) | |
|
53 | 52 | |
|
54 | 53 | tmpl_post = pkg_resources.resource_string( |
|
55 | 54 | 'vcsserver', '/'.join( |
@@ -70,14 +69,13 b' def install_git_hooks(repo_path, bare, e' | |||
|
70 | 69 | log.debug('writing git %s hook file at %s !', h_type, _hook_file) |
|
71 | 70 | try: |
|
72 | 71 | with open(_hook_file, 'wb') as f: |
|
73 | template = template.replace( | |
|
74 | '_TMPL_', vcsserver.__version__) | |
|
75 |
template = template.replace('_ |
|
|
76 |
template = template.replace('_ |
|
|
77 | template = template.replace('_PATH_', path) | |
|
72 | template = template.replace(b'_TMPL_', safe_bytes(vcsserver.__version__)) | |
|
73 | template = template.replace(b'_DATE_', safe_bytes(timestamp)) | |
|
74 | template = template.replace(b'_ENV_', safe_bytes(executable)) | |
|
75 | template = template.replace(b'_PATH_', safe_bytes(path)) | |
|
78 | 76 | f.write(template) |
|
79 | 77 | os.chmod(_hook_file, 0o755) |
|
80 |
except |
|
|
78 | except OSError: | |
|
81 | 79 | log.exception('error writing hook file %s', _hook_file) |
|
82 | 80 | else: |
|
83 | 81 | log.debug('skipping writing hook file') |
@@ -102,7 +100,7 b' def install_svn_hooks(repo_path, executa' | |||
|
102 | 100 | executable = executable or sys.executable |
|
103 | 101 | hooks_path = get_svn_hooks_path(repo_path) |
|
104 | 102 | if not os.path.isdir(hooks_path): |
|
105 | os.makedirs(hooks_path, mode=0o777) | |
|
103 | os.makedirs(hooks_path, mode=0o777, exist_ok=True) | |
|
106 | 104 | |
|
107 | 105 | tmpl_post = pkg_resources.resource_string( |
|
108 | 106 | 'vcsserver', '/'.join( |
@@ -124,15 +122,14 b' def install_svn_hooks(repo_path, executa' | |||
|
124 | 122 | |
|
125 | 123 | try: |
|
126 | 124 | with open(_hook_file, 'wb') as f: |
|
127 | template = template.replace( | |
|
128 | '_TMPL_', vcsserver.__version__) | |
|
129 |
template = template.replace('_ |
|
|
130 |
template = template.replace('_ |
|
|
131 | template = template.replace('_PATH_', path) | |
|
125 | template = template.replace(b'_TMPL_', safe_bytes(vcsserver.__version__)) | |
|
126 | template = template.replace(b'_DATE_', safe_bytes(timestamp)) | |
|
127 | template = template.replace(b'_ENV_', safe_bytes(executable)) | |
|
128 | template = template.replace(b'_PATH_', safe_bytes(path)) | |
|
132 | 129 | |
|
133 | 130 | f.write(template) |
|
134 | 131 | os.chmod(_hook_file, 0o755) |
|
135 |
except |
|
|
132 | except OSError: | |
|
136 | 133 | log.exception('error writing hook file %s', _hook_file) |
|
137 | 134 | else: |
|
138 | 135 | log.debug('skipping writing hook file') |
@@ -141,16 +138,16 b' def install_svn_hooks(repo_path, executa' | |||
|
141 | 138 | |
|
142 | 139 | |
|
143 | 140 | def get_version_from_hook(hook_path): |
|
144 | version = '' | |
|
141 | version = b'' | |
|
145 | 142 | hook_content = read_hook_content(hook_path) |
|
146 |
matches = re.search(r' |
|
|
143 | matches = re.search(rb'RC_HOOK_VER\s*=\s*(.*)', hook_content) | |
|
147 | 144 | if matches: |
|
148 | 145 | try: |
|
149 | 146 | version = matches.groups()[0] |
|
150 | 147 | log.debug('got version %s from hooks.', version) |
|
151 | 148 | except Exception: |
|
152 | 149 | log.exception("Exception while reading the hook version.") |
|
153 | return version.replace("'", "") | |
|
150 | return version.replace(b"'", b"") | |
|
154 | 151 | |
|
155 | 152 | |
|
156 | 153 | def check_rhodecode_hook(hook_path): |
@@ -169,8 +166,8 b' def check_rhodecode_hook(hook_path):' | |||
|
169 | 166 | return False |
|
170 | 167 | |
|
171 | 168 | |
|
172 | def read_hook_content(hook_path): | |
|
173 | content = '' | |
|
169 | def read_hook_content(hook_path) -> bytes: | |
|
170 | content = b'' | |
|
174 | 171 | if os.path.isfile(hook_path): |
|
175 | 172 | with open(hook_path, 'rb') as f: |
|
176 | 173 | content = f.read() |
@@ -11,7 +11,7 b' try:' | |||
|
11 | 11 | except ImportError: |
|
12 | 12 | if os.environ.get('RC_DEBUG_GIT_HOOK'): |
|
13 | 13 | import traceback |
|
14 |
print |
|
|
14 | print(traceback.format_exc()) | |
|
15 | 15 | hooks = None |
|
16 | 16 | |
|
17 | 17 | |
@@ -42,7 +42,7 b' def main():' | |||
|
42 | 42 | # TODO: johbo: Improve handling of this special case |
|
43 | 43 | if not getattr(error, '_vcs_kind', None) == 'repo_locked': |
|
44 | 44 | raise |
|
45 |
print |
|
|
45 | print(f'ERROR: {error}') | |
|
46 | 46 | sys.exit(1) |
|
47 | 47 | sys.exit(0) |
|
48 | 48 |
@@ -11,7 +11,7 b' try:' | |||
|
11 | 11 | except ImportError: |
|
12 | 12 | if os.environ.get('RC_DEBUG_GIT_HOOK'): |
|
13 | 13 | import traceback |
|
14 |
print |
|
|
14 | print(traceback.format_exc()) | |
|
15 | 15 | hooks = None |
|
16 | 16 | |
|
17 | 17 | |
@@ -42,7 +42,7 b' def main():' | |||
|
42 | 42 | # TODO: johbo: Improve handling of this special case |
|
43 | 43 | if not getattr(error, '_vcs_kind', None) == 'repo_locked': |
|
44 | 44 | raise |
|
45 |
print |
|
|
45 | print(f'ERROR: {error}') | |
|
46 | 46 | sys.exit(1) |
|
47 | 47 | sys.exit(0) |
|
48 | 48 |
@@ -12,7 +12,7 b' try:' | |||
|
12 | 12 | except ImportError: |
|
13 | 13 | if os.environ.get('RC_DEBUG_SVN_HOOK'): |
|
14 | 14 | import traceback |
|
15 |
print |
|
|
15 | print(traceback.format_exc()) | |
|
16 | 16 | hooks = None |
|
17 | 17 | |
|
18 | 18 | |
@@ -40,7 +40,7 b' def main():' | |||
|
40 | 40 | # TODO: johbo: Improve handling of this special case |
|
41 | 41 | if not getattr(error, '_vcs_kind', None) == 'repo_locked': |
|
42 | 42 | raise |
|
43 |
print |
|
|
43 | print(f'ERROR: {error}') | |
|
44 | 44 | sys.exit(1) |
|
45 | 45 | sys.exit(0) |
|
46 | 46 |
@@ -12,7 +12,7 b' try:' | |||
|
12 | 12 | except ImportError: |
|
13 | 13 | if os.environ.get('RC_DEBUG_SVN_HOOK'): |
|
14 | 14 | import traceback |
|
15 |
print |
|
|
15 | print(traceback.format_exc()) | |
|
16 | 16 | hooks = None |
|
17 | 17 | |
|
18 | 18 | |
@@ -43,7 +43,7 b' def main():' | |||
|
43 | 43 | # TODO: johbo: Improve handling of this special case |
|
44 | 44 | if not getattr(error, '_vcs_kind', None) == 'repo_locked': |
|
45 | 45 | raise |
|
46 |
print |
|
|
46 | print(f'ERROR: {error}') | |
|
47 | 47 | sys.exit(1) |
|
48 | 48 | sys.exit(0) |
|
49 | 49 |
@@ -1,7 +1,5 b'' | |||
|
1 | # -*- coding: utf-8 -*- | |
|
2 | ||
|
3 | 1 |
|
|
4 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
5 | 3 | # |
|
6 | 4 | # This program is free software; you can redistribute it and/or modify |
|
7 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -24,51 +22,70 b' import logging' | |||
|
24 | 22 | import collections |
|
25 | 23 | import importlib |
|
26 | 24 | import base64 |
|
25 | import msgpack | |
|
26 | import dataclasses | |
|
27 | import pygit2 | |
|
27 | 28 | |
|
28 | from httplib import HTTPConnection | |
|
29 | import http.client | |
|
29 | 30 | |
|
30 | 31 | |
|
31 | 32 | import mercurial.scmutil |
|
32 | 33 | import mercurial.node |
|
33 | import simplejson as json | |
|
34 | 34 | |
|
35 | from vcsserver.lib.rc_json import json | |
|
35 | 36 | from vcsserver import exceptions, subprocessio, settings |
|
37 | from vcsserver.str_utils import ascii_str, safe_str | |
|
38 | from vcsserver.remote.git import Repository | |
|
36 | 39 | |
|
37 | 40 | log = logging.getLogger(__name__) |
|
38 | 41 | |
|
39 | 42 | |
|
40 | 43 | class HooksHttpClient(object): |
|
44 | proto = 'msgpack.v1' | |
|
41 | 45 | connection = None |
|
42 | 46 | |
|
43 | 47 | def __init__(self, hooks_uri): |
|
44 | 48 | self.hooks_uri = hooks_uri |
|
45 | 49 | |
|
50 | def __repr__(self): | |
|
51 | return f'{self.__class__}(hook_uri={self.hooks_uri}, proto={self.proto})' | |
|
52 | ||
|
46 | 53 | def __call__(self, method, extras): |
|
47 | connection = HTTPConnection(self.hooks_uri) | |
|
48 | body = self._serialize(method, extras) | |
|
49 | try: | |
|
50 | connection.request('POST', '/', body) | |
|
51 | except Exception: | |
|
52 | log.error('Hooks calling Connection failed on %s', connection.__dict__) | |
|
53 | raise | |
|
54 | response = connection.getresponse() | |
|
55 | ||
|
56 | response_data = response.read() | |
|
54 | connection = http.client.HTTPConnection(self.hooks_uri) | |
|
55 | # binary msgpack body | |
|
56 | headers, body = self._serialize(method, extras) | |
|
57 | log.debug('Doing a new hooks call using HTTPConnection to %s', self.hooks_uri) | |
|
57 | 58 | |
|
58 | 59 | try: |
|
59 | return json.loads(response_data) | |
|
60 | except Exception: | |
|
61 | log.exception('Failed to decode hook response json data. ' | |
|
62 | 'response_code:%s, raw_data:%s', | |
|
63 | response.status, response_data) | |
|
64 | raise | |
|
60 | try: | |
|
61 | connection.request('POST', '/', body, headers) | |
|
62 | except Exception as error: | |
|
63 | log.error('Hooks calling Connection failed on %s, org error: %s', connection.__dict__, error) | |
|
64 | raise | |
|
65 | 65 | |
|
66 | def _serialize(self, hook_name, extras): | |
|
66 | response = connection.getresponse() | |
|
67 | try: | |
|
68 | return msgpack.load(response) | |
|
69 | except Exception: | |
|
70 | response_data = response.read() | |
|
71 | log.exception('Failed to decode hook response json data. ' | |
|
72 | 'response_code:%s, raw_data:%s', | |
|
73 | response.status, response_data) | |
|
74 | raise | |
|
75 | finally: | |
|
76 | connection.close() | |
|
77 | ||
|
78 | @classmethod | |
|
79 | def _serialize(cls, hook_name, extras): | |
|
67 | 80 | data = { |
|
68 | 81 | 'method': hook_name, |
|
69 | 82 | 'extras': extras |
|
70 | 83 | } |
|
71 | return json.dumps(data) | |
|
84 | headers = { | |
|
85 | "rc-hooks-protocol": cls.proto, | |
|
86 | "Connection": "keep-alive" | |
|
87 | } | |
|
88 | return headers, msgpack.packb(data) | |
|
72 | 89 | |
|
73 | 90 | |
|
74 | 91 | class HooksDummyClient(object): |
@@ -98,7 +115,7 b' class HgMessageWriter(RemoteMessageWrite' | |||
|
98 | 115 | def __init__(self, ui): |
|
99 | 116 | self.ui = ui |
|
100 | 117 | |
|
101 | def write(self, message): | |
|
118 | def write(self, message: str): | |
|
102 | 119 | # TODO: Check why the quiet flag is set by default. |
|
103 | 120 | old = self.ui.quiet |
|
104 | 121 | self.ui.quiet = False |
@@ -112,8 +129,8 b' class GitMessageWriter(RemoteMessageWrit' | |||
|
112 | 129 | def __init__(self, stdout=None): |
|
113 | 130 | self.stdout = stdout or sys.stdout |
|
114 | 131 | |
|
115 | def write(self, message): | |
|
116 |
self.stdout.write(message |
|
|
132 | def write(self, message: str): | |
|
133 | self.stdout.write(message) | |
|
117 | 134 | |
|
118 | 135 | |
|
119 | 136 | class SvnMessageWriter(RemoteMessageWriter): |
@@ -130,6 +147,7 b' class SvnMessageWriter(RemoteMessageWrit' | |||
|
130 | 147 | def _handle_exception(result): |
|
131 | 148 | exception_class = result.get('exception') |
|
132 | 149 | exception_traceback = result.get('exception_traceback') |
|
150 | log.debug('Handling hook-call exception: %s', exception_class) | |
|
133 | 151 | |
|
134 | 152 | if exception_traceback: |
|
135 | 153 | log.error('Got traceback from remote call:%s', exception_traceback) |
@@ -141,13 +159,15 b' def _handle_exception(result):' | |||
|
141 | 159 | elif exception_class == 'RepositoryError': |
|
142 | 160 | raise exceptions.VcsException()(*result['exception_args']) |
|
143 | 161 | elif exception_class: |
|
144 | raise Exception('Got remote exception "%s" with args "%s"' % | |
|
145 |
|
|
|
162 | raise Exception( | |
|
163 | f"""Got remote exception "{exception_class}" with args "{result['exception_args']}" """ | |
|
164 | ) | |
|
146 | 165 | |
|
147 | 166 | |
|
148 | 167 | def _get_hooks_client(extras): |
|
149 | 168 | hooks_uri = extras.get('hooks_uri') |
|
150 | 169 | is_shadow_repo = extras.get('is_shadow_repo') |
|
170 | ||
|
151 | 171 | if hooks_uri: |
|
152 | 172 | return HooksHttpClient(extras['hooks_uri']) |
|
153 | 173 | elif is_shadow_repo: |
@@ -161,7 +181,6 b' def _call_hook(hook_name, extras, writer' | |||
|
161 | 181 | log.debug('Hooks, using client:%s', hooks_client) |
|
162 | 182 | result = hooks_client(hook_name, extras) |
|
163 | 183 | log.debug('Hooks got result: %s', result) |
|
164 | ||
|
165 | 184 | _handle_exception(result) |
|
166 | 185 | writer.write(result['output']) |
|
167 | 186 | |
@@ -169,7 +188,7 b' def _call_hook(hook_name, extras, writer' | |||
|
169 | 188 | |
|
170 | 189 | |
|
171 | 190 | def _extras_from_ui(ui): |
|
172 | hook_data = ui.config('rhodecode', 'RC_SCM_DATA') | |
|
191 | hook_data = ui.config(b'rhodecode', b'RC_SCM_DATA') | |
|
173 | 192 | if not hook_data: |
|
174 | 193 | # maybe it's inside environ ? |
|
175 | 194 | env_hook_data = os.environ.get('RC_SCM_DATA') |
@@ -192,8 +211,8 b' def _rev_range_hash(repo, node, check_he' | |||
|
192 | 211 | for rev in range(start, end): |
|
193 | 212 | revs.append(rev) |
|
194 | 213 | ctx = get_ctx(repo, rev) |
|
195 | commit_id = mercurial.node.hex(ctx.node()) | |
|
196 | branch = ctx.branch() | |
|
214 | commit_id = ascii_str(mercurial.node.hex(ctx.node())) | |
|
215 | branch = safe_str(ctx.branch()) | |
|
197 | 216 | commits.append((commit_id, branch)) |
|
198 | 217 | |
|
199 | 218 | parent_heads = [] |
@@ -217,9 +236,9 b' def _check_heads(repo, start, end, commi' | |||
|
217 | 236 | for p in parents: |
|
218 | 237 | branch = get_ctx(repo, p).branch() |
|
219 | 238 | # The heads descending from that parent, on the same branch |
|
220 |
parent_heads = |
|
|
221 |
reachable = |
|
|
222 |
for x in |
|
|
239 | parent_heads = {p} | |
|
240 | reachable = {p} | |
|
241 | for x in range(p + 1, end): | |
|
223 | 242 | if get_ctx(repo, x).branch() != branch: |
|
224 | 243 | continue |
|
225 | 244 | for pp in changelog.parentrevs(x): |
@@ -295,14 +314,16 b' def pre_push(ui, repo, node=None, **kwar' | |||
|
295 | 314 | detect_force_push = extras.get('detect_force_push') |
|
296 | 315 | |
|
297 | 316 | rev_data = [] |
|
298 | if node and kwargs.get('hooktype') == 'pretxnchangegroup': | |
|
317 | hook_type: str = safe_str(kwargs.get('hooktype')) | |
|
318 | ||
|
319 | if node and hook_type == 'pretxnchangegroup': | |
|
299 | 320 | branches = collections.defaultdict(list) |
|
300 | 321 | commits, _heads = _rev_range_hash(repo, node, check_heads=detect_force_push) |
|
301 | 322 | for commit_id, branch in commits: |
|
302 | 323 | branches[branch].append(commit_id) |
|
303 | 324 | |
|
304 | 325 | for branch, commits in branches.items(): |
|
305 | old_rev = kwargs.get('node_last') or commits[0] | |
|
326 | old_rev = ascii_str(kwargs.get('node_last')) or commits[0] | |
|
306 | 327 | rev_data.append({ |
|
307 | 328 | 'total_commits': len(commits), |
|
308 | 329 | 'old_rev': old_rev, |
@@ -319,10 +340,10 b' def pre_push(ui, repo, node=None, **kwar' | |||
|
319 | 340 | extras.get('repo_store', ''), extras.get('repository', '')) |
|
320 | 341 | push_ref['hg_env'] = _get_hg_env( |
|
321 | 342 | old_rev=push_ref['old_rev'], |
|
322 | new_rev=push_ref['new_rev'], txnid=kwargs.get('txnid'), | |
|
343 | new_rev=push_ref['new_rev'], txnid=ascii_str(kwargs.get('txnid')), | |
|
323 | 344 | repo_path=repo_path) |
|
324 | 345 | |
|
325 |
extras['hook_type'] = |
|
|
346 | extras['hook_type'] = hook_type or 'pre_push' | |
|
326 | 347 | extras['commit_ids'] = rev_data |
|
327 | 348 | |
|
328 | 349 | return _call_hook('pre_push', extras, HgMessageWriter(ui)) |
@@ -363,6 +384,7 b' def post_push(ui, repo, node, **kwargs):' | |||
|
363 | 384 | branches = [] |
|
364 | 385 | bookmarks = [] |
|
365 | 386 | tags = [] |
|
387 | hook_type: str = safe_str(kwargs.get('hooktype')) | |
|
366 | 388 | |
|
367 | 389 | commits, _heads = _rev_range_hash(repo, node) |
|
368 | 390 | for commit_id, branch in commits: |
@@ -370,11 +392,12 b' def post_push(ui, repo, node, **kwargs):' | |||
|
370 | 392 | if branch not in branches: |
|
371 | 393 | branches.append(branch) |
|
372 | 394 | |
|
373 |
if hasattr(ui, '_rc_pushkey_b |
|
|
374 |
bookmarks = ui._rc_pushkey_b |
|
|
395 | if hasattr(ui, '_rc_pushkey_bookmarks'): | |
|
396 | bookmarks = ui._rc_pushkey_bookmarks | |
|
375 | 397 | |
|
376 |
extras['hook_type'] = |
|
|
398 | extras['hook_type'] = hook_type or 'post_push' | |
|
377 | 399 | extras['commit_ids'] = commit_ids |
|
400 | ||
|
378 | 401 | extras['new_refs'] = { |
|
379 | 402 | 'branches': branches, |
|
380 | 403 | 'bookmarks': bookmarks, |
@@ -395,9 +418,10 b' def post_push_ssh(ui, repo, node, **kwar' | |||
|
395 | 418 | |
|
396 | 419 | def key_push(ui, repo, **kwargs): |
|
397 | 420 | from vcsserver.hgcompat import get_ctx |
|
398 | if kwargs['new'] != '0' and kwargs['namespace'] == 'bookmarks': | |
|
421 | ||
|
422 | if kwargs['new'] != b'0' and kwargs['namespace'] == b'bookmarks': | |
|
399 | 423 | # store new bookmarks in our UI object propagated later to post_push |
|
400 |
ui._rc_pushkey_b |
|
|
424 | ui._rc_pushkey_bookmarks = get_ctx(repo, kwargs['key']).bookmarks() | |
|
401 | 425 | return |
|
402 | 426 | |
|
403 | 427 | |
@@ -426,10 +450,13 b' def handle_git_post_receive(unused_repo_' | |||
|
426 | 450 | pass |
|
427 | 451 | |
|
428 | 452 | |
|
429 | HookResponse = collections.namedtuple('HookResponse', ('status', 'output')) | |
|
453 | @dataclasses.dataclass | |
|
454 | class HookResponse: | |
|
455 | status: int | |
|
456 | output: str | |
|
430 | 457 | |
|
431 | 458 | |
|
432 | def git_pre_pull(extras): | |
|
459 | def git_pre_pull(extras) -> HookResponse: | |
|
433 | 460 | """ |
|
434 | 461 | Pre pull hook. |
|
435 | 462 | |
@@ -439,20 +466,23 b' def git_pre_pull(extras):' | |||
|
439 | 466 | :return: status code of the hook. 0 for success. |
|
440 | 467 | :rtype: int |
|
441 | 468 | """ |
|
469 | ||
|
442 | 470 | if 'pull' not in extras['hooks']: |
|
443 | 471 | return HookResponse(0, '') |
|
444 | 472 | |
|
445 |
stdout = io. |
|
|
473 | stdout = io.StringIO() | |
|
446 | 474 | try: |
|
447 | status = _call_hook('pre_pull', extras, GitMessageWriter(stdout)) | |
|
475 | status_code = _call_hook('pre_pull', extras, GitMessageWriter(stdout)) | |
|
476 | ||
|
448 | 477 | except Exception as error: |
|
449 | status = 128 | |
|
450 | stdout.write('ERROR: %s\n' % str(error)) | |
|
478 | log.exception('Failed to call pre_pull hook') | |
|
479 | status_code = 128 | |
|
480 | stdout.write(f'ERROR: {error}\n') | |
|
451 | 481 | |
|
452 | return HookResponse(status, stdout.getvalue()) | |
|
482 | return HookResponse(status_code, stdout.getvalue()) | |
|
453 | 483 | |
|
454 | 484 | |
|
455 | def git_post_pull(extras): | |
|
485 | def git_post_pull(extras) -> HookResponse: | |
|
456 | 486 | """ |
|
457 | 487 | Post pull hook. |
|
458 | 488 | |
@@ -465,12 +495,12 b' def git_post_pull(extras):' | |||
|
465 | 495 | if 'pull' not in extras['hooks']: |
|
466 | 496 | return HookResponse(0, '') |
|
467 | 497 | |
|
468 |
stdout = io. |
|
|
498 | stdout = io.StringIO() | |
|
469 | 499 | try: |
|
470 | 500 | status = _call_hook('post_pull', extras, GitMessageWriter(stdout)) |
|
471 | 501 | except Exception as error: |
|
472 | 502 | status = 128 |
|
473 |
stdout.write('ERROR: |
|
|
503 | stdout.write(f'ERROR: {error}\n') | |
|
474 | 504 | |
|
475 | 505 | return HookResponse(status, stdout.getvalue()) |
|
476 | 506 | |
@@ -495,15 +525,11 b' def _parse_git_ref_lines(revision_lines)' | |||
|
495 | 525 | return rev_data |
|
496 | 526 | |
|
497 | 527 | |
|
498 | def git_pre_receive(unused_repo_path, revision_lines, env): | |
|
528 | def git_pre_receive(unused_repo_path, revision_lines, env) -> int: | |
|
499 | 529 | """ |
|
500 | 530 | Pre push hook. |
|
501 | 531 | |
|
502 | :param extras: dictionary containing the keys defined in simplevcs | |
|
503 | :type extras: dict | |
|
504 | ||
|
505 | 532 | :return: status code of the hook. 0 for success. |
|
506 | :rtype: int | |
|
507 | 533 | """ |
|
508 | 534 | extras = json.loads(env['RC_SCM_DATA']) |
|
509 | 535 | rev_data = _parse_git_ref_lines(revision_lines) |
@@ -527,7 +553,7 b' def git_pre_receive(unused_repo_path, re' | |||
|
527 | 553 | if type_ == 'heads' and not (new_branch or delete_branch): |
|
528 | 554 | old_rev = push_ref['old_rev'] |
|
529 | 555 | new_rev = push_ref['new_rev'] |
|
530 |
cmd = [settings.GIT_EXECUTABLE, 'rev-list', old_rev, '^{}' |
|
|
556 | cmd = [settings.GIT_EXECUTABLE, 'rev-list', old_rev, f'^{new_rev}'] | |
|
531 | 557 | stdout, stderr = subprocessio.run_command( |
|
532 | 558 | cmd, env=os.environ.copy()) |
|
533 | 559 | # means we're having some non-reachable objects, this forced push was used |
@@ -536,18 +562,18 b' def git_pre_receive(unused_repo_path, re' | |||
|
536 | 562 | |
|
537 | 563 | extras['hook_type'] = 'pre_receive' |
|
538 | 564 | extras['commit_ids'] = rev_data |
|
539 | return _call_hook('pre_push', extras, GitMessageWriter()) | |
|
565 | ||
|
566 | stdout = sys.stdout | |
|
567 | status_code = _call_hook('pre_push', extras, GitMessageWriter(stdout)) | |
|
568 | ||
|
569 | return status_code | |
|
540 | 570 | |
|
541 | 571 | |
|
542 | def git_post_receive(unused_repo_path, revision_lines, env): | |
|
572 | def git_post_receive(unused_repo_path, revision_lines, env) -> int: | |
|
543 | 573 | """ |
|
544 | 574 | Post push hook. |
|
545 | 575 | |
|
546 | :param extras: dictionary containing the keys defined in simplevcs | |
|
547 | :type extras: dict | |
|
548 | ||
|
549 | 576 | :return: status code of the hook. 0 for success. |
|
550 | :rtype: int | |
|
551 | 577 | """ |
|
552 | 578 | extras = json.loads(env['RC_SCM_DATA']) |
|
553 | 579 | if 'push' not in extras['hooks']: |
@@ -567,26 +593,28 b' def git_post_receive(unused_repo_path, r' | |||
|
567 | 593 | type_ = push_ref['type'] |
|
568 | 594 | |
|
569 | 595 | if type_ == 'heads': |
|
596 | # starting new branch case | |
|
570 | 597 | if push_ref['old_rev'] == empty_commit_id: |
|
571 | # starting new branch case | |
|
572 | if push_ref['name'] not in branches: | |
|
573 | branches.append(push_ref['name']) | |
|
598 | push_ref_name = push_ref['name'] | |
|
599 | ||
|
600 | if push_ref_name not in branches: | |
|
601 | branches.append(push_ref_name) | |
|
574 | 602 | |
|
575 | # Fix up head revision if needed | |
|
576 | cmd = [settings.GIT_EXECUTABLE, 'show', 'HEAD'] | |
|
577 | try: | |
|
578 | subprocessio.run_command(cmd, env=os.environ.copy()) | |
|
579 |
except |
|
|
580 | cmd = [settings.GIT_EXECUTABLE, 'symbolic-ref', '"HEAD"', | |
|
581 | '"refs/heads/%s"' % push_ref['name']] | |
|
582 | print("Setting default branch to %s" % push_ref['name']) | |
|
583 | subprocessio.run_command(cmd, env=os.environ.copy()) | |
|
603 | need_head_set = '' | |
|
604 | with Repository(os.getcwd()) as repo: | |
|
605 | try: | |
|
606 | repo.head | |
|
607 | except pygit2.GitError: | |
|
608 | need_head_set = f'refs/heads/{push_ref_name}' | |
|
584 | 609 | |
|
585 | cmd = [settings.GIT_EXECUTABLE, 'for-each-ref', | |
|
586 | '--format=%(refname)', 'refs/heads/*'] | |
|
610 | if need_head_set: | |
|
611 | repo.set_head(need_head_set) | |
|
612 | print(f"Setting default branch to {push_ref_name}") | |
|
613 | ||
|
614 | cmd = [settings.GIT_EXECUTABLE, 'for-each-ref', '--format=%(refname)', 'refs/heads/*'] | |
|
587 | 615 | stdout, stderr = subprocessio.run_command( |
|
588 | 616 | cmd, env=os.environ.copy()) |
|
589 | heads = stdout | |
|
617 | heads = safe_str(stdout) | |
|
590 | 618 | heads = heads.replace(push_ref['ref'], '') |
|
591 | 619 | heads = ' '.join(head for head |
|
592 | 620 | in heads.splitlines() if head) or '.' |
@@ -595,9 +623,10 b' def git_post_receive(unused_repo_path, r' | |||
|
595 | 623 | '--not', heads] |
|
596 | 624 | stdout, stderr = subprocessio.run_command( |
|
597 | 625 | cmd, env=os.environ.copy()) |
|
598 | git_revs.extend(stdout.splitlines()) | |
|
626 | git_revs.extend(list(map(ascii_str, stdout.splitlines()))) | |
|
627 | ||
|
628 | # delete branch case | |
|
599 | 629 | elif push_ref['new_rev'] == empty_commit_id: |
|
600 | # delete branch case | |
|
601 | 630 | git_revs.append('delete_branch=>%s' % push_ref['name']) |
|
602 | 631 | else: |
|
603 | 632 | if push_ref['name'] not in branches: |
@@ -608,7 +637,25 b' def git_post_receive(unused_repo_path, r' | |||
|
608 | 637 | '--reverse', '--pretty=format:%H'] |
|
609 | 638 | stdout, stderr = subprocessio.run_command( |
|
610 | 639 | cmd, env=os.environ.copy()) |
|
611 | git_revs.extend(stdout.splitlines()) | |
|
640 | # we get bytes from stdout, we need str to be consistent | |
|
641 | log_revs = list(map(ascii_str, stdout.splitlines())) | |
|
642 | git_revs.extend(log_revs) | |
|
643 | ||
|
644 | # Pure pygit2 impl. but still 2-3x slower :/ | |
|
645 | # results = [] | |
|
646 | # | |
|
647 | # with Repository(os.getcwd()) as repo: | |
|
648 | # repo_new_rev = repo[push_ref['new_rev']] | |
|
649 | # repo_old_rev = repo[push_ref['old_rev']] | |
|
650 | # walker = repo.walk(repo_new_rev.id, pygit2.GIT_SORT_TOPOLOGICAL) | |
|
651 | # | |
|
652 | # for commit in walker: | |
|
653 | # if commit.id == repo_old_rev.id: | |
|
654 | # break | |
|
655 | # results.append(commit.id.hex) | |
|
656 | # # reverse the order, can't use GIT_SORT_REVERSE | |
|
657 | # log_revs = results[::-1] | |
|
658 | ||
|
612 | 659 | elif type_ == 'tags': |
|
613 | 660 | if push_ref['name'] not in tags: |
|
614 | 661 | tags.append(push_ref['name']) |
@@ -622,13 +669,16 b' def git_post_receive(unused_repo_path, r' | |||
|
622 | 669 | 'tags': tags, |
|
623 | 670 | } |
|
624 | 671 | |
|
672 | stdout = sys.stdout | |
|
673 | ||
|
625 | 674 | if 'repo_size' in extras['hooks']: |
|
626 | 675 | try: |
|
627 | _call_hook('repo_size', extras, GitMessageWriter()) | |
|
628 | except: | |
|
676 | _call_hook('repo_size', extras, GitMessageWriter(stdout)) | |
|
677 | except Exception: | |
|
629 | 678 | pass |
|
630 | 679 | |
|
631 |
|
|
|
680 | status_code = _call_hook('post_push', extras, GitMessageWriter(stdout)) | |
|
681 | return status_code | |
|
632 | 682 | |
|
633 | 683 | |
|
634 | 684 | def _get_extras_from_txn_id(path, txn_id): |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -15,9 +15,9 b'' | |||
|
15 | 15 | # along with this program; if not, write to the Free Software Foundation, |
|
16 | 16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
17 | 17 | |
|
18 | import io | |
|
18 | 19 | import os |
|
19 | 20 | import sys |
|
20 | import base64 | |
|
21 | 21 | import locale |
|
22 | 22 | import logging |
|
23 | 23 | import uuid |
@@ -28,17 +28,20 b' import tempfile' | |||
|
28 | 28 | import psutil |
|
29 | 29 | |
|
30 | 30 | from itertools import chain |
|
31 | from cStringIO import StringIO | |
|
32 | 31 | |
|
33 | import simplejson as json | |
|
34 | 32 | import msgpack |
|
33 | import configparser | |
|
34 | ||
|
35 | 35 | from pyramid.config import Configurator |
|
36 | 36 | from pyramid.wsgi import wsgiapp |
|
37 | from pyramid.compat import configparser | |
|
38 | 37 | from pyramid.response import Response |
|
38 | ||
|
39 | from vcsserver.base import BytesEnvelope, BinaryEnvelope | |
|
40 | from vcsserver.lib.rc_json import json | |
|
39 | 41 | from vcsserver.config.settings_maker import SettingsMaker |
|
40 | from vcsserver.utils import safe_int | |
|
42 | from vcsserver.str_utils import safe_int | |
|
41 | 43 | from vcsserver.lib.statsd_client import StatsdClient |
|
44 | from vcsserver.tweens.request_wrapper import get_call_context, get_headers_call_context | |
|
42 | 45 | |
|
43 | 46 | log = logging.getLogger(__name__) |
|
44 | 47 | |
@@ -62,23 +65,39 b' from vcsserver.exceptions import HTTPRep' | |||
|
62 | 65 | from vcsserver.lib.exc_tracking import store_exception |
|
63 | 66 | from vcsserver.server import VcsServer |
|
64 | 67 | |
|
68 | strict_vcs = True | |
|
69 | ||
|
70 | git_import_err = None | |
|
65 | 71 | try: |
|
66 | from vcsserver.git import GitFactory, GitRemote | |
|
67 | except ImportError: | |
|
72 | from vcsserver.remote.git import GitFactory, GitRemote | |
|
73 | except ImportError as e: | |
|
68 | 74 | GitFactory = None |
|
69 | 75 | GitRemote = None |
|
76 | git_import_err = e | |
|
77 | if strict_vcs: | |
|
78 | raise | |
|
70 | 79 | |
|
80 | ||
|
81 | hg_import_err = None | |
|
71 | 82 | try: |
|
72 | from vcsserver.hg import MercurialFactory, HgRemote | |
|
73 | except ImportError: | |
|
83 | from vcsserver.remote.hg import MercurialFactory, HgRemote | |
|
84 | except ImportError as e: | |
|
74 | 85 | MercurialFactory = None |
|
75 | 86 | HgRemote = None |
|
87 | hg_import_err = e | |
|
88 | if strict_vcs: | |
|
89 | raise | |
|
76 | 90 | |
|
91 | ||
|
92 | svn_import_err = None | |
|
77 | 93 | try: |
|
78 | from vcsserver.svn import SubversionFactory, SvnRemote | |
|
79 | except ImportError: | |
|
94 | from vcsserver.remote.svn import SubversionFactory, SvnRemote | |
|
95 | except ImportError as e: | |
|
80 | 96 | SubversionFactory = None |
|
81 | 97 | SvnRemote = None |
|
98 | svn_import_err = e | |
|
99 | if strict_vcs: | |
|
100 | raise | |
|
82 | 101 | |
|
83 | 102 | |
|
84 | 103 | def _is_request_chunked(environ): |
@@ -106,13 +125,13 b' class VCS(object):' | |||
|
106 | 125 | git_factory = GitFactory() |
|
107 | 126 | self._git_remote = GitRemote(git_factory) |
|
108 | 127 | else: |
|
109 |
log. |
|
|
128 | log.error("Git client import failed: %s", git_import_err) | |
|
110 | 129 | |
|
111 | 130 | if MercurialFactory and HgRemote: |
|
112 | 131 | hg_factory = MercurialFactory() |
|
113 | 132 | self._hg_remote = HgRemote(hg_factory) |
|
114 | 133 | else: |
|
115 |
log. |
|
|
134 | log.error("Mercurial client import failed: %s", hg_import_err) | |
|
116 | 135 | |
|
117 | 136 | if SubversionFactory and SvnRemote: |
|
118 | 137 | svn_factory = SubversionFactory() |
@@ -121,7 +140,7 b' class VCS(object):' | |||
|
121 | 140 | hg_factory = MercurialFactory() |
|
122 | 141 | self._svn_remote = SvnRemote(svn_factory, hg_factory=hg_factory) |
|
123 | 142 | else: |
|
124 |
log. |
|
|
143 | log.error("Subversion client import failed: %s", svn_import_err) | |
|
125 | 144 | |
|
126 | 145 | self._vcsserver = VcsServer() |
|
127 | 146 | |
@@ -129,8 +148,7 b' class VCS(object):' | |||
|
129 | 148 | if self.locale: |
|
130 | 149 | log.info('Settings locale: `LC_ALL` to %s', self.locale) |
|
131 | 150 | else: |
|
132 | log.info( | |
|
133 | 'Configuring locale subsystem based on environment variables') | |
|
151 | log.info('Configuring locale subsystem based on environment variables') | |
|
134 | 152 | try: |
|
135 | 153 | # If self.locale is the empty string, then the locale |
|
136 | 154 | # module will use the environment variables. See the |
@@ -142,8 +160,7 b' class VCS(object):' | |||
|
142 | 160 | 'Locale set to language code "%s" with encoding "%s".', |
|
143 | 161 | language_code, encoding) |
|
144 | 162 | except locale.Error: |
|
145 | log.exception( | |
|
146 | 'Cannot set locale, not configuring the locale system') | |
|
163 | log.exception('Cannot set locale, not configuring the locale system') | |
|
147 | 164 | |
|
148 | 165 | |
|
149 | 166 | class WsgiProxy(object): |
@@ -189,7 +206,7 b' class VCSViewPredicate(object):' | |||
|
189 | 206 | self.remotes = val |
|
190 | 207 | |
|
191 | 208 | def text(self): |
|
192 |
return 'vcs view method = |
|
|
209 | return f'vcs view method = {list(self.remotes.keys())}' | |
|
193 | 210 | |
|
194 | 211 | phash = text |
|
195 | 212 | |
@@ -213,9 +230,11 b' class HTTPApplication(object):' | |||
|
213 | 230 | self.config = Configurator(settings=settings) |
|
214 | 231 | # Init our statsd at very start |
|
215 | 232 | self.config.registry.statsd = StatsdClient.statsd |
|
233 | self.config.registry.vcs_call_context = {} | |
|
216 | 234 | |
|
217 | 235 | self.global_config = global_config |
|
218 | 236 | self.config.include('vcsserver.lib.rc_cache') |
|
237 | self.config.include('vcsserver.lib.rc_cache.archive_cache') | |
|
219 | 238 | |
|
220 | 239 | settings_locale = settings.get('locale', '') or 'en_US.UTF-8' |
|
221 | 240 | vcs = VCS(locale_conf=settings_locale, cache_config=settings) |
@@ -302,6 +321,7 b' class HTTPApplication(object):' | |||
|
302 | 321 | def _vcs_view_params(self, request): |
|
303 | 322 | remote = self._remotes[request.matchdict['backend']] |
|
304 | 323 | payload = msgpack.unpackb(request.body, use_list=True) |
|
324 | ||
|
305 | 325 | method = payload.get('method') |
|
306 | 326 | params = payload['params'] |
|
307 | 327 | wire = params.get('wire') |
@@ -309,6 +329,11 b' class HTTPApplication(object):' | |||
|
309 | 329 | kwargs = params.get('kwargs') |
|
310 | 330 | context_uid = None |
|
311 | 331 | |
|
332 | request.registry.vcs_call_context = { | |
|
333 | 'method': method, | |
|
334 | 'repo_name': payload.get('_repo_name'), | |
|
335 | } | |
|
336 | ||
|
312 | 337 | if wire: |
|
313 | 338 | try: |
|
314 | 339 | wire['context'] = context_uid = uuid.UUID(wire['context']) |
@@ -319,22 +344,33 b' class HTTPApplication(object):' | |||
|
319 | 344 | |
|
320 | 345 | # NOTE(marcink): trading complexity for slight performance |
|
321 | 346 | if log.isEnabledFor(logging.DEBUG): |
|
322 | no_args_methods = [ | |
|
323 | ||
|
324 | ] | |
|
325 | if method in no_args_methods: | |
|
347 | # also we SKIP printing out any of those methods args since they maybe excessive | |
|
348 | just_args_methods = { | |
|
349 | 'commitctx': ('content', 'removed', 'updated'), | |
|
350 | 'commit': ('content', 'removed', 'updated') | |
|
351 | } | |
|
352 | if method in just_args_methods: | |
|
353 | skip_args = just_args_methods[method] | |
|
326 | 354 | call_args = '' |
|
355 | call_kwargs = {} | |
|
356 | for k in kwargs: | |
|
357 | if k in skip_args: | |
|
358 | # replace our skip key with dummy | |
|
359 | call_kwargs[k] = f'RemovedParam({k})' | |
|
360 | else: | |
|
361 | call_kwargs[k] = kwargs[k] | |
|
327 | 362 | else: |
|
328 | 363 | call_args = args[1:] |
|
364 | call_kwargs = kwargs | |
|
329 | 365 | |
|
330 | 366 | log.debug('Method requested:`%s` with args:%s kwargs:%s context_uid: %s, repo_state_uid:%s', |
|
331 | method, call_args, kwargs, context_uid, repo_state_uid) | |
|
367 | method, call_args, call_kwargs, context_uid, repo_state_uid) | |
|
332 | 368 | |
|
333 | 369 | statsd = request.registry.statsd |
|
334 | 370 | if statsd: |
|
335 | 371 | statsd.incr( |
|
336 | 372 | 'vcsserver_method_total', tags=[ |
|
337 |
"method:{}" |
|
|
373 | f"method:{method}", | |
|
338 | 374 | ]) |
|
339 | 375 | return payload, remote, method, args, kwargs |
|
340 | 376 | |
@@ -384,7 +420,7 b' class HTTPApplication(object):' | |||
|
384 | 420 | resp = { |
|
385 | 421 | 'id': payload_id, |
|
386 | 422 | 'error': { |
|
387 |
'message': e |
|
|
423 | 'message': str(e), | |
|
388 | 424 | 'traceback': tb_info, |
|
389 | 425 | 'org_exc': org_exc_name, |
|
390 | 426 | 'org_exc_tb': org_exc_tb, |
@@ -401,7 +437,7 b' class HTTPApplication(object):' | |||
|
401 | 437 | 'id': payload_id, |
|
402 | 438 | 'result': resp |
|
403 | 439 | } |
|
404 | ||
|
440 | log.debug('Serving data for method %s', method) | |
|
405 | 441 | return resp |
|
406 | 442 | |
|
407 | 443 | def vcs_stream_view(self, request): |
@@ -416,7 +452,7 b' class HTTPApplication(object):' | |||
|
416 | 452 | raise |
|
417 | 453 | |
|
418 | 454 | def get_chunked_data(method_resp): |
|
419 |
stream = |
|
|
455 | stream = io.BytesIO(method_resp) | |
|
420 | 456 | while 1: |
|
421 | 457 | chunk = stream.read(chunk_size) |
|
422 | 458 | if not chunk: |
@@ -453,7 +489,7 b' class HTTPApplication(object):' | |||
|
453 | 489 | except Exception: |
|
454 | 490 | log.exception('Failed to read .ini file for display') |
|
455 | 491 | |
|
456 | environ = os.environ.items() | |
|
492 | environ = list(os.environ.items()) | |
|
457 | 493 | |
|
458 | 494 | resp = { |
|
459 | 495 | 'id': payload.get('id'), |
@@ -468,14 +504,28 b' class HTTPApplication(object):' | |||
|
468 | 504 | return resp |
|
469 | 505 | |
|
470 | 506 | def _msgpack_renderer_factory(self, info): |
|
507 | ||
|
471 | 508 | def _render(value, system): |
|
509 | bin_type = False | |
|
510 | res = value.get('result') | |
|
511 | if isinstance(res, BytesEnvelope): | |
|
512 | log.debug('Result is wrapped in BytesEnvelope type') | |
|
513 | bin_type = True | |
|
514 | elif isinstance(res, BinaryEnvelope): | |
|
515 | log.debug('Result is wrapped in BinaryEnvelope type') | |
|
516 | value['result'] = res.val | |
|
517 | bin_type = True | |
|
518 | ||
|
472 | 519 | request = system.get('request') |
|
473 | 520 | if request is not None: |
|
474 | 521 | response = request.response |
|
475 | 522 | ct = response.content_type |
|
476 | 523 | if ct == response.default_content_type: |
|
477 | 524 | response.content_type = 'application/x-msgpack' |
|
478 | return msgpack.packb(value) | |
|
525 | if bin_type: | |
|
526 | response.content_type = 'application/x-msgpack-bin' | |
|
527 | ||
|
528 | return msgpack.packb(value, use_bin_type=bin_type) | |
|
479 | 529 | return _render |
|
480 | 530 | |
|
481 | 531 | def set_env_from_config(self, environ, config): |
@@ -528,16 +578,17 b' class HTTPApplication(object):' | |||
|
528 | 578 | @wsgiapp |
|
529 | 579 | def _hg_stream(environ, start_response): |
|
530 | 580 | log.debug('http-app: handling hg stream') |
|
531 | repo_path = environ['HTTP_X_RC_REPO_PATH'] | |
|
532 | repo_name = environ['HTTP_X_RC_REPO_NAME'] | |
|
533 | packed_config = base64.b64decode( | |
|
534 | environ['HTTP_X_RC_REPO_CONFIG']) | |
|
535 |
config = |
|
|
581 | call_context = get_headers_call_context(environ) | |
|
582 | ||
|
583 | repo_path = call_context['repo_path'] | |
|
584 | repo_name = call_context['repo_name'] | |
|
585 | config = call_context['repo_config'] | |
|
586 | ||
|
536 | 587 | app = scm_app.create_hg_wsgi_app( |
|
537 | 588 | repo_path, repo_name, config) |
|
538 | 589 | |
|
539 | 590 | # Consistent path information for hgweb |
|
540 |
environ['PATH_INFO'] = |
|
|
591 | environ['PATH_INFO'] = call_context['path_info'] | |
|
541 | 592 | environ['REPO_NAME'] = repo_name |
|
542 | 593 | self.set_env_from_config(environ, config) |
|
543 | 594 | |
@@ -557,13 +608,14 b' class HTTPApplication(object):' | |||
|
557 | 608 | @wsgiapp |
|
558 | 609 | def _git_stream(environ, start_response): |
|
559 | 610 | log.debug('http-app: handling git stream') |
|
560 | repo_path = environ['HTTP_X_RC_REPO_PATH'] | |
|
561 | repo_name = environ['HTTP_X_RC_REPO_NAME'] | |
|
562 | packed_config = base64.b64decode( | |
|
563 | environ['HTTP_X_RC_REPO_CONFIG']) | |
|
564 | config = msgpack.unpackb(packed_config) | |
|
611 | ||
|
612 | call_context = get_headers_call_context(environ) | |
|
565 | 613 | |
|
566 | environ['PATH_INFO'] = environ['HTTP_X_RC_PATH_INFO'] | |
|
614 | repo_path = call_context['repo_path'] | |
|
615 | repo_name = call_context['repo_name'] | |
|
616 | config = call_context['repo_config'] | |
|
617 | ||
|
618 | environ['PATH_INFO'] = call_context['path_info'] | |
|
567 | 619 | self.set_env_from_config(environ, config) |
|
568 | 620 | |
|
569 | 621 | content_type = environ.get('CONTENT_TYPE', '') |
@@ -599,15 +651,18 b' class HTTPApplication(object):' | |||
|
599 | 651 | |
|
600 | 652 | def handle_vcs_exception(self, exception, request): |
|
601 | 653 | _vcs_kind = getattr(exception, '_vcs_kind', '') |
|
654 | ||
|
602 | 655 | if _vcs_kind == 'repo_locked': |
|
603 | # Get custom repo-locked status code if present. | |
|
604 |
status_code = |
|
|
656 | headers_call_context = get_headers_call_context(request.environ) | |
|
657 | status_code = safe_int(headers_call_context['locked_status_code']) | |
|
658 | ||
|
605 | 659 | return HTTPRepoLocked( |
|
606 |
title=exception |
|
|
660 | title=str(exception), status_code=status_code, headers=[('X-Rc-Locked', '1')]) | |
|
607 | 661 | |
|
608 | 662 | elif _vcs_kind == 'repo_branch_protected': |
|
609 | 663 | # Get custom repo-branch-protected status code if present. |
|
610 |
return HTTPRepoBranchProtected( |
|
|
664 | return HTTPRepoBranchProtected( | |
|
665 | title=str(exception), headers=[('X-Rc-Branch-Protection', '1')]) | |
|
611 | 666 | |
|
612 | 667 | exc_info = request.exc_info |
|
613 | 668 | store_exception(id(exc_info), exc_info) |
@@ -623,9 +678,9 b' class HTTPApplication(object):' | |||
|
623 | 678 | |
|
624 | 679 | statsd = request.registry.statsd |
|
625 | 680 | if statsd: |
|
626 |
exc_type = "{ |
|
|
681 | exc_type = f"{exception.__class__.__module__}.{exception.__class__.__name__}" | |
|
627 | 682 | statsd.incr('vcsserver_exception_total', |
|
628 |
tags=["type:{}" |
|
|
683 | tags=[f"type:{exc_type}"]) | |
|
629 | 684 | raise exception |
|
630 | 685 | |
|
631 | 686 | |
@@ -716,8 +771,7 b' def main(global_config, **settings):' | |||
|
716 | 771 | |
|
717 | 772 | pyramid_app = HTTPApplication(settings=settings, global_config=global_config).wsgi_app() |
|
718 | 773 | total_time = time.time() - start_time |
|
719 |
log.info('Pyramid app |
|
|
720 | getattr(pyramid_app, 'func_name', 'pyramid_app'), total_time) | |
|
774 | log.info('Pyramid app created and configured in %.2fs', total_time) | |
|
721 | 775 | return pyramid_app |
|
722 | 776 | |
|
723 | 777 |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -84,7 +84,7 b' class JsonEncoder(json.JSONEncoder):' | |||
|
84 | 84 | return str(obj) |
|
85 | 85 | |
|
86 | 86 | try: |
|
87 |
return super( |
|
|
87 | return super().default(obj) | |
|
88 | 88 | |
|
89 | 89 | except TypeError: |
|
90 | 90 | try: |
@@ -132,7 +132,7 b' class JsonFormatter(ExceptionAwareFormat' | |||
|
132 | 132 | self.json_ensure_ascii = kwargs.pop("json_ensure_ascii", True) |
|
133 | 133 | self.prefix = kwargs.pop("prefix", "") |
|
134 | 134 | reserved_attrs = kwargs.pop("reserved_attrs", RESERVED_ATTRS) |
|
135 | self.reserved_attrs = dict(zip(reserved_attrs, reserved_attrs)) | |
|
135 | self.reserved_attrs = dict(list(zip(reserved_attrs, reserved_attrs))) | |
|
136 | 136 | self.timestamp = kwargs.pop("timestamp", True) |
|
137 | 137 | |
|
138 | 138 | # super(JsonFormatter, self).__init__(*args, **kwargs) |
@@ -141,8 +141,8 b' class JsonFormatter(ExceptionAwareFormat' | |||
|
141 | 141 | self.json_encoder = JsonEncoder |
|
142 | 142 | |
|
143 | 143 | self._required_fields = self.parse() |
|
144 | self._skip_fields = dict(zip(self._required_fields, | |
|
145 | self._required_fields)) | |
|
144 | self._skip_fields = dict(list(zip(self._required_fields, | |
|
145 | self._required_fields))) | |
|
146 | 146 | self._skip_fields.update(self.reserved_attrs) |
|
147 | 147 | |
|
148 | 148 | def _str_to_fn(self, fn_as_str): |
@@ -200,7 +200,7 b' class JsonFormatter(ExceptionAwareFormat' | |||
|
200 | 200 | |
|
201 | 201 | def serialize_log_record(self, log_record): |
|
202 | 202 | """Returns the final representation of the log record.""" |
|
203 |
return " |
|
|
203 | return "{}{}".format(self.prefix, self.jsonify_log_record(log_record)) | |
|
204 | 204 | |
|
205 | 205 | def format(self, record): |
|
206 | 206 | """Formats a log record and serializes to json""" |
@@ -23,14 +23,8 b' loggers = {' | |||
|
23 | 23 | ] |
|
24 | 24 | } |
|
25 | 25 | |
|
26 | PY3 = sys.version_info[0] == 3 | |
|
27 | ||
|
28 | if PY3: | |
|
29 | text_type = str | |
|
30 | binary_type = bytes | |
|
31 | else: | |
|
32 | text_type = unicode # noqa | |
|
33 | binary_type = str | |
|
26 | text_type = str | |
|
27 | binary_type = bytes | |
|
34 | 28 | |
|
35 | 29 | |
|
36 | 30 | # Check if the id match. If not, return an error code. |
@@ -225,7 +219,7 b' class Lock(object):' | |||
|
225 | 219 | |
|
226 | 220 | if self._held: |
|
227 | 221 | owner_id = self.get_owner_id() |
|
228 |
raise AlreadyAcquired("Already acquired from this Lock instance. Lock id: {}" |
|
|
222 | raise AlreadyAcquired(f"Already acquired from this Lock instance. Lock id: {owner_id}") | |
|
229 | 223 | |
|
230 | 224 | if not blocking and timeout is not None: |
|
231 | 225 | raise TimeoutNotUsable("Timeout cannot be used if blocking=False") |
@@ -1,5 +1,3 b'' | |||
|
1 | from __future__ import absolute_import, division, unicode_literals | |
|
2 | ||
|
3 | 1 |
|
|
4 | 2 | |
|
5 | 3 | from .stream import TCPStatsClient, UnixSocketStatsClient # noqa |
@@ -1,5 +1,3 b'' | |||
|
1 | from __future__ import absolute_import, division, unicode_literals | |
|
2 | ||
|
3 | 1 |
|
|
4 | 2 | import random |
|
5 | 3 | from collections import deque |
@@ -49,7 +47,7 b' class StatsClientBase(object):' | |||
|
49 | 47 | statsd = StatsdClient.statsd |
|
50 | 48 | with statsd.timer('bucket_name', auto_send=True) as tmr: |
|
51 | 49 | # This block will be timed. |
|
52 |
for i in |
|
|
50 | for i in range(0, 100000): | |
|
53 | 51 | i ** 2 |
|
54 | 52 | # you can access time here... |
|
55 | 53 | elapsed_ms = tmr.ms |
@@ -1,5 +1,3 b'' | |||
|
1 | from __future__ import absolute_import, division, unicode_literals | |
|
2 | ||
|
3 | 1 |
|
|
4 | 2 | |
|
5 | 3 | from .base import StatsClientBase, PipelineBase |
@@ -1,14 +1,5 b'' | |||
|
1 | from __future__ import absolute_import, division, unicode_literals | |
|
2 | ||
|
3 | 1 |
|
|
4 | ||
|
5 | # Use timer that's not susceptible to time of day adjustments. | |
|
6 | try: | |
|
7 | # perf_counter is only present on Py3.3+ | |
|
8 | from time import perf_counter as time_now | |
|
9 | except ImportError: | |
|
10 | # fall back to using time | |
|
11 | from time import time as time_now | |
|
2 | from time import perf_counter as time_now | |
|
12 | 3 | |
|
13 | 4 | |
|
14 | 5 | def safe_wraps(wrapper, *args, **kwargs): |
@@ -1,5 +1,3 b'' | |||
|
1 | from __future__ import absolute_import, division, unicode_literals | |
|
2 | ||
|
3 | 1 |
|
|
4 | 2 | |
|
5 | 3 | from .base import StatsClientBase, PipelineBase |
@@ -8,7 +6,7 b' from .base import StatsClientBase, Pipel' | |||
|
8 | 6 | class Pipeline(PipelineBase): |
|
9 | 7 | |
|
10 | 8 | def __init__(self, client): |
|
11 |
super( |
|
|
9 | super().__init__(client) | |
|
12 | 10 | self._maxudpsize = client._maxudpsize |
|
13 | 11 | |
|
14 | 12 | def _send(self): |
@@ -1,7 +1,5 b'' | |||
|
1 | # -*- coding: utf-8 -*- | |
|
2 | ||
|
3 | 1 |
|
|
4 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
5 | 3 | # |
|
6 | 4 | # This program is free software; you can redistribute it and/or modify |
|
7 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -26,8 +24,6 b' import logging' | |||
|
26 | 24 | import traceback |
|
27 | 25 | import tempfile |
|
28 | 26 | |
|
29 | from pyramid import compat | |
|
30 | ||
|
31 | 27 | log = logging.getLogger(__name__) |
|
32 | 28 | |
|
33 | 29 | # NOTE: Any changes should be synced with exc_tracking at rhodecode.lib.exc_tracking |
@@ -77,7 +73,8 b' def _store_exception(exc_id, exc_info, p' | |||
|
77 | 73 | detailed_tb = getattr(exc_value, '_org_exc_tb', None) |
|
78 | 74 | |
|
79 | 75 | if detailed_tb: |
|
80 | if isinstance(detailed_tb, compat.string_types): | |
|
76 | remote_tb = detailed_tb | |
|
77 | if isinstance(detailed_tb, str): | |
|
81 | 78 | remote_tb = [detailed_tb] |
|
82 | 79 | |
|
83 | 80 | tb += ( |
@@ -127,10 +124,10 b' def store_exception(exc_id, exc_info, pr' | |||
|
127 | 124 | def _find_exc_file(exc_id, prefix=global_prefix): |
|
128 | 125 | exc_store_path = get_exc_store() |
|
129 | 126 | if prefix: |
|
130 |
exc_id = '{}_{}' |
|
|
127 | exc_id = f'{exc_id}_{prefix}' | |
|
131 | 128 | else: |
|
132 | 129 | # search without a prefix |
|
133 |
exc_id = '{}' |
|
|
130 | exc_id = f'{exc_id}' | |
|
134 | 131 | |
|
135 | 132 | # we need to search the store for such start pattern as above |
|
136 | 133 | for fname in os.listdir(exc_store_path): |
@@ -1,7 +1,5 b'' | |||
|
1 | # -*- coding: utf-8 -*- | |
|
2 | ||
|
3 | 1 |
|
|
4 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
5 | 3 | # |
|
6 | 4 | # This program is free software; you can redistribute it and/or modify |
|
7 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -22,7 +20,7 b' import logging' | |||
|
22 | 20 | |
|
23 | 21 | from repoze.lru import LRUCache |
|
24 | 22 | |
|
25 | from vcsserver.utils import safe_str | |
|
23 | from vcsserver.str_utils import safe_str | |
|
26 | 24 | |
|
27 | 25 | log = logging.getLogger(__name__) |
|
28 | 26 | |
@@ -45,7 +43,7 b' class LRUDict(LRUCache):' | |||
|
45 | 43 | del self.data[key] |
|
46 | 44 | |
|
47 | 45 | def keys(self): |
|
48 | return self.data.keys() | |
|
46 | return list(self.data.keys()) | |
|
49 | 47 | |
|
50 | 48 | |
|
51 | 49 | class LRUDictDebug(LRUDict): |
@@ -53,11 +51,11 b' class LRUDictDebug(LRUDict):' | |||
|
53 | 51 | Wrapper to provide some debug options |
|
54 | 52 | """ |
|
55 | 53 | def _report_keys(self): |
|
56 |
elems_cnt = ' |
|
|
54 | elems_cnt = f'{len(list(self.keys()))}/{self.size}' | |
|
57 | 55 | # trick for pformat print it more nicely |
|
58 | 56 | fmt = '\n' |
|
59 | 57 | for cnt, elem in enumerate(self.keys()): |
|
60 |
fmt += ' |
|
|
58 | fmt += f'{cnt+1} - {safe_str(elem)}\n' | |
|
61 | 59 | log.debug('current LRU keys (%s):%s', elems_cnt, fmt) |
|
62 | 60 | |
|
63 | 61 | def __getitem__(self, key): |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -16,31 +16,57 b'' | |||
|
16 | 16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
17 | 17 | |
|
18 | 18 | import logging |
|
19 | import threading | |
|
20 | ||
|
19 | 21 | from dogpile.cache import register_backend |
|
20 | 22 | |
|
23 | from . import region_meta | |
|
24 | from .utils import ( | |
|
25 | backend_key_generator, | |
|
26 | clear_cache_namespace, | |
|
27 | get_default_cache_settings, | |
|
28 | get_or_create_region, | |
|
29 | make_region, | |
|
30 | str2bool, | |
|
31 | ) | |
|
32 | ||
|
33 | module_name = 'vcsserver' | |
|
34 | ||
|
21 | 35 | register_backend( |
|
22 |
"dogpile.cache.rc.memory_lru", " |
|
|
36 | "dogpile.cache.rc.memory_lru", f"{module_name}.lib.rc_cache.backends", | |
|
23 | 37 | "LRUMemoryBackend") |
|
24 | 38 | |
|
25 | 39 | register_backend( |
|
26 |
"dogpile.cache.rc.file_namespace", " |
|
|
40 | "dogpile.cache.rc.file_namespace", f"{module_name}.lib.rc_cache.backends", | |
|
27 | 41 | "FileNamespaceBackend") |
|
28 | 42 | |
|
29 | 43 | register_backend( |
|
30 |
"dogpile.cache.rc.redis", " |
|
|
44 | "dogpile.cache.rc.redis", f"{module_name}.lib.rc_cache.backends", | |
|
31 | 45 | "RedisPickleBackend") |
|
32 | 46 | |
|
33 | 47 | register_backend( |
|
34 |
"dogpile.cache.rc.redis_msgpack", " |
|
|
48 | "dogpile.cache.rc.redis_msgpack", f"{module_name}.lib.rc_cache.backends", | |
|
35 | 49 | "RedisMsgPackBackend") |
|
36 | 50 | |
|
37 | 51 | |
|
38 | 52 | log = logging.getLogger(__name__) |
|
39 | 53 | |
|
40 | from . import region_meta | |
|
41 | from .utils import ( | |
|
42 | get_default_cache_settings, backend_key_generator, get_or_create_region, | |
|
43 | clear_cache_namespace, make_region) | |
|
54 | ||
|
55 | CLEAR_DELETE = 'delete' | |
|
56 | CLEAR_INVALIDATE = 'invalidate' | |
|
57 | ||
|
58 | ||
|
59 | def async_creation_runner(cache, somekey, creator, mutex): | |
|
60 | ||
|
61 | def runner(): | |
|
62 | try: | |
|
63 | value = creator() | |
|
64 | cache.set(somekey, value) | |
|
65 | finally: | |
|
66 | mutex.release() | |
|
67 | ||
|
68 | thread = threading.Thread(target=runner) | |
|
69 | thread.start() | |
|
44 | 70 | |
|
45 | 71 | |
|
46 | 72 | def configure_dogpile_cache(settings): |
@@ -62,13 +88,20 b' def configure_dogpile_cache(settings):' | |||
|
62 | 88 | |
|
63 | 89 | new_region = make_region( |
|
64 | 90 | name=namespace_name, |
|
65 | function_key_generator=None | |
|
91 | function_key_generator=None, | |
|
92 | async_creation_runner=None | |
|
66 | 93 | ) |
|
67 | 94 | |
|
68 |
new_region.configure_from_config(settings, 'rc_cache.{}.' |
|
|
95 | new_region.configure_from_config(settings, f'rc_cache.{namespace_name}.') | |
|
69 | 96 | new_region.function_key_generator = backend_key_generator(new_region.actual_backend) |
|
97 | ||
|
98 | async_creator = str2bool(settings.pop(f'rc_cache.{namespace_name}.async_creator', 'false')) | |
|
99 | if async_creator: | |
|
100 | log.debug('configuring region %s with async creator', new_region) | |
|
101 | new_region.async_creation_runner = async_creation_runner | |
|
102 | ||
|
70 | 103 | if log.isEnabledFor(logging.DEBUG): |
|
71 |
region_args = dict(backend=new_region.actual_backend |
|
|
104 | region_args = dict(backend=new_region.actual_backend, | |
|
72 | 105 | region_invalidator=new_region.region_invalidator.__class__) |
|
73 | 106 | log.debug('dogpile: registering a new region `%s` %s', namespace_name, region_args) |
|
74 | 107 |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -15,25 +15,31 b'' | |||
|
15 | 15 | # along with this program; if not, write to the Free Software Foundation, |
|
16 | 16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
17 | 17 | |
|
18 | import time | |
|
19 | 18 | import errno |
|
19 | import fcntl | |
|
20 | import functools | |
|
20 | 21 | import logging |
|
22 | import os | |
|
23 | import pickle | |
|
24 | #import time | |
|
21 | 25 | |
|
26 | #import gevent | |
|
22 | 27 | import msgpack |
|
23 | 28 | import redis |
|
24 | 29 | |
|
25 | from dogpile.cache.api import CachedValue | |
|
26 | from dogpile.cache.backends import memory as memory_backend | |
|
30 | flock_org = fcntl.flock | |
|
31 | from typing import Union | |
|
32 | ||
|
33 | from dogpile.cache.api import Deserializer, Serializer | |
|
27 | 34 | from dogpile.cache.backends import file as file_backend |
|
35 | from dogpile.cache.backends import memory as memory_backend | |
|
28 | 36 | from dogpile.cache.backends import redis as redis_backend |
|
29 |
from dogpile.cache.backends.file import |
|
|
37 | from dogpile.cache.backends.file import FileLock | |
|
30 | 38 | from dogpile.cache.util import memoized_property |
|
31 | 39 | |
|
32 | from pyramid.settings import asbool | |
|
33 | ||
|
34 | 40 | from vcsserver.lib.memory_lru_dict import LRUDict, LRUDictDebug |
|
35 |
from vcsserver.utils import safe_ |
|
|
36 | ||
|
41 | from vcsserver.str_utils import safe_bytes, safe_str | |
|
42 | from vcsserver.type_utils import str2bool | |
|
37 | 43 | |
|
38 | 44 | _default_max_size = 1024 |
|
39 | 45 | |
@@ -45,14 +51,20 b' class LRUMemoryBackend(memory_backend.Me' | |||
|
45 | 51 | pickle_values = False |
|
46 | 52 | |
|
47 | 53 | def __init__(self, arguments): |
|
48 | max_size = arguments.pop('max_size', _default_max_size) | |
|
54 | self.max_size = arguments.pop('max_size', _default_max_size) | |
|
49 | 55 | |
|
50 | 56 | LRUDictClass = LRUDict |
|
51 | 57 | if arguments.pop('log_key_count', None): |
|
52 | 58 | LRUDictClass = LRUDictDebug |
|
53 | 59 | |
|
54 | arguments['cache_dict'] = LRUDictClass(max_size) | |
|
55 |
super( |
|
|
60 | arguments['cache_dict'] = LRUDictClass(self.max_size) | |
|
61 | super().__init__(arguments) | |
|
62 | ||
|
63 | def __repr__(self): | |
|
64 | return f'{self.__class__}(maxsize=`{self.max_size}`)' | |
|
65 | ||
|
66 | def __str__(self): | |
|
67 | return self.__repr__() | |
|
56 | 68 | |
|
57 | 69 | def delete(self, key): |
|
58 | 70 | try: |
@@ -66,55 +78,22 b' class LRUMemoryBackend(memory_backend.Me' | |||
|
66 | 78 | self.delete(key) |
|
67 | 79 | |
|
68 | 80 | |
|
69 |
class PickleSerializer |
|
|
70 | ||
|
71 | def _dumps(self, value, safe=False): | |
|
72 | try: | |
|
73 | return compat.pickle.dumps(value) | |
|
74 | except Exception: | |
|
75 | if safe: | |
|
76 | return NO_VALUE | |
|
77 | else: | |
|
78 | raise | |
|
79 | ||
|
80 | def _loads(self, value, safe=True): | |
|
81 | try: | |
|
82 | return compat.pickle.loads(value) | |
|
83 | except Exception: | |
|
84 | if safe: | |
|
85 | return NO_VALUE | |
|
86 | else: | |
|
87 | raise | |
|
81 | class PickleSerializer: | |
|
82 | serializer: None | Serializer = staticmethod( # type: ignore | |
|
83 | functools.partial(pickle.dumps, protocol=pickle.HIGHEST_PROTOCOL) | |
|
84 | ) | |
|
85 | deserializer: None | Deserializer = staticmethod( # type: ignore | |
|
86 | functools.partial(pickle.loads) | |
|
87 | ) | |
|
88 | 88 | |
|
89 | 89 | |
|
90 | 90 | class MsgPackSerializer(object): |
|
91 | ||
|
92 | def _dumps(self, value, safe=False): | |
|
93 | try: | |
|
94 | return msgpack.packb(value) | |
|
95 | except Exception: | |
|
96 | if safe: | |
|
97 | return NO_VALUE | |
|
98 | else: | |
|
99 | raise | |
|
100 | ||
|
101 | def _loads(self, value, safe=True): | |
|
102 | """ | |
|
103 | pickle maintained the `CachedValue` wrapper of the tuple | |
|
104 | msgpack does not, so it must be added back in. | |
|
105 | """ | |
|
106 | try: | |
|
107 | value = msgpack.unpackb(value, use_list=False) | |
|
108 | return CachedValue(*value) | |
|
109 | except Exception: | |
|
110 | if safe: | |
|
111 | return NO_VALUE | |
|
112 | else: | |
|
113 | raise | |
|
114 | ||
|
115 | ||
|
116 | import fcntl | |
|
117 | flock_org = fcntl.flock | |
|
91 | serializer: None | Serializer = staticmethod( # type: ignore | |
|
92 | msgpack.packb | |
|
93 | ) | |
|
94 | deserializer: None | Deserializer = staticmethod( # type: ignore | |
|
95 | functools.partial(msgpack.unpackb, use_list=False) | |
|
96 | ) | |
|
118 | 97 | |
|
119 | 98 | |
|
120 | 99 | class CustomLockFactory(FileLock): |
@@ -129,30 +108,40 b' class FileNamespaceBackend(PickleSeriali' | |||
|
129 | 108 | arguments['lock_factory'] = CustomLockFactory |
|
130 | 109 | db_file = arguments.get('filename') |
|
131 | 110 | |
|
132 |
log.debug('initialing %s |
|
|
111 | log.debug('initialing cache-backend=%s db in %s', self.__class__.__name__, db_file) | |
|
112 | db_file_dir = os.path.dirname(db_file) | |
|
113 | if not os.path.isdir(db_file_dir): | |
|
114 | os.makedirs(db_file_dir) | |
|
115 | ||
|
133 | 116 | try: |
|
134 |
super( |
|
|
117 | super().__init__(arguments) | |
|
135 | 118 | except Exception: |
|
136 | 119 | log.exception('Failed to initialize db at: %s', db_file) |
|
137 | 120 | raise |
|
138 | 121 | |
|
139 | 122 | def __repr__(self): |
|
140 |
return '{ |
|
|
123 | return f'{self.__class__}(file=`{self.filename}`)' | |
|
124 | ||
|
125 | def __str__(self): | |
|
126 | return self.__repr__() | |
|
141 | 127 | |
|
142 |
def |
|
|
143 | prefix = '{}:{}'.format(self.key_prefix, prefix) | |
|
128 | def _get_keys_pattern(self, prefix: bytes = b''): | |
|
129 | return b'%b:%b' % (safe_bytes(self.key_prefix), safe_bytes(prefix)) | |
|
144 | 130 | |
|
145 | def cond(v): | |
|
131 | def list_keys(self, prefix: bytes = b''): | |
|
132 | prefix = self._get_keys_pattern(prefix) | |
|
133 | ||
|
134 | def cond(dbm_key: bytes): | |
|
146 | 135 | if not prefix: |
|
147 | 136 | return True |
|
148 | 137 | |
|
149 |
if |
|
|
138 | if dbm_key.startswith(prefix): | |
|
150 | 139 | return True |
|
151 | 140 | return False |
|
152 | 141 | |
|
153 | 142 | with self._dbm_file(True) as dbm: |
|
154 | 143 | try: |
|
155 | return filter(cond, dbm.keys()) | |
|
144 | return list(filter(cond, dbm.keys())) | |
|
156 | 145 | except Exception: |
|
157 | 146 | log.error('Failed to fetch DBM keys from DB: %s', self.get_store()) |
|
158 | 147 | raise |
@@ -160,49 +149,27 b' class FileNamespaceBackend(PickleSeriali' | |||
|
160 | 149 | def get_store(self): |
|
161 | 150 | return self.filename |
|
162 | 151 | |
|
163 | def _dbm_get(self, key): | |
|
164 | with self._dbm_file(False) as dbm: | |
|
165 | if hasattr(dbm, 'get'): | |
|
166 | value = dbm.get(key, NO_VALUE) | |
|
167 | else: | |
|
168 | # gdbm objects lack a .get method | |
|
169 | try: | |
|
170 | value = dbm[key] | |
|
171 | except KeyError: | |
|
172 | value = NO_VALUE | |
|
173 | if value is not NO_VALUE: | |
|
174 | value = self._loads(value) | |
|
175 | return value | |
|
176 | ||
|
177 | def get(self, key): | |
|
178 | try: | |
|
179 | return self._dbm_get(key) | |
|
180 | except Exception: | |
|
181 | log.error('Failed to fetch DBM key %s from DB: %s', key, self.get_store()) | |
|
182 | raise | |
|
183 | ||
|
184 | def set(self, key, value): | |
|
185 | with self._dbm_file(True) as dbm: | |
|
186 | dbm[key] = self._dumps(value) | |
|
187 | ||
|
188 | def set_multi(self, mapping): | |
|
189 | with self._dbm_file(True) as dbm: | |
|
190 | for key, value in mapping.items(): | |
|
191 | dbm[key] = self._dumps(value) | |
|
192 | ||
|
193 | 152 | |
|
194 | 153 | class BaseRedisBackend(redis_backend.RedisBackend): |
|
195 | 154 | key_prefix = '' |
|
196 | 155 | |
|
197 | 156 | def __init__(self, arguments): |
|
198 | super(BaseRedisBackend, self).__init__(arguments) | |
|
157 | self.db_conn = arguments.get('host', '') or arguments.get('url', '') or 'redis-host' | |
|
158 | super().__init__(arguments) | |
|
159 | ||
|
199 | 160 | self._lock_timeout = self.lock_timeout |
|
200 |
self._lock_auto_renewal = |
|
|
161 | self._lock_auto_renewal = str2bool(arguments.pop("lock_auto_renewal", True)) | |
|
201 | 162 | |
|
202 | 163 | if self._lock_auto_renewal and not self._lock_timeout: |
|
203 | 164 | # set default timeout for auto_renewal |
|
204 | 165 | self._lock_timeout = 30 |
|
205 | 166 | |
|
167 | def __repr__(self): | |
|
168 | return f'{self.__class__}(conn=`{self.db_conn}`)' | |
|
169 | ||
|
170 | def __str__(self): | |
|
171 | return self.__repr__() | |
|
172 | ||
|
206 | 173 | def _create_client(self): |
|
207 | 174 | args = {} |
|
208 | 175 | |
@@ -216,58 +183,29 b' class BaseRedisBackend(redis_backend.Red' | |||
|
216 | 183 | ) |
|
217 | 184 | |
|
218 | 185 | connection_pool = redis.ConnectionPool(**args) |
|
219 | ||
|
220 |
|
|
|
186 | self.writer_client = redis.StrictRedis( | |
|
187 | connection_pool=connection_pool | |
|
188 | ) | |
|
189 | self.reader_client = self.writer_client | |
|
221 | 190 | |
|
222 |
def |
|
|
223 |
|
|
|
224 | return self.client.keys(prefix) | |
|
191 | def _get_keys_pattern(self, prefix: bytes = b''): | |
|
192 | return b'%b:%b*' % (safe_bytes(self.key_prefix), safe_bytes(prefix)) | |
|
193 | ||
|
194 | def list_keys(self, prefix: bytes = b''): | |
|
195 | prefix = self._get_keys_pattern(prefix) | |
|
196 | return self.reader_client.keys(prefix) | |
|
225 | 197 | |
|
226 | 198 | def get_store(self): |
|
227 | return self.client.connection_pool | |
|
228 | ||
|
229 | def get(self, key): | |
|
230 | value = self.client.get(key) | |
|
231 | if value is None: | |
|
232 | return NO_VALUE | |
|
233 | return self._loads(value) | |
|
234 | ||
|
235 | def get_multi(self, keys): | |
|
236 | if not keys: | |
|
237 | return [] | |
|
238 | values = self.client.mget(keys) | |
|
239 | loads = self._loads | |
|
240 | return [ | |
|
241 | loads(v) if v is not None else NO_VALUE | |
|
242 | for v in values] | |
|
243 | ||
|
244 | def set(self, key, value): | |
|
245 | if self.redis_expiration_time: | |
|
246 | self.client.setex(key, self.redis_expiration_time, | |
|
247 | self._dumps(value)) | |
|
248 | else: | |
|
249 | self.client.set(key, self._dumps(value)) | |
|
250 | ||
|
251 | def set_multi(self, mapping): | |
|
252 | dumps = self._dumps | |
|
253 | mapping = dict( | |
|
254 | (k, dumps(v)) | |
|
255 | for k, v in mapping.items() | |
|
256 | ) | |
|
257 | ||
|
258 | if not self.redis_expiration_time: | |
|
259 | self.client.mset(mapping) | |
|
260 | else: | |
|
261 | pipe = self.client.pipeline() | |
|
262 | for key, value in mapping.items(): | |
|
263 | pipe.setex(key, self.redis_expiration_time, value) | |
|
264 | pipe.execute() | |
|
199 | return self.reader_client.connection_pool | |
|
265 | 200 | |
|
266 | 201 | def get_mutex(self, key): |
|
267 | 202 | if self.distributed_lock: |
|
268 |
lock_key = |
|
|
269 |
return get_mutex_lock( |
|
|
270 | auto_renewal=self._lock_auto_renewal) | |
|
203 | lock_key = f'_lock_{safe_str(key)}' | |
|
204 | return get_mutex_lock( | |
|
205 | self.writer_client, lock_key, | |
|
206 | self._lock_timeout, | |
|
207 | auto_renewal=self._lock_auto_renewal | |
|
208 | ) | |
|
271 | 209 | else: |
|
272 | 210 | return None |
|
273 | 211 | |
@@ -283,7 +221,7 b' class RedisMsgPackBackend(MsgPackSeriali' | |||
|
283 | 221 | |
|
284 | 222 | |
|
285 | 223 | def get_mutex_lock(client, lock_key, lock_timeout, auto_renewal=False): |
|
286 | import redis_lock | |
|
224 | from vcsserver.lib._vendor import redis_lock | |
|
287 | 225 | |
|
288 | 226 | class _RedisLockWrapper(object): |
|
289 | 227 | """LockWrapper for redis_lock""" |
@@ -299,10 +237,10 b' def get_mutex_lock(client, lock_key, loc' | |||
|
299 | 237 | ) |
|
300 | 238 | |
|
301 | 239 | def __repr__(self): |
|
302 |
return "{ |
|
|
240 | return f"{self.__class__.__name__}:{lock_key}" | |
|
303 | 241 | |
|
304 | 242 | def __str__(self): |
|
305 |
return "{ |
|
|
243 | return f"{self.__class__.__name__}:{lock_key}" | |
|
306 | 244 | |
|
307 | 245 | def __init__(self): |
|
308 | 246 | self.lock = self.get_lock() |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -15,115 +15,69 b'' | |||
|
15 | 15 | # along with this program; if not, write to the Free Software Foundation, |
|
16 | 16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
17 | 17 | |
|
18 | import os | |
|
19 | import time | |
|
18 | import functools | |
|
20 | 19 | import logging |
|
21 |
import |
|
|
20 | import os | |
|
21 | import threading | |
|
22 | import time | |
|
22 | 23 | |
|
24 | import decorator | |
|
23 | 25 | from dogpile.cache import CacheRegion |
|
24 | from dogpile.cache.util import compat | |
|
26 | ||
|
25 | 27 | |
|
26 |
from vcsserver.utils import |
|
|
28 | from vcsserver.utils import sha1 | |
|
29 | from vcsserver.str_utils import safe_bytes | |
|
30 | from vcsserver.type_utils import str2bool | |
|
27 | 31 | |
|
28 |
from |
|
|
32 | from . import region_meta | |
|
29 | 33 | |
|
30 | 34 | log = logging.getLogger(__name__) |
|
31 | 35 | |
|
32 | 36 | |
|
33 | 37 | class RhodeCodeCacheRegion(CacheRegion): |
|
34 | 38 | |
|
39 | def __repr__(self): | |
|
40 | return f'{self.__class__}(name={self.name})' | |
|
41 | ||
|
35 | 42 | def conditional_cache_on_arguments( |
|
36 | 43 | self, namespace=None, |
|
37 | 44 | expiration_time=None, |
|
38 | 45 | should_cache_fn=None, |
|
39 |
to_str= |
|
|
46 | to_str=str, | |
|
40 | 47 | function_key_generator=None, |
|
41 | 48 | condition=True): |
|
42 | 49 | """ |
|
43 | 50 | Custom conditional decorator, that will not touch any dogpile internals if |
|
44 |
condition isn't meet. This works a bit different |
|
|
51 | condition isn't meet. This works a bit different from should_cache_fn | |
|
45 | 52 | And it's faster in cases we don't ever want to compute cached values |
|
46 | 53 | """ |
|
47 |
expiration_time_is_callable = |
|
|
54 | expiration_time_is_callable = callable(expiration_time) | |
|
55 | if not namespace: | |
|
56 | namespace = getattr(self, '_default_namespace', None) | |
|
48 | 57 | |
|
49 | 58 | if function_key_generator is None: |
|
50 | 59 | function_key_generator = self.function_key_generator |
|
51 | 60 | |
|
52 | # workaround for py2 and cython problems, this block should be removed | |
|
53 | # once we've migrated to py3 | |
|
54 | if 'cython' == 'cython': | |
|
55 | def decorator(fn): | |
|
56 | if to_str is compat.string_type: | |
|
57 | # backwards compatible | |
|
58 | key_generator = function_key_generator(namespace, fn) | |
|
59 | else: | |
|
60 | key_generator = function_key_generator(namespace, fn, to_str=to_str) | |
|
61 | ||
|
62 | @functools.wraps(fn) | |
|
63 | def decorate(*arg, **kw): | |
|
64 | key = key_generator(*arg, **kw) | |
|
65 | ||
|
66 | @functools.wraps(fn) | |
|
67 | def creator(): | |
|
68 | return fn(*arg, **kw) | |
|
69 | ||
|
70 | if not condition: | |
|
71 | return creator() | |
|
72 | ||
|
73 | timeout = expiration_time() if expiration_time_is_callable \ | |
|
74 | else expiration_time | |
|
75 | ||
|
76 | return self.get_or_create(key, creator, timeout, should_cache_fn) | |
|
77 | ||
|
78 | def invalidate(*arg, **kw): | |
|
79 | key = key_generator(*arg, **kw) | |
|
80 | self.delete(key) | |
|
81 | ||
|
82 | def set_(value, *arg, **kw): | |
|
83 | key = key_generator(*arg, **kw) | |
|
84 | self.set(key, value) | |
|
85 | ||
|
86 | def get(*arg, **kw): | |
|
87 | key = key_generator(*arg, **kw) | |
|
88 | return self.get(key) | |
|
89 | ||
|
90 | def refresh(*arg, **kw): | |
|
91 | key = key_generator(*arg, **kw) | |
|
92 | value = fn(*arg, **kw) | |
|
93 | self.set(key, value) | |
|
94 | return value | |
|
95 | ||
|
96 | decorate.set = set_ | |
|
97 | decorate.invalidate = invalidate | |
|
98 | decorate.refresh = refresh | |
|
99 | decorate.get = get | |
|
100 | decorate.original = fn | |
|
101 | decorate.key_generator = key_generator | |
|
102 | decorate.__wrapped__ = fn | |
|
103 | ||
|
104 | return decorate | |
|
105 | return decorator | |
|
106 | ||
|
107 | def get_or_create_for_user_func(key_generator, user_func, *arg, **kw): | |
|
61 | def get_or_create_for_user_func(func_key_generator, user_func, *arg, **kw): | |
|
108 | 62 | |
|
109 | 63 | if not condition: |
|
110 |
log.debug('Calling un-cached method:%s', user_func. |
|
|
64 | log.debug('Calling un-cached method:%s', user_func.__name__) | |
|
111 | 65 | start = time.time() |
|
112 | 66 | result = user_func(*arg, **kw) |
|
113 | 67 | total = time.time() - start |
|
114 |
log.debug('un-cached method:%s took %.4fs', user_func. |
|
|
68 | log.debug('un-cached method:%s took %.4fs', user_func.__name__, total) | |
|
115 | 69 | return result |
|
116 | 70 | |
|
117 | key = key_generator(*arg, **kw) | |
|
71 | key = func_key_generator(*arg, **kw) | |
|
118 | 72 | |
|
119 | 73 | timeout = expiration_time() if expiration_time_is_callable \ |
|
120 | 74 | else expiration_time |
|
121 | 75 | |
|
122 |
log.debug('Calling cached method:`%s`', user_func. |
|
|
76 | log.debug('Calling cached method:`%s`', user_func.__name__) | |
|
123 | 77 | return self.get_or_create(key, user_func, timeout, should_cache_fn, (arg, kw)) |
|
124 | 78 | |
|
125 | 79 | def cache_decorator(user_func): |
|
126 |
if to_str is |
|
|
80 | if to_str is str: | |
|
127 | 81 | # backwards compatible |
|
128 | 82 | key_generator = function_key_generator(namespace, user_func) |
|
129 | 83 | else: |
@@ -176,7 +130,7 b' def get_default_cache_settings(settings,' | |||
|
176 | 130 | if key.startswith(prefix): |
|
177 | 131 | name = key.split(prefix)[1].strip() |
|
178 | 132 | val = settings[key] |
|
179 |
if isinstance(val, |
|
|
133 | if isinstance(val, str): | |
|
180 | 134 | val = val.strip() |
|
181 | 135 | cache_settings[name] = val |
|
182 | 136 | return cache_settings |
@@ -186,7 +140,21 b' def compute_key_from_params(*args):' | |||
|
186 | 140 | """ |
|
187 | 141 | Helper to compute key from given params to be used in cache manager |
|
188 | 142 | """ |
|
189 |
return sha1("_".join(map( |
|
|
143 | return sha1(safe_bytes("_".join(map(str, args)))) | |
|
144 | ||
|
145 | ||
|
146 | def custom_key_generator(backend, namespace, fn): | |
|
147 | func_name = fn.__name__ | |
|
148 | ||
|
149 | def generate_key(*args): | |
|
150 | backend_pref = getattr(backend, 'key_prefix', None) or 'backend_prefix' | |
|
151 | namespace_pref = namespace or 'default_namespace' | |
|
152 | arg_key = compute_key_from_params(*args) | |
|
153 | final_key = f"{backend_pref}:{namespace_pref}:{func_name}_{arg_key}" | |
|
154 | ||
|
155 | return final_key | |
|
156 | ||
|
157 | return generate_key | |
|
190 | 158 | |
|
191 | 159 | |
|
192 | 160 | def backend_key_generator(backend): |
@@ -194,49 +162,50 b' def backend_key_generator(backend):' | |||
|
194 | 162 | Special wrapper that also sends over the backend to the key generator |
|
195 | 163 | """ |
|
196 | 164 | def wrapper(namespace, fn): |
|
197 | return key_generator(backend, namespace, fn) | |
|
165 | return custom_key_generator(backend, namespace, fn) | |
|
198 | 166 | return wrapper |
|
199 | 167 | |
|
200 | 168 | |
|
201 | def key_generator(backend, namespace, fn): | |
|
202 | fname = fn.__name__ | |
|
169 | def get_or_create_region(region_name, region_namespace: str = None, use_async_runner=False): | |
|
170 | from .backends import FileNamespaceBackend | |
|
171 | from . import async_creation_runner | |
|
203 | 172 | |
|
204 | def generate_key(*args): | |
|
205 | backend_prefix = getattr(backend, 'key_prefix', None) or 'backend_prefix' | |
|
206 | namespace_pref = namespace or 'default_namespace' | |
|
207 | arg_key = compute_key_from_params(*args) | |
|
208 | final_key = "{}:{}:{}_{}".format(backend_prefix, namespace_pref, fname, arg_key) | |
|
209 | ||
|
210 | return final_key | |
|
211 | ||
|
212 | return generate_key | |
|
213 | ||
|
214 | ||
|
215 | def get_or_create_region(region_name, region_namespace=None): | |
|
216 | from vcsserver.lib.rc_cache.backends import FileNamespaceBackend | |
|
217 | 173 | region_obj = region_meta.dogpile_cache_regions.get(region_name) |
|
218 | 174 | if not region_obj: |
|
219 | raise EnvironmentError( | |
|
220 |
|
|
|
221 | region_name, region_meta.dogpile_cache_regions.keys())) | |
|
175 | reg_keys = list(region_meta.dogpile_cache_regions.keys()) | |
|
176 | raise OSError(f'Region `{region_name}` not in configured: {reg_keys}.') | |
|
177 | ||
|
178 | region_uid_name = f'{region_name}:{region_namespace}' | |
|
222 | 179 | |
|
223 | region_uid_name = '{}:{}'.format(region_name, region_namespace) | |
|
224 | 180 | if isinstance(region_obj.actual_backend, FileNamespaceBackend): |
|
181 | if not region_namespace: | |
|
182 | raise ValueError(f'{FileNamespaceBackend} used requires to specify region_namespace param') | |
|
183 | ||
|
225 | 184 | region_exist = region_meta.dogpile_cache_regions.get(region_namespace) |
|
226 | 185 | if region_exist: |
|
227 | 186 | log.debug('Using already configured region: %s', region_namespace) |
|
228 | 187 | return region_exist |
|
229 | cache_dir = region_meta.dogpile_config_defaults['cache_dir'] | |
|
188 | ||
|
230 | 189 | expiration_time = region_obj.expiration_time |
|
231 | 190 | |
|
232 | if not os.path.isdir(cache_dir): | |
|
233 | os.makedirs(cache_dir) | |
|
191 | cache_dir = region_meta.dogpile_config_defaults['cache_dir'] | |
|
192 | namespace_cache_dir = cache_dir | |
|
193 | ||
|
194 | # we default the namespace_cache_dir to our default cache dir. | |
|
195 | # however if this backend is configured with filename= param, we prioritize that | |
|
196 | # so all caches within that particular region, even those namespaced end up in the same path | |
|
197 | if region_obj.actual_backend.filename: | |
|
198 | namespace_cache_dir = os.path.dirname(region_obj.actual_backend.filename) | |
|
199 | ||
|
200 | if not os.path.isdir(namespace_cache_dir): | |
|
201 | os.makedirs(namespace_cache_dir) | |
|
234 | 202 | new_region = make_region( |
|
235 | 203 | name=region_uid_name, |
|
236 | 204 | function_key_generator=backend_key_generator(region_obj.actual_backend) |
|
237 | 205 | ) |
|
206 | ||
|
238 | 207 | namespace_filename = os.path.join( |
|
239 |
cache_dir, "{}.cache |
|
|
208 | namespace_cache_dir, f"{region_name}_{region_namespace}.cache_db") | |
|
240 | 209 | # special type that allows 1db per namespace |
|
241 | 210 | new_region.configure( |
|
242 | 211 | backend='dogpile.cache.rc.file_namespace', |
@@ -248,16 +217,31 b' def get_or_create_region(region_name, re' | |||
|
248 | 217 | log.debug('configuring new region: %s', region_uid_name) |
|
249 | 218 | region_obj = region_meta.dogpile_cache_regions[region_namespace] = new_region |
|
250 | 219 | |
|
220 | region_obj._default_namespace = region_namespace | |
|
221 | if use_async_runner: | |
|
222 | region_obj.async_creation_runner = async_creation_runner | |
|
251 | 223 | return region_obj |
|
252 | 224 | |
|
253 | 225 | |
|
254 |
def clear_cache_namespace(cache_region, cache_namespace_uid, |
|
|
255 | region = get_or_create_region(cache_region, cache_namespace_uid) | |
|
256 | cache_keys = region.backend.list_keys(prefix=cache_namespace_uid) | |
|
257 | num_delete_keys = len(cache_keys) | |
|
258 | if invalidate: | |
|
259 | region.invalidate(hard=False) | |
|
260 | else: | |
|
261 | if num_delete_keys: | |
|
262 | region.delete_multi(cache_keys) | |
|
263 | return num_delete_keys | |
|
226 | def clear_cache_namespace(cache_region: str | RhodeCodeCacheRegion, cache_namespace_uid: str, method: str): | |
|
227 | from . import CLEAR_DELETE, CLEAR_INVALIDATE | |
|
228 | ||
|
229 | if not isinstance(cache_region, RhodeCodeCacheRegion): | |
|
230 | cache_region = get_or_create_region(cache_region, cache_namespace_uid) | |
|
231 | log.debug('clearing cache region: %s with method=%s', cache_region, method) | |
|
232 | ||
|
233 | num_affected_keys = None | |
|
234 | ||
|
235 | if method == CLEAR_INVALIDATE: | |
|
236 | # NOTE: The CacheRegion.invalidate() method’s default mode of | |
|
237 | # operation is to set a timestamp local to this CacheRegion in this Python process only. | |
|
238 | # It does not impact other Python processes or regions as the timestamp is only stored locally in memory. | |
|
239 | cache_region.invalidate(hard=True) | |
|
240 | ||
|
241 | if method == CLEAR_DELETE: | |
|
242 | cache_keys = cache_region.backend.list_keys(prefix=cache_namespace_uid) | |
|
243 | num_affected_keys = len(cache_keys) | |
|
244 | if num_affected_keys: | |
|
245 | cache_region.delete_multi(cache_keys) | |
|
246 | ||
|
247 | return num_affected_keys |
@@ -1,7 +1,5 b'' | |||
|
1 | # -*- coding: utf-8 -*- | |
|
2 | ||
|
3 | 1 |
|
|
4 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
5 | 3 | # |
|
6 | 4 | # This program is free software; you can redistribute it and/or modify |
|
7 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -1,3 +1,20 b'' | |||
|
1 | # RhodeCode VCSServer provides access to different vcs backends via network. | |
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | # | |
|
4 | # This program is free software; you can redistribute it and/or modify | |
|
5 | # it under the terms of the GNU General Public License as published by | |
|
6 | # the Free Software Foundation; either version 3 of the License, or | |
|
7 | # (at your option) any later version. | |
|
8 | # | |
|
9 | # This program is distributed in the hope that it will be useful, | |
|
10 | # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
|
11 | # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
|
12 | # GNU General Public License for more details. | |
|
13 | # | |
|
14 | # You should have received a copy of the GNU General Public License | |
|
15 | # along with this program; if not, write to the Free Software Foundation, | |
|
16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA | |
|
17 | ||
|
1 | 18 | from vcsserver.lib._vendor.statsd import client_from_config |
|
2 | 19 | |
|
3 | 20 | |
@@ -12,7 +29,7 b' class _Singleton(type):' | |||
|
12 | 29 | |
|
13 | 30 | def __call__(cls, *args, **kwargs): |
|
14 | 31 | if cls not in cls._instances: |
|
15 |
cls._instances[cls] = super( |
|
|
32 | cls._instances[cls] = super().__call__(*args, **kwargs) | |
|
16 | 33 | return cls._instances[cls] |
|
17 | 34 | |
|
18 | 35 | |
@@ -24,22 +41,26 b' class StatsdClientClass(Singleton):' | |||
|
24 | 41 | setup_run = False |
|
25 | 42 | statsd_client = None |
|
26 | 43 | statsd = None |
|
44 | strict_mode_init = False | |
|
27 | 45 | |
|
28 | 46 | def __getattribute__(self, name): |
|
29 | 47 | |
|
30 | 48 | if name.startswith("statsd"): |
|
31 | 49 | if self.setup_run: |
|
32 |
return super( |
|
|
50 | return super().__getattribute__(name) | |
|
33 | 51 | else: |
|
52 | if self.strict_mode_init: | |
|
53 | raise StatsdClientNotInitialised(f"requested key was {name}") | |
|
34 | 54 | return None |
|
35 | #raise StatsdClientNotInitialised("requested key was %s" % name) | |
|
36 | 55 | |
|
37 |
return super( |
|
|
56 | return super().__getattribute__(name) | |
|
38 | 57 | |
|
39 | 58 | def setup(self, settings): |
|
40 | 59 | """ |
|
41 | 60 | Initialize the client |
|
42 | 61 | """ |
|
62 | strict_init_mode = settings.pop('statsd_strict_init', False) | |
|
63 | ||
|
43 | 64 | statsd = client_from_config(settings) |
|
44 | 65 | self.statsd = statsd |
|
45 | 66 | self.statsd_client = statsd |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -21,11 +21,13 b' import os' | |||
|
21 | 21 | import socket |
|
22 | 22 | import logging |
|
23 | 23 | |
|
24 | import simplejson as json | |
|
25 | 24 | import dulwich.protocol |
|
25 | from dulwich.protocol import CAPABILITY_SIDE_BAND, CAPABILITY_SIDE_BAND_64K | |
|
26 | 26 | from webob import Request, Response, exc |
|
27 | 27 | |
|
28 | from vcsserver.lib.rc_json import json | |
|
28 | 29 | from vcsserver import hooks, subprocessio |
|
30 | from vcsserver.str_utils import ascii_bytes | |
|
29 | 31 | |
|
30 | 32 | |
|
31 | 33 | log = logging.getLogger(__name__) |
@@ -54,7 +56,7 b' class FileWrapper(object):' | |||
|
54 | 56 | return data |
|
55 | 57 | |
|
56 | 58 | def __repr__(self): |
|
57 |
return '<FileWrapper |
|
|
59 | return '<FileWrapper {} len: {}, read: {}>'.format( | |
|
58 | 60 | self.fd, self.content_length, self.content_length - self.remain |
|
59 | 61 | ) |
|
60 | 62 | |
@@ -62,26 +64,25 b' class FileWrapper(object):' | |||
|
62 | 64 | class GitRepository(object): |
|
63 | 65 | """WSGI app for handling Git smart protocol endpoints.""" |
|
64 | 66 | |
|
65 | git_folder_signature = frozenset( | |
|
66 | ('config', 'head', 'info', 'objects', 'refs')) | |
|
67 | git_folder_signature = frozenset(('config', 'head', 'info', 'objects', 'refs')) | |
|
67 | 68 | commands = frozenset(('git-upload-pack', 'git-receive-pack')) |
|
68 |
valid_accepts = frozenset( |
|
|
69 | c for c in commands)) | |
|
69 | valid_accepts = frozenset(f'application/x-{c}-result' for c in commands) | |
|
70 | 70 | |
|
71 | 71 | # The last bytes are the SHA1 of the first 12 bytes. |
|
72 | 72 | EMPTY_PACK = ( |
|
73 | 'PACK\x00\x00\x00\x02\x00\x00\x00\x00' + | |
|
74 |
' |
|
|
73 | b'PACK\x00\x00\x00\x02\x00\x00\x00\x00\x02\x9d\x08' + | |
|
74 | b'\x82;\xd8\xa8\xea\xb5\x10\xadj\xc7\\\x82<\xfd>\xd3\x1e' | |
|
75 | 75 | ) |
|
76 | SIDE_BAND_CAPS = frozenset(('side-band', 'side-band-64k')) | |
|
76 | FLUSH_PACKET = b"0000" | |
|
77 | 77 | |
|
78 | def __init__(self, repo_name, content_path, git_path, update_server_info, | |
|
79 | extras): | |
|
78 | SIDE_BAND_CAPS = frozenset((CAPABILITY_SIDE_BAND, CAPABILITY_SIDE_BAND_64K)) | |
|
79 | ||
|
80 | def __init__(self, repo_name, content_path, git_path, update_server_info, extras): | |
|
80 | 81 | files = frozenset(f.lower() for f in os.listdir(content_path)) |
|
81 | 82 | valid_dir_signature = self.git_folder_signature.issubset(files) |
|
82 | 83 | |
|
83 | 84 | if not valid_dir_signature: |
|
84 |
raise OSError(' |
|
|
85 | raise OSError(f'{content_path} missing git signature') | |
|
85 | 86 | |
|
86 | 87 | self.content_path = content_path |
|
87 | 88 | self.repo_name = repo_name |
@@ -123,7 +124,7 b' class GitRepository(object):' | |||
|
123 | 124 | # It reads binary, per number of bytes specified. |
|
124 | 125 | # if you do add '\n' as part of data, count it. |
|
125 | 126 | server_advert = '# service=%s\n' % git_command |
|
126 |
packet_len = |
|
|
127 | packet_len = hex(len(server_advert) + 4)[2:].rjust(4, '0').lower() | |
|
127 | 128 | try: |
|
128 | 129 | gitenv = dict(os.environ) |
|
129 | 130 | # forget all configs |
@@ -133,15 +134,15 b' class GitRepository(object):' | |||
|
133 | 134 | out = subprocessio.SubprocessIOChunker( |
|
134 | 135 | command, |
|
135 | 136 | env=gitenv, |
|
136 |
starting_values=[packet_len + server_advert + |
|
|
137 | starting_values=[ascii_bytes(packet_len + server_advert) + self.FLUSH_PACKET], | |
|
137 | 138 | shell=False |
|
138 | 139 | ) |
|
139 |
except |
|
|
140 | except OSError: | |
|
140 | 141 | log.exception('Error processing command') |
|
141 | 142 | raise exc.HTTPExpectationFailed() |
|
142 | 143 | |
|
143 | 144 | resp = Response() |
|
144 |
resp.content_type = 'application/x- |
|
|
145 | resp.content_type = f'application/x-{git_command}-advertisement' | |
|
145 | 146 | resp.charset = None |
|
146 | 147 | resp.app_iter = out |
|
147 | 148 | |
@@ -166,34 +167,103 b' class GitRepository(object):' | |||
|
166 | 167 | We also print in the error output a message explaining why the command |
|
167 | 168 | was aborted. |
|
168 | 169 | |
|
169 | If aditionally, the user is accepting messages we send them the output | |
|
170 | If additionally, the user is accepting messages we send them the output | |
|
170 | 171 | of the pre-pull hook. |
|
171 | 172 | |
|
172 | 173 | Note that for clients not supporting side-band we just send them the |
|
173 | 174 | emtpy PACK file. |
|
174 | 175 | """ |
|
176 | ||
|
175 | 177 | if self.SIDE_BAND_CAPS.intersection(capabilities): |
|
176 | 178 | response = [] |
|
177 | 179 | proto = dulwich.protocol.Protocol(None, response.append) |
|
178 |
proto.write_pkt_line( |
|
|
179 | self._write_sideband_to_proto(pre_pull_messages, proto, | |
|
180 | capabilities) | |
|
180 | proto.write_pkt_line(dulwich.protocol.NAK_LINE) | |
|
181 | ||
|
182 | self._write_sideband_to_proto(proto, ascii_bytes(pre_pull_messages, allow_bytes=True), capabilities) | |
|
181 | 183 | # N.B.(skreft): Do not change the sideband channel to 3, as that |
|
182 | 184 | # produces a fatal error in the client: |
|
183 | 185 | # fatal: error in sideband demultiplexer |
|
184 |
proto.write_sideband( |
|
|
185 | proto.write_sideband(1, self.EMPTY_PACK) | |
|
186 | proto.write_sideband( | |
|
187 | dulwich.protocol.SIDE_BAND_CHANNEL_PROGRESS, | |
|
188 | ascii_bytes('Pre pull hook failed: aborting\n', allow_bytes=True)) | |
|
189 | proto.write_sideband( | |
|
190 | dulwich.protocol.SIDE_BAND_CHANNEL_DATA, | |
|
191 | ascii_bytes(self.EMPTY_PACK, allow_bytes=True)) | |
|
186 | 192 | |
|
187 | # writes 0000 | |
|
193 | # writes b"0000" as default | |
|
188 | 194 | proto.write_pkt_line(None) |
|
189 | 195 | |
|
190 | 196 | return response |
|
191 | 197 | else: |
|
192 | return [self.EMPTY_PACK] | |
|
198 | return [ascii_bytes(self.EMPTY_PACK, allow_bytes=True)] | |
|
199 | ||
|
200 | def _build_post_pull_response(self, response, capabilities, start_message, end_message): | |
|
201 | """ | |
|
202 | Given a list response we inject the post-pull messages. | |
|
203 | ||
|
204 | We only inject the messages if the client supports sideband, and the | |
|
205 | response has the format: | |
|
206 | 0008NAK\n...0000 | |
|
207 | ||
|
208 | Note that we do not check the no-progress capability as by default, git | |
|
209 | sends it, which effectively would block all messages. | |
|
210 | """ | |
|
211 | ||
|
212 | if not self.SIDE_BAND_CAPS.intersection(capabilities): | |
|
213 | return response | |
|
214 | ||
|
215 | if not start_message and not end_message: | |
|
216 | return response | |
|
217 | ||
|
218 | try: | |
|
219 | iter(response) | |
|
220 | # iterator probably will work, we continue | |
|
221 | except TypeError: | |
|
222 | raise TypeError(f'response must be an iterator: got {type(response)}') | |
|
223 | if isinstance(response, (list, tuple)): | |
|
224 | raise TypeError(f'response must be an iterator: got {type(response)}') | |
|
225 | ||
|
226 | def injected_response(): | |
|
193 | 227 | |
|
194 | def _write_sideband_to_proto(self, data, proto, capabilities): | |
|
228 | do_loop = 1 | |
|
229 | header_injected = 0 | |
|
230 | next_item = None | |
|
231 | has_item = False | |
|
232 | item = b'' | |
|
233 | ||
|
234 | while do_loop: | |
|
235 | ||
|
236 | try: | |
|
237 | next_item = next(response) | |
|
238 | except StopIteration: | |
|
239 | do_loop = 0 | |
|
240 | ||
|
241 | if has_item: | |
|
242 | # last item ! alter it now | |
|
243 | if do_loop == 0 and item.endswith(self.FLUSH_PACKET): | |
|
244 | new_response = [item[:-4]] | |
|
245 | new_response.extend(self._get_messages(end_message, capabilities)) | |
|
246 | new_response.append(self.FLUSH_PACKET) | |
|
247 | item = b''.join(new_response) | |
|
248 | ||
|
249 | yield item | |
|
250 | ||
|
251 | has_item = True | |
|
252 | item = next_item | |
|
253 | ||
|
254 | # alter item if it's the initial chunk | |
|
255 | if not header_injected and item.startswith(b'0008NAK\n'): | |
|
256 | new_response = [b'0008NAK\n'] | |
|
257 | new_response.extend(self._get_messages(start_message, capabilities)) | |
|
258 | new_response.append(item[8:]) | |
|
259 | item = b''.join(new_response) | |
|
260 | header_injected = 1 | |
|
261 | ||
|
262 | return injected_response() | |
|
263 | ||
|
264 | def _write_sideband_to_proto(self, proto, data, capabilities): | |
|
195 | 265 | """ |
|
196 |
Write the data to the proto's sideband number 2 |
|
|
266 | Write the data to the proto's sideband number 2 == SIDE_BAND_CHANNEL_PROGRESS | |
|
197 | 267 | |
|
198 | 268 | We do not use dulwich's write_sideband directly as it only supports |
|
199 | 269 | side-band-64k. |
@@ -204,68 +274,27 b' class GitRepository(object):' | |||
|
204 | 274 | # N.B.(skreft): The values below are explained in the pack protocol |
|
205 | 275 | # documentation, section Packfile Data. |
|
206 | 276 | # https://github.com/git/git/blob/master/Documentation/technical/pack-protocol.txt |
|
207 |
if |
|
|
277 | if CAPABILITY_SIDE_BAND_64K in capabilities: | |
|
208 | 278 | chunk_size = 65515 |
|
209 |
elif |
|
|
279 | elif CAPABILITY_SIDE_BAND in capabilities: | |
|
210 | 280 | chunk_size = 995 |
|
211 | 281 | else: |
|
212 | 282 | return |
|
213 | 283 | |
|
214 | chunker = ( | |
|
215 | data[i:i + chunk_size] for i in xrange(0, len(data), chunk_size)) | |
|
284 | chunker = (data[i:i + chunk_size] for i in range(0, len(data), chunk_size)) | |
|
216 | 285 | |
|
217 | 286 | for chunk in chunker: |
|
218 | proto.write_sideband(2, chunk) | |
|
287 | proto.write_sideband(dulwich.protocol.SIDE_BAND_CHANNEL_PROGRESS, ascii_bytes(chunk, allow_bytes=True)) | |
|
219 | 288 | |
|
220 | 289 | def _get_messages(self, data, capabilities): |
|
221 | 290 | """Return a list with packets for sending data in sideband number 2.""" |
|
222 | 291 | response = [] |
|
223 | 292 | proto = dulwich.protocol.Protocol(None, response.append) |
|
224 | 293 | |
|
225 |
self._write_sideband_to_proto( |
|
|
294 | self._write_sideband_to_proto(proto, data, capabilities) | |
|
226 | 295 | |
|
227 | 296 | return response |
|
228 | 297 | |
|
229 | def _inject_messages_to_response(self, response, capabilities, | |
|
230 | start_messages, end_messages): | |
|
231 | """ | |
|
232 | Given a list response we inject the pre/post-pull messages. | |
|
233 | ||
|
234 | We only inject the messages if the client supports sideband, and the | |
|
235 | response has the format: | |
|
236 | 0008NAK\n...0000 | |
|
237 | ||
|
238 | Note that we do not check the no-progress capability as by default, git | |
|
239 | sends it, which effectively would block all messages. | |
|
240 | """ | |
|
241 | if not self.SIDE_BAND_CAPS.intersection(capabilities): | |
|
242 | return response | |
|
243 | ||
|
244 | if not start_messages and not end_messages: | |
|
245 | return response | |
|
246 | ||
|
247 | # make a list out of response if it's an iterator | |
|
248 | # so we can investigate it for message injection. | |
|
249 | if hasattr(response, '__iter__'): | |
|
250 | response = list(response) | |
|
251 | ||
|
252 | if (not response[0].startswith('0008NAK\n') or | |
|
253 | not response[-1].endswith('0000')): | |
|
254 | return response | |
|
255 | ||
|
256 | new_response = ['0008NAK\n'] | |
|
257 | new_response.extend(self._get_messages(start_messages, capabilities)) | |
|
258 | if len(response) == 1: | |
|
259 | new_response.append(response[0][8:-4]) | |
|
260 | else: | |
|
261 | new_response.append(response[0][8:]) | |
|
262 | new_response.extend(response[1:-1]) | |
|
263 | new_response.append(response[-1][:-4]) | |
|
264 | new_response.extend(self._get_messages(end_messages, capabilities)) | |
|
265 | new_response.append('0000') | |
|
266 | ||
|
267 | return new_response | |
|
268 | ||
|
269 | 298 | def backend(self, request, environ): |
|
270 | 299 | """ |
|
271 | 300 | WSGI Response producer for HTTP POST Git Smart HTTP requests. |
@@ -304,14 +333,15 b' class GitRepository(object):' | |||
|
304 | 333 | inputstream = request.body_file_seekable |
|
305 | 334 | |
|
306 | 335 | resp = Response() |
|
307 |
resp.content_type = |
|
|
308 | git_command.encode('utf8')) | |
|
336 | resp.content_type = f'application/x-{git_command}-result' | |
|
309 | 337 | resp.charset = None |
|
310 | 338 | |
|
311 | 339 | pre_pull_messages = '' |
|
340 | # Upload-pack == clone | |
|
312 | 341 | if git_command == 'git-upload-pack': |
|
313 |
|
|
|
314 | if status != 0: | |
|
342 | hook_response = hooks.git_pre_pull(self.extras) | |
|
343 | if hook_response.status != 0: | |
|
344 | pre_pull_messages = hook_response.output | |
|
315 | 345 | resp.app_iter = self._build_failed_pre_pull_response( |
|
316 | 346 | capabilities, pre_pull_messages) |
|
317 | 347 | return resp |
@@ -326,7 +356,7 b' class GitRepository(object):' | |||
|
326 | 356 | |
|
327 | 357 | out = subprocessio.SubprocessIOChunker( |
|
328 | 358 | cmd, |
|
329 | inputstream=inputstream, | |
|
359 | input_stream=inputstream, | |
|
330 | 360 | env=gitenv, |
|
331 | 361 | cwd=self.content_path, |
|
332 | 362 | shell=False, |
@@ -346,7 +376,7 b' class GitRepository(object):' | |||
|
346 | 376 | log.debug('handling cmd %s', cmd) |
|
347 | 377 | output = subprocessio.SubprocessIOChunker( |
|
348 | 378 | cmd, |
|
349 | inputstream=inputstream, | |
|
379 | input_stream=inputstream, | |
|
350 | 380 | env=gitenv, |
|
351 | 381 | cwd=self.content_path, |
|
352 | 382 | shell=False, |
@@ -357,10 +387,11 b' class GitRepository(object):' | |||
|
357 | 387 | for _ in output: |
|
358 | 388 | pass |
|
359 | 389 | |
|
390 | # Upload-pack == clone | |
|
360 | 391 | if git_command == 'git-upload-pack': |
|
361 |
|
|
|
362 | resp.app_iter = self._inject_messages_to_response( | |
|
363 |
|
|
|
392 | hook_response = hooks.git_post_pull(self.extras) | |
|
393 | post_pull_messages = hook_response.output | |
|
394 | resp.app_iter = self._build_post_pull_response(out, capabilities, pre_pull_messages, post_pull_messages) | |
|
364 | 395 | else: |
|
365 | 396 | resp.app_iter = out |
|
366 | 397 |
This diff has been collapsed as it changes many lines, (688 lines changed) Show them Hide them | |||
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -18,12 +18,12 b'' | |||
|
18 | 18 | import collections |
|
19 | 19 | import logging |
|
20 | 20 | import os |
|
21 | import posixpath as vcspath | |
|
22 | 21 | import re |
|
23 | 22 | import stat |
|
24 | 23 | import traceback |
|
25 | import urllib | |
|
26 |
import urllib |
|
|
24 | import urllib.request | |
|
25 | import urllib.parse | |
|
26 | import urllib.error | |
|
27 | 27 | from functools import wraps |
|
28 | 28 | |
|
29 | 29 | import more_itertools |
@@ -31,7 +31,7 b' import pygit2' | |||
|
31 | 31 | from pygit2 import Repository as LibGit2Repo |
|
32 | 32 | from pygit2 import index as LibGit2Index |
|
33 | 33 | from dulwich import index, objects |
|
34 | from dulwich.client import HttpGitClient, LocalGitClient | |
|
34 | from dulwich.client import HttpGitClient, LocalGitClient, FetchPackResult | |
|
35 | 35 | from dulwich.errors import ( |
|
36 | 36 | NotGitRepository, ChecksumMismatch, WrongObjectException, |
|
37 | 37 | MissingCommitError, ObjectMissing, HangupException, |
@@ -40,8 +40,8 b' from dulwich.repo import Repo as Dulwich' | |||
|
40 | 40 | from dulwich.server import update_server_info |
|
41 | 41 | |
|
42 | 42 | from vcsserver import exceptions, settings, subprocessio |
|
43 |
from vcsserver.utils import safe_str, safe_int, safe_ |
|
|
44 |
from vcsserver.base import RepoFactory, obfuscate_qs, ArchiveNode, archive_ |
|
|
43 | from vcsserver.str_utils import safe_str, safe_int, safe_bytes, ascii_bytes | |
|
44 | from vcsserver.base import RepoFactory, obfuscate_qs, ArchiveNode, store_archive_in_cache, BytesEnvelope, BinaryEnvelope | |
|
45 | 45 | from vcsserver.hgcompat import ( |
|
46 | 46 | hg_url as url_parser, httpbasicauthhandler, httpdigestauthhandler) |
|
47 | 47 | from vcsserver.git_lfs.lib import LFSOidStore |
@@ -50,19 +50,12 b' from vcsserver.vcs_base import RemoteBas' | |||
|
50 | 50 | DIR_STAT = stat.S_IFDIR |
|
51 | 51 | FILE_MODE = stat.S_IFMT |
|
52 | 52 | GIT_LINK = objects.S_IFGITLINK |
|
53 | PEELED_REF_MARKER = '^{}' | |
|
54 | ||
|
53 | PEELED_REF_MARKER = b'^{}' | |
|
54 | HEAD_MARKER = b'HEAD' | |
|
55 | 55 | |
|
56 | 56 | log = logging.getLogger(__name__) |
|
57 | 57 | |
|
58 | 58 | |
|
59 | def str_to_dulwich(value): | |
|
60 | """ | |
|
61 | Dulwich 0.10.1a requires `unicode` objects to be passed in. | |
|
62 | """ | |
|
63 | return value.decode(settings.WIRE_ENCODING) | |
|
64 | ||
|
65 | ||
|
66 | 59 | def reraise_safe_exceptions(func): |
|
67 | 60 | """Converts Dulwich exceptions to something neutral.""" |
|
68 | 61 | |
@@ -76,8 +69,8 b' def reraise_safe_exceptions(func):' | |||
|
76 | 69 | except (HangupException, UnexpectedCommandError) as e: |
|
77 | 70 | exc = exceptions.VcsException(org_exc=e) |
|
78 | 71 | raise exc(safe_str(e)) |
|
79 |
except Exception |
|
|
80 |
# NOTE(marcink): bec |
|
|
72 | except Exception: | |
|
73 | # NOTE(marcink): because of how dulwich handles some exceptions | |
|
81 | 74 | # (KeyError on empty repos), we cannot track this and catch all |
|
82 | 75 | # exceptions, it's an exceptions from other handlers |
|
83 | 76 | #if not hasattr(e, '_vcs_kind'): |
@@ -114,10 +107,14 b' class GitFactory(RepoFactory):' | |||
|
114 | 107 | |
|
115 | 108 | def _create_repo(self, wire, create, use_libgit2=False): |
|
116 | 109 | if use_libgit2: |
|
117 |
re |
|
|
110 | repo = Repository(safe_bytes(wire['path'])) | |
|
118 | 111 | else: |
|
119 | repo_path = str_to_dulwich(wire['path']) | |
|
120 | return Repo(repo_path) | |
|
112 | # dulwich mode | |
|
113 | repo_path = safe_str(wire['path'], to_encoding=settings.WIRE_ENCODING) | |
|
114 | repo = Repo(repo_path) | |
|
115 | ||
|
116 | log.debug('repository created: got GIT object: %s', repo) | |
|
117 | return repo | |
|
121 | 118 | |
|
122 | 119 | def repo(self, wire, create=False, use_libgit2=False): |
|
123 | 120 | """ |
@@ -129,6 +126,28 b' class GitFactory(RepoFactory):' | |||
|
129 | 126 | return self.repo(wire, use_libgit2=True) |
|
130 | 127 | |
|
131 | 128 | |
|
129 | def create_signature_from_string(author_str, **kwargs): | |
|
130 | """ | |
|
131 | Creates a pygit2.Signature object from a string of the format 'Name <email>'. | |
|
132 | ||
|
133 | :param author_str: String of the format 'Name <email>' | |
|
134 | :return: pygit2.Signature object | |
|
135 | """ | |
|
136 | match = re.match(r'^(.+) <(.+)>$', author_str) | |
|
137 | if match is None: | |
|
138 | raise ValueError(f"Invalid format: {author_str}") | |
|
139 | ||
|
140 | name, email = match.groups() | |
|
141 | return pygit2.Signature(name, email, **kwargs) | |
|
142 | ||
|
143 | ||
|
144 | def get_obfuscated_url(url_obj): | |
|
145 | url_obj.passwd = b'*****' if url_obj.passwd else url_obj.passwd | |
|
146 | url_obj.query = obfuscate_qs(url_obj.query) | |
|
147 | obfuscated_uri = str(url_obj) | |
|
148 | return obfuscated_uri | |
|
149 | ||
|
150 | ||
|
132 | 151 | class GitRemote(RemoteBase): |
|
133 | 152 | |
|
134 | 153 | def __init__(self, factory): |
@@ -141,10 +160,17 b' class GitRemote(RemoteBase):' | |||
|
141 | 160 | "parents": self.parents, |
|
142 | 161 | "_commit": self.revision, |
|
143 | 162 | } |
|
163 | self._bulk_file_methods = { | |
|
164 | "size": self.get_node_size, | |
|
165 | "data": self.get_node_data, | |
|
166 | "flags": self.get_node_flags, | |
|
167 | "is_binary": self.get_node_is_binary, | |
|
168 | "md5": self.md5_hash | |
|
169 | } | |
|
144 | 170 | |
|
145 | 171 | def _wire_to_config(self, wire): |
|
146 | 172 | if 'config' in wire: |
|
147 |
return |
|
|
173 | return {x[0] + '_' + x[1]: x[2] for x in wire['config']} | |
|
148 | 174 | return {} |
|
149 | 175 | |
|
150 | 176 | def _remote_conf(self, config): |
@@ -153,17 +179,17 b' class GitRemote(RemoteBase):' | |||
|
153 | 179 | ] |
|
154 | 180 | ssl_cert_dir = config.get('vcs_ssl_dir') |
|
155 | 181 | if ssl_cert_dir: |
|
156 |
params.extend(['-c', 'http.sslCAinfo={}' |
|
|
182 | params.extend(['-c', f'http.sslCAinfo={ssl_cert_dir}']) | |
|
157 | 183 | return params |
|
158 | 184 | |
|
159 | 185 | @reraise_safe_exceptions |
|
160 | 186 | def discover_git_version(self): |
|
161 | 187 | stdout, _ = self.run_git_command( |
|
162 | 188 | {}, ['--version'], _bare=True, _safe=True) |
|
163 | prefix = 'git version' | |
|
189 | prefix = b'git version' | |
|
164 | 190 | if stdout.startswith(prefix): |
|
165 | 191 | stdout = stdout[len(prefix):] |
|
166 | return stdout.strip() | |
|
192 | return safe_str(stdout.strip()) | |
|
167 | 193 | |
|
168 | 194 | @reraise_safe_exceptions |
|
169 | 195 | def is_empty(self, wire): |
@@ -186,20 +212,27 b' class GitRemote(RemoteBase):' | |||
|
186 | 212 | def assert_correct_path(self, wire): |
|
187 | 213 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
188 | 214 | region = self._region(wire) |
|
215 | ||
|
189 | 216 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
190 | def _assert_correct_path(_context_uid, _repo_id): | |
|
191 |
|
|
|
192 |
|
|
|
193 | with repo_init as repo: | |
|
194 |
|
|
|
195 | except pygit2.GitError: | |
|
196 | path = wire.get('path') | |
|
197 | tb = traceback.format_exc() | |
|
198 | log.debug("Invalid Git path `%s`, tb: %s", path, tb) | |
|
217 | def _assert_correct_path(_context_uid, _repo_id, fast_check): | |
|
218 | if fast_check: | |
|
219 | path = safe_str(wire['path']) | |
|
220 | if pygit2.discover_repository(path): | |
|
221 | return True | |
|
199 | 222 | return False |
|
223 | else: | |
|
224 | try: | |
|
225 | repo_init = self._factory.repo_libgit2(wire) | |
|
226 | with repo_init: | |
|
227 | pass | |
|
228 | except pygit2.GitError: | |
|
229 | path = wire.get('path') | |
|
230 | tb = traceback.format_exc() | |
|
231 | log.debug("Invalid Git path `%s`, tb: %s", path, tb) | |
|
232 | return False | |
|
233 | return True | |
|
200 | 234 | |
|
201 | return True | |
|
202 | return _assert_correct_path(context_uid, repo_id) | |
|
235 | return _assert_correct_path(context_uid, repo_id, True) | |
|
203 | 236 | |
|
204 | 237 | @reraise_safe_exceptions |
|
205 | 238 | def bare(self, wire): |
@@ -208,17 +241,69 b' class GitRemote(RemoteBase):' | |||
|
208 | 241 | return repo.is_bare |
|
209 | 242 | |
|
210 | 243 | @reraise_safe_exceptions |
|
244 | def get_node_data(self, wire, commit_id, path): | |
|
245 | repo_init = self._factory.repo_libgit2(wire) | |
|
246 | with repo_init as repo: | |
|
247 | commit = repo[commit_id] | |
|
248 | blob_obj = commit.tree[path] | |
|
249 | ||
|
250 | if blob_obj.type != pygit2.GIT_OBJ_BLOB: | |
|
251 | raise exceptions.LookupException()( | |
|
252 | f'Tree for commit_id:{commit_id} is not a blob: {blob_obj.type_str}') | |
|
253 | ||
|
254 | return BytesEnvelope(blob_obj.data) | |
|
255 | ||
|
256 | @reraise_safe_exceptions | |
|
257 | def get_node_size(self, wire, commit_id, path): | |
|
258 | repo_init = self._factory.repo_libgit2(wire) | |
|
259 | with repo_init as repo: | |
|
260 | commit = repo[commit_id] | |
|
261 | blob_obj = commit.tree[path] | |
|
262 | ||
|
263 | if blob_obj.type != pygit2.GIT_OBJ_BLOB: | |
|
264 | raise exceptions.LookupException()( | |
|
265 | f'Tree for commit_id:{commit_id} is not a blob: {blob_obj.type_str}') | |
|
266 | ||
|
267 | return blob_obj.size | |
|
268 | ||
|
269 | @reraise_safe_exceptions | |
|
270 | def get_node_flags(self, wire, commit_id, path): | |
|
271 | repo_init = self._factory.repo_libgit2(wire) | |
|
272 | with repo_init as repo: | |
|
273 | commit = repo[commit_id] | |
|
274 | blob_obj = commit.tree[path] | |
|
275 | ||
|
276 | if blob_obj.type != pygit2.GIT_OBJ_BLOB: | |
|
277 | raise exceptions.LookupException()( | |
|
278 | f'Tree for commit_id:{commit_id} is not a blob: {blob_obj.type_str}') | |
|
279 | ||
|
280 | return blob_obj.filemode | |
|
281 | ||
|
282 | @reraise_safe_exceptions | |
|
283 | def get_node_is_binary(self, wire, commit_id, path): | |
|
284 | repo_init = self._factory.repo_libgit2(wire) | |
|
285 | with repo_init as repo: | |
|
286 | commit = repo[commit_id] | |
|
287 | blob_obj = commit.tree[path] | |
|
288 | ||
|
289 | if blob_obj.type != pygit2.GIT_OBJ_BLOB: | |
|
290 | raise exceptions.LookupException()( | |
|
291 | f'Tree for commit_id:{commit_id} is not a blob: {blob_obj.type_str}') | |
|
292 | ||
|
293 | return blob_obj.is_binary | |
|
294 | ||
|
295 | @reraise_safe_exceptions | |
|
211 | 296 | def blob_as_pretty_string(self, wire, sha): |
|
212 | 297 | repo_init = self._factory.repo_libgit2(wire) |
|
213 | 298 | with repo_init as repo: |
|
214 | 299 | blob_obj = repo[sha] |
|
215 |
|
|
|
216 | return blob | |
|
300 | return BytesEnvelope(blob_obj.data) | |
|
217 | 301 | |
|
218 | 302 | @reraise_safe_exceptions |
|
219 | 303 | def blob_raw_length(self, wire, sha): |
|
220 | 304 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
221 | 305 | region = self._region(wire) |
|
306 | ||
|
222 | 307 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
223 | 308 | def _blob_raw_length(_repo_id, _sha): |
|
224 | 309 | |
@@ -230,10 +315,10 b' class GitRemote(RemoteBase):' | |||
|
230 | 315 | return _blob_raw_length(repo_id, sha) |
|
231 | 316 | |
|
232 | 317 | def _parse_lfs_pointer(self, raw_content): |
|
318 | spec_string = b'version https://git-lfs.github.com/spec' | |
|
319 | if raw_content and raw_content.startswith(spec_string): | |
|
233 | 320 | |
|
234 | spec_string = 'version https://git-lfs.github.com/spec' | |
|
235 | if raw_content and raw_content.startswith(spec_string): | |
|
236 | pattern = re.compile(r""" | |
|
321 | pattern = re.compile(rb""" | |
|
237 | 322 | (?:\n)? |
|
238 | 323 | ^version[ ]https://git-lfs\.github\.com/spec/(?P<spec_ver>v\d+)\n |
|
239 | 324 | ^oid[ ] sha256:(?P<oid_hash>[0-9a-f]{64})\n |
@@ -249,8 +334,8 b' class GitRemote(RemoteBase):' | |||
|
249 | 334 | @reraise_safe_exceptions |
|
250 | 335 | def is_large_file(self, wire, commit_id): |
|
251 | 336 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
337 | region = self._region(wire) | |
|
252 | 338 | |
|
253 | region = self._region(wire) | |
|
254 | 339 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
255 | 340 | def _is_large_file(_repo_id, _sha): |
|
256 | 341 | repo_init = self._factory.repo_libgit2(wire) |
@@ -266,8 +351,8 b' class GitRemote(RemoteBase):' | |||
|
266 | 351 | @reraise_safe_exceptions |
|
267 | 352 | def is_binary(self, wire, tree_id): |
|
268 | 353 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
354 | region = self._region(wire) | |
|
269 | 355 | |
|
270 | region = self._region(wire) | |
|
271 | 356 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
272 | 357 | def _is_binary(_repo_id, _tree_id): |
|
273 | 358 | repo_init = self._factory.repo_libgit2(wire) |
@@ -278,6 +363,26 b' class GitRemote(RemoteBase):' | |||
|
278 | 363 | return _is_binary(repo_id, tree_id) |
|
279 | 364 | |
|
280 | 365 | @reraise_safe_exceptions |
|
366 | def md5_hash(self, wire, commit_id, path): | |
|
367 | cache_on, context_uid, repo_id = self._cache_on(wire) | |
|
368 | region = self._region(wire) | |
|
369 | ||
|
370 | @region.conditional_cache_on_arguments(condition=cache_on) | |
|
371 | def _md5_hash(_repo_id, _commit_id, _path): | |
|
372 | repo_init = self._factory.repo_libgit2(wire) | |
|
373 | with repo_init as repo: | |
|
374 | commit = repo[_commit_id] | |
|
375 | blob_obj = commit.tree[_path] | |
|
376 | ||
|
377 | if blob_obj.type != pygit2.GIT_OBJ_BLOB: | |
|
378 | raise exceptions.LookupException()( | |
|
379 | f'Tree for commit_id:{_commit_id} is not a blob: {blob_obj.type_str}') | |
|
380 | ||
|
381 | return '' | |
|
382 | ||
|
383 | return _md5_hash(repo_id, commit_id, path) | |
|
384 | ||
|
385 | @reraise_safe_exceptions | |
|
281 | 386 | def in_largefiles_store(self, wire, oid): |
|
282 | 387 | conf = self._wire_to_config(wire) |
|
283 | 388 | repo_init = self._factory.repo_libgit2(wire) |
@@ -305,90 +410,105 b' class GitRemote(RemoteBase):' | |||
|
305 | 410 | store = LFSOidStore( |
|
306 | 411 | oid=oid, repo=repo_name, store_location=store_location) |
|
307 | 412 | return store.oid_path |
|
308 |
raise ValueError('Unable to fetch oid with path {}' |
|
|
413 | raise ValueError(f'Unable to fetch oid with path {oid}') | |
|
309 | 414 | |
|
310 | 415 | @reraise_safe_exceptions |
|
311 | 416 | def bulk_request(self, wire, rev, pre_load): |
|
312 | 417 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
313 | 418 | region = self._region(wire) |
|
419 | ||
|
314 | 420 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
315 | 421 | def _bulk_request(_repo_id, _rev, _pre_load): |
|
316 | 422 | result = {} |
|
317 | 423 | for attr in pre_load: |
|
318 | 424 | try: |
|
319 | 425 | method = self._bulk_methods[attr] |
|
426 | wire.update({'cache': False}) # disable cache for bulk calls so we don't double cache | |
|
320 | 427 | args = [wire, rev] |
|
321 | 428 | result[attr] = method(*args) |
|
322 | 429 | except KeyError as e: |
|
323 | raise exceptions.VcsException(e)( | |
|
324 | "Unknown bulk attribute: %s" % attr) | |
|
430 | raise exceptions.VcsException(e)(f"Unknown bulk attribute: {attr}") | |
|
325 | 431 | return result |
|
326 | 432 | |
|
327 | 433 | return _bulk_request(repo_id, rev, sorted(pre_load)) |
|
328 | 434 | |
|
329 | def _build_opener(self, url): | |
|
435 | @reraise_safe_exceptions | |
|
436 | def bulk_file_request(self, wire, commit_id, path, pre_load): | |
|
437 | cache_on, context_uid, repo_id = self._cache_on(wire) | |
|
438 | region = self._region(wire) | |
|
439 | ||
|
440 | @region.conditional_cache_on_arguments(condition=cache_on) | |
|
441 | def _bulk_file_request(_repo_id, _commit_id, _path, _pre_load): | |
|
442 | result = {} | |
|
443 | for attr in pre_load: | |
|
444 | try: | |
|
445 | method = self._bulk_file_methods[attr] | |
|
446 | wire.update({'cache': False}) # disable cache for bulk calls so we don't double cache | |
|
447 | result[attr] = method(wire, _commit_id, _path) | |
|
448 | except KeyError as e: | |
|
449 | raise exceptions.VcsException(e)(f'Unknown bulk attribute: "{attr}"') | |
|
450 | return BinaryEnvelope(result) | |
|
451 | ||
|
452 | return _bulk_file_request(repo_id, commit_id, path, sorted(pre_load)) | |
|
453 | ||
|
454 | def _build_opener(self, url: str): | |
|
330 | 455 | handlers = [] |
|
331 | url_obj = url_parser(url) | |
|
332 |
|
|
|
456 | url_obj = url_parser(safe_bytes(url)) | |
|
457 | authinfo = url_obj.authinfo()[1] | |
|
333 | 458 | |
|
334 | 459 | if authinfo: |
|
335 | 460 | # create a password manager |
|
336 |
passmgr = urllib |
|
|
461 | passmgr = urllib.request.HTTPPasswordMgrWithDefaultRealm() | |
|
337 | 462 | passmgr.add_password(*authinfo) |
|
338 | 463 | |
|
339 | 464 | handlers.extend((httpbasicauthhandler(passmgr), |
|
340 | 465 | httpdigestauthhandler(passmgr))) |
|
341 | 466 | |
|
342 |
return urllib |
|
|
343 | ||
|
344 | def _type_id_to_name(self, type_id): | |
|
345 | return { | |
|
346 | 1: b'commit', | |
|
347 | 2: b'tree', | |
|
348 | 3: b'blob', | |
|
349 | 4: b'tag' | |
|
350 | }[type_id] | |
|
467 | return urllib.request.build_opener(*handlers) | |
|
351 | 468 | |
|
352 | 469 | @reraise_safe_exceptions |
|
353 | 470 | def check_url(self, url, config): |
|
354 | url_obj = url_parser(url) | |
|
355 | test_uri, _ = url_obj.authinfo() | |
|
356 | url_obj.passwd = '*****' if url_obj.passwd else url_obj.passwd | |
|
357 | url_obj.query = obfuscate_qs(url_obj.query) | |
|
358 | cleaned_uri = str(url_obj) | |
|
359 |
log.info("Checking URL for remote cloning/import: %s", |
|
|
471 | url_obj = url_parser(safe_bytes(url)) | |
|
472 | ||
|
473 | test_uri = safe_str(url_obj.authinfo()[0]) | |
|
474 | obfuscated_uri = get_obfuscated_url(url_obj) | |
|
475 | ||
|
476 | log.info("Checking URL for remote cloning/import: %s", obfuscated_uri) | |
|
360 | 477 | |
|
361 | 478 | if not test_uri.endswith('info/refs'): |
|
362 | 479 | test_uri = test_uri.rstrip('/') + '/info/refs' |
|
363 | 480 | |
|
364 |
o = self._build_opener( |
|
|
481 | o = self._build_opener(test_uri) | |
|
365 | 482 | o.addheaders = [('User-Agent', 'git/1.7.8.0')] # fake some git |
|
366 | 483 | |
|
367 | 484 | q = {"service": 'git-upload-pack'} |
|
368 | qs = '?%s' % urllib.urlencode(q) | |
|
369 |
cu = " |
|
|
370 |
req = urllib |
|
|
485 | qs = '?%s' % urllib.parse.urlencode(q) | |
|
486 | cu = f"{test_uri}{qs}" | |
|
487 | req = urllib.request.Request(cu, None, {}) | |
|
371 | 488 | |
|
372 | 489 | try: |
|
373 |
log.debug("Trying to open URL %s", |
|
|
490 | log.debug("Trying to open URL %s", obfuscated_uri) | |
|
374 | 491 | resp = o.open(req) |
|
375 | 492 | if resp.code != 200: |
|
376 | 493 | raise exceptions.URLError()('Return Code is not 200') |
|
377 | 494 | except Exception as e: |
|
378 |
log.warning("URL cannot be opened: %s", |
|
|
495 | log.warning("URL cannot be opened: %s", obfuscated_uri, exc_info=True) | |
|
379 | 496 | # means it cannot be cloned |
|
380 |
raise exceptions.URLError(e)("[ |
|
|
497 | raise exceptions.URLError(e)(f"[{obfuscated_uri}] org_exc: {e}") | |
|
381 | 498 | |
|
382 | 499 | # now detect if it's proper git repo |
|
383 | gitdata = resp.read() | |
|
384 | if 'service=git-upload-pack' in gitdata: | |
|
500 | gitdata: bytes = resp.read() | |
|
501 | ||
|
502 | if b'service=git-upload-pack' in gitdata: | |
|
385 | 503 | pass |
|
386 | elif re.findall(r'[0-9a-fA-F]{40}\s+refs', gitdata): | |
|
504 | elif re.findall(br'[0-9a-fA-F]{40}\s+refs', gitdata): | |
|
387 | 505 | # old style git can return some other format ! |
|
388 | 506 | pass |
|
389 | 507 | else: |
|
390 | raise exceptions.URLError()( | |
|
391 | "url [%s] does not look like an git" % (cleaned_uri,)) | |
|
508 | e = None | |
|
509 | raise exceptions.URLError(e)( | |
|
510 | "url [%s] does not look like an hg repo org_exc: %s" | |
|
511 | % (obfuscated_uri, e)) | |
|
392 | 512 | |
|
393 | 513 | return True |
|
394 | 514 | |
@@ -415,6 +535,7 b' class GitRemote(RemoteBase):' | |||
|
415 | 535 | def branch(self, wire, commit_id): |
|
416 | 536 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
417 | 537 | region = self._region(wire) |
|
538 | ||
|
418 | 539 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
419 | 540 | def _branch(_context_uid, _repo_id, _commit_id): |
|
420 | 541 | regex = re.compile('^refs/heads') |
@@ -422,7 +543,7 b' class GitRemote(RemoteBase):' | |||
|
422 | 543 | def filter_with(ref): |
|
423 | 544 | return regex.match(ref[0]) and ref[1] == _commit_id |
|
424 | 545 | |
|
425 | branches = filter(filter_with, self.get_refs(wire).items()) | |
|
546 | branches = list(filter(filter_with, list(self.get_refs(wire).items()))) | |
|
426 | 547 | return [x[0].split('refs/heads/')[-1] for x in branches] |
|
427 | 548 | |
|
428 | 549 | return _branch(context_uid, repo_id, commit_id) |
@@ -431,6 +552,7 b' class GitRemote(RemoteBase):' | |||
|
431 | 552 | def commit_branches(self, wire, commit_id): |
|
432 | 553 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
433 | 554 | region = self._region(wire) |
|
555 | ||
|
434 | 556 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
435 | 557 | def _commit_branches(_context_uid, _repo_id, _commit_id): |
|
436 | 558 | repo_init = self._factory.repo_libgit2(wire) |
@@ -449,152 +571,136 b' class GitRemote(RemoteBase):' | |||
|
449 | 571 | repo.object_store.add_object(blob) |
|
450 | 572 | return blob.id |
|
451 | 573 | |
|
452 | # TODO: this is quite complex, check if that can be simplified | |
|
574 | @reraise_safe_exceptions | |
|
575 | def create_commit(self, wire, author, committer, message, branch, new_tree_id, date_args: list[int, int] = None): | |
|
576 | repo_init = self._factory.repo_libgit2(wire) | |
|
577 | with repo_init as repo: | |
|
578 | ||
|
579 | if date_args: | |
|
580 | current_time, offset = date_args | |
|
581 | ||
|
582 | kw = { | |
|
583 | 'time': current_time, | |
|
584 | 'offset': offset | |
|
585 | } | |
|
586 | author = create_signature_from_string(author, **kw) | |
|
587 | committer = create_signature_from_string(committer, **kw) | |
|
588 | ||
|
589 | tree = new_tree_id | |
|
590 | if isinstance(tree, (bytes, str)): | |
|
591 | # validate this tree is in the repo... | |
|
592 | tree = repo[safe_str(tree)].id | |
|
593 | ||
|
594 | parents = [] | |
|
595 | # ensure we COMMIT on top of given branch head | |
|
596 | # check if this repo has ANY branches, otherwise it's a new branch case we need to make | |
|
597 | if branch in repo.branches.local: | |
|
598 | parents += [repo.branches[branch].target] | |
|
599 | elif [x for x in repo.branches.local]: | |
|
600 | parents += [repo.head.target] | |
|
601 | #else: | |
|
602 | # in case we want to commit on new branch we create it on top of HEAD | |
|
603 | #repo.branches.local.create(branch, repo.revparse_single('HEAD')) | |
|
604 | ||
|
605 | # # Create a new commit | |
|
606 | commit_oid = repo.create_commit( | |
|
607 | f'refs/heads/{branch}', # the name of the reference to update | |
|
608 | author, # the author of the commit | |
|
609 | committer, # the committer of the commit | |
|
610 | message, # the commit message | |
|
611 | tree, # the tree produced by the index | |
|
612 | parents # list of parents for the new commit, usually just one, | |
|
613 | ) | |
|
614 | ||
|
615 | new_commit_id = safe_str(commit_oid) | |
|
616 | ||
|
617 | return new_commit_id | |
|
618 | ||
|
453 | 619 | @reraise_safe_exceptions |
|
454 | 620 | def commit(self, wire, commit_data, branch, commit_tree, updated, removed): |
|
455 | # Defines the root tree | |
|
456 | class _Root(object): | |
|
457 | def __repr__(self): | |
|
458 | return 'ROOT TREE' | |
|
459 | ROOT = _Root() | |
|
460 | 621 | |
|
461 | repo = self._factory.repo(wire) | |
|
462 | object_store = repo.object_store | |
|
463 | ||
|
464 | # Create tree and populates it with blobs | |
|
465 | ||
|
466 | if commit_tree and repo[commit_tree]: | |
|
467 | git_commit = repo[commit_data['parents'][0]] | |
|
468 | commit_tree = repo[git_commit.tree] # root tree | |
|
469 | else: | |
|
470 | commit_tree = objects.Tree() | |
|
471 | ||
|
472 | for node in updated: | |
|
473 | # Compute subdirs if needed | |
|
474 | dirpath, nodename = vcspath.split(node['path']) | |
|
475 | dirnames = map(safe_str, dirpath and dirpath.split('/') or []) | |
|
476 | parent = commit_tree | |
|
477 | ancestors = [('', parent)] | |
|
622 | def mode2pygit(mode): | |
|
623 | """ | |
|
624 | git only supports two filemode 644 and 755 | |
|
478 | 625 |
|
|
479 | # Tries to dig for the deepest existing tree | |
|
480 | while dirnames: | |
|
481 | curdir = dirnames.pop(0) | |
|
482 |
|
|
|
483 | dir_id = parent[curdir][1] | |
|
484 | except KeyError: | |
|
485 | # put curdir back into dirnames and stops | |
|
486 | dirnames.insert(0, curdir) | |
|
487 | break | |
|
488 | else: | |
|
489 | # If found, updates parent | |
|
490 | parent = repo[dir_id] | |
|
491 | ancestors.append((curdir, parent)) | |
|
492 | # Now parent is deepest existing tree and we need to create | |
|
493 | # subtrees for dirnames (in reverse order) | |
|
494 | # [this only applies for nodes from added] | |
|
495 | new_trees = [] | |
|
626 | 0o100755 -> 33261 | |
|
627 | 0o100644 -> 33188 | |
|
628 | """ | |
|
629 | return { | |
|
630 | 0o100644: pygit2.GIT_FILEMODE_BLOB, | |
|
631 | 0o100755: pygit2.GIT_FILEMODE_BLOB_EXECUTABLE, | |
|
632 | 0o120000: pygit2.GIT_FILEMODE_LINK | |
|
633 | }.get(mode) or pygit2.GIT_FILEMODE_BLOB | |
|
496 | 634 | |
|
497 | blob = objects.Blob.from_string(node['content']) | |
|
635 | repo_init = self._factory.repo_libgit2(wire) | |
|
636 | with repo_init as repo: | |
|
637 | repo_index = repo.index | |
|
498 | 638 | |
|
499 |
|
|
|
500 | # If there are trees which should be created we need to build | |
|
501 | # them now (in reverse order) | |
|
502 | reversed_dirnames = list(reversed(dirnames)) | |
|
503 | curtree = objects.Tree() | |
|
504 | curtree[node['node_path']] = node['mode'], blob.id | |
|
505 | new_trees.append(curtree) | |
|
506 | for dirname in reversed_dirnames[:-1]: | |
|
507 | newtree = objects.Tree() | |
|
508 | newtree[dirname] = (DIR_STAT, curtree.id) | |
|
509 | new_trees.append(newtree) | |
|
510 | curtree = newtree | |
|
511 | parent[reversed_dirnames[-1]] = (DIR_STAT, curtree.id) | |
|
512 | else: | |
|
513 | parent.add(name=node['node_path'], mode=node['mode'], hexsha=blob.id) | |
|
639 | for pathspec in updated: | |
|
640 | blob_id = repo.create_blob(pathspec['content']) | |
|
641 | ie = pygit2.IndexEntry(pathspec['path'], blob_id, mode2pygit(pathspec['mode'])) | |
|
642 | repo_index.add(ie) | |
|
514 | 643 | |
|
515 | new_trees.append(parent) | |
|
516 | # Update ancestors | |
|
517 | reversed_ancestors = reversed( | |
|
518 | [(a[1], b[1], b[0]) for a, b in zip(ancestors, ancestors[1:])]) | |
|
519 | for parent, tree, path in reversed_ancestors: | |
|
520 | parent[path] = (DIR_STAT, tree.id) | |
|
521 | object_store.add_object(tree) | |
|
644 | for pathspec in removed: | |
|
645 | repo_index.remove(pathspec) | |
|
522 | 646 | |
|
523 | object_store.add_object(blob) | |
|
524 |
|
|
|
525 | object_store.add_object(tree) | |
|
647 | # Write changes to the index | |
|
648 | repo_index.write() | |
|
649 | ||
|
650 | # Create a tree from the updated index | |
|
651 | commit_tree = repo_index.write_tree() | |
|
652 | ||
|
653 | new_tree_id = commit_tree | |
|
526 | 654 | |
|
527 | for node_path in removed: | |
|
528 | paths = node_path.split('/') | |
|
529 | tree = commit_tree # start with top-level | |
|
530 | trees = [{'tree': tree, 'path': ROOT}] | |
|
531 | # Traverse deep into the forest... | |
|
532 | # resolve final tree by iterating the path. | |
|
533 | # e.g a/b/c.txt will get | |
|
534 | # - root as tree then | |
|
535 | # - 'a' as tree, | |
|
536 | # - 'b' as tree, | |
|
537 | # - stop at c as blob. | |
|
538 | for path in paths: | |
|
539 | try: | |
|
540 | obj = repo[tree[path][1]] | |
|
541 | if isinstance(obj, objects.Tree): | |
|
542 | trees.append({'tree': obj, 'path': path}) | |
|
543 | tree = obj | |
|
544 | except KeyError: | |
|
545 | break | |
|
546 | #PROBLEM: | |
|
547 | """ | |
|
548 | We're not editing same reference tree object | |
|
549 | """ | |
|
550 | # Cut down the blob and all rotten trees on the way back... | |
|
551 | for path, tree_data in reversed(zip(paths, trees)): | |
|
552 | tree = tree_data['tree'] | |
|
553 | tree.__delitem__(path) | |
|
554 | # This operation edits the tree, we need to mark new commit back | |
|
655 | author = commit_data['author'] | |
|
656 | committer = commit_data['committer'] | |
|
657 | message = commit_data['message'] | |
|
658 | ||
|
659 | date_args = [int(commit_data['commit_time']), int(commit_data['commit_timezone'])] | |
|
555 | 660 | |
|
556 | if len(tree) > 0: | |
|
557 | # This tree still has elements - don't remove it or any | |
|
558 | # of it's parents | |
|
559 | break | |
|
560 | ||
|
561 | object_store.add_object(commit_tree) | |
|
661 | new_commit_id = self.create_commit(wire, author, committer, message, branch, | |
|
662 | new_tree_id, date_args=date_args) | |
|
562 | 663 | |
|
563 | # Create commit | |
|
564 | commit = objects.Commit() | |
|
565 | commit.tree = commit_tree.id | |
|
566 | for k, v in commit_data.items(): | |
|
567 | setattr(commit, k, v) | |
|
568 | object_store.add_object(commit) | |
|
664 | # libgit2, ensure the branch is there and exists | |
|
665 | self.create_branch(wire, branch, new_commit_id) | |
|
569 | 666 | |
|
570 | self.create_branch(wire, branch, commit.id) | |
|
667 | # libgit2, set new ref to this created commit | |
|
668 | self.set_refs(wire, f'refs/heads/{branch}', new_commit_id) | |
|
571 | 669 | |
|
572 | # dulwich set-ref | |
|
573 | ref = 'refs/heads/%s' % branch | |
|
574 | repo.refs[ref] = commit.id | |
|
575 | ||
|
576 | return commit.id | |
|
670 | return new_commit_id | |
|
577 | 671 | |
|
578 | 672 | @reraise_safe_exceptions |
|
579 | 673 | def pull(self, wire, url, apply_refs=True, refs=None, update_after=False): |
|
580 | 674 | if url != 'default' and '://' not in url: |
|
581 | 675 | client = LocalGitClient(url) |
|
582 | 676 | else: |
|
583 | url_obj = url_parser(url) | |
|
677 | url_obj = url_parser(safe_bytes(url)) | |
|
584 | 678 | o = self._build_opener(url) |
|
585 |
url |
|
|
679 | url = url_obj.authinfo()[0] | |
|
586 | 680 | client = HttpGitClient(base_url=url, opener=o) |
|
587 | 681 | repo = self._factory.repo(wire) |
|
588 | 682 | |
|
589 | 683 | determine_wants = repo.object_store.determine_wants_all |
|
590 | 684 | if refs: |
|
591 | def determine_wants_requested(references): | |
|
592 | return [references[r] for r in references if r in refs] | |
|
685 | refs = [ascii_bytes(x) for x in refs] | |
|
686 | ||
|
687 | def determine_wants_requested(remote_refs): | |
|
688 | determined = [] | |
|
689 | for ref_name, ref_hash in remote_refs.items(): | |
|
690 | bytes_ref_name = safe_bytes(ref_name) | |
|
691 | ||
|
692 | if bytes_ref_name in refs: | |
|
693 | bytes_ref_hash = safe_bytes(ref_hash) | |
|
694 | determined.append(bytes_ref_hash) | |
|
695 | return determined | |
|
696 | ||
|
697 | # swap with our custom requested wants | |
|
593 | 698 | determine_wants = determine_wants_requested |
|
594 | 699 | |
|
595 | 700 | try: |
|
596 | 701 | remote_refs = client.fetch( |
|
597 | 702 | path=url, target=repo, determine_wants=determine_wants) |
|
703 | ||
|
598 | 704 | except NotGitRepository as e: |
|
599 | 705 | log.warning( |
|
600 | 706 | 'Trying to fetch from "%s" failed, not a Git repository.', url) |
@@ -619,18 +725,21 b' class GitRemote(RemoteBase):' | |||
|
619 | 725 | |
|
620 | 726 | if refs and not update_after: |
|
621 | 727 | # mikhail: explicitly set the head to the last ref. |
|
622 |
repo[ |
|
|
728 | repo[HEAD_MARKER] = remote_refs[refs[-1]] | |
|
623 | 729 | |
|
624 | 730 | if update_after: |
|
625 | # we want to checkout HEAD | |
|
626 |
repo[ |
|
|
731 | # we want to check out HEAD | |
|
732 | repo[HEAD_MARKER] = remote_refs[HEAD_MARKER] | |
|
627 | 733 | index.build_index_from_tree(repo.path, repo.index_path(), |
|
628 |
repo.object_store, repo[ |
|
|
734 | repo.object_store, repo[HEAD_MARKER].tree) | |
|
735 | ||
|
736 | if isinstance(remote_refs, FetchPackResult): | |
|
737 | return remote_refs.refs | |
|
629 | 738 | return remote_refs |
|
630 | 739 | |
|
631 | 740 | @reraise_safe_exceptions |
|
632 | 741 | def sync_fetch(self, wire, url, refs=None, all_refs=False): |
|
633 |
|
|
|
742 | self._factory.repo(wire) | |
|
634 | 743 | if refs and not isinstance(refs, (list, tuple)): |
|
635 | 744 | refs = [refs] |
|
636 | 745 | |
@@ -649,7 +758,7 b' class GitRemote(RemoteBase):' | |||
|
649 | 758 | fetch_refs = [] |
|
650 | 759 | |
|
651 | 760 | for ref_line in output.splitlines(): |
|
652 | sha, ref = ref_line.split('\t') | |
|
761 | sha, ref = ref_line.split(b'\t') | |
|
653 | 762 | sha = sha.strip() |
|
654 | 763 | if ref in remote_refs: |
|
655 | 764 | # duplicate, skip |
@@ -658,23 +767,23 b' class GitRemote(RemoteBase):' | |||
|
658 | 767 | log.debug("Skipping peeled reference %s", ref) |
|
659 | 768 | continue |
|
660 | 769 | # don't sync HEAD |
|
661 |
if ref in [ |
|
|
770 | if ref in [HEAD_MARKER]: | |
|
662 | 771 | continue |
|
663 | 772 | |
|
664 | 773 | remote_refs[ref] = sha |
|
665 | 774 | |
|
666 | 775 | if refs and sha in refs: |
|
667 | 776 | # we filter fetch using our specified refs |
|
668 |
fetch_refs.append('{ |
|
|
777 | fetch_refs.append(f'{safe_str(ref)}:{safe_str(ref)}') | |
|
669 | 778 | elif not refs: |
|
670 |
fetch_refs.append('{ |
|
|
779 | fetch_refs.append(f'{safe_str(ref)}:{safe_str(ref)}') | |
|
671 | 780 | log.debug('Finished obtaining fetch refs, total: %s', len(fetch_refs)) |
|
672 | 781 | |
|
673 | 782 | if fetch_refs: |
|
674 | 783 | for chunk in more_itertools.chunked(fetch_refs, 1024 * 4): |
|
675 | 784 | fetch_refs_chunks = list(chunk) |
|
676 | 785 | log.debug('Fetching %s refs from import url', len(fetch_refs_chunks)) |
|
677 |
|
|
|
786 | self.run_git_command( | |
|
678 | 787 | wire, ['fetch', url, '--force', '--prune', '--'] + fetch_refs_chunks, |
|
679 | 788 | fail_on_stderr=False, |
|
680 | 789 | _copts=self._remote_conf(config), |
@@ -711,17 +820,18 b' class GitRemote(RemoteBase):' | |||
|
711 | 820 | wire_remote = wire.copy() |
|
712 | 821 | wire_remote['path'] = path2 |
|
713 | 822 | repo_remote = self._factory.repo(wire_remote) |
|
714 |
LocalGitClient(thin_packs=False).fetch( |
|
|
823 | LocalGitClient(thin_packs=False).fetch(path2, repo_remote) | |
|
715 | 824 | |
|
716 | 825 | revs = [ |
|
717 | 826 | x.commit.id |
|
718 | for x in repo_remote.get_walker(include=[rev2], exclude=[rev1])] | |
|
827 | for x in repo_remote.get_walker(include=[safe_bytes(rev2)], exclude=[safe_bytes(rev1)])] | |
|
719 | 828 | return revs |
|
720 | 829 | |
|
721 | 830 | @reraise_safe_exceptions |
|
722 | 831 | def get_object(self, wire, sha, maybe_unreachable=False): |
|
723 | 832 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
724 | 833 | region = self._region(wire) |
|
834 | ||
|
725 | 835 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
726 | 836 | def _get_object(_context_uid, _repo_id, _sha): |
|
727 | 837 | repo_init = self._factory.repo_libgit2(wire) |
@@ -766,11 +876,11 b' class GitRemote(RemoteBase):' | |||
|
766 | 876 | raise exceptions.LookupException(e)(missing_commit_err) |
|
767 | 877 | |
|
768 | 878 | commit_id = commit.hex |
|
769 |
type_ |
|
|
879 | type_str = commit.type_str | |
|
770 | 880 | |
|
771 | 881 | return { |
|
772 | 882 | 'id': commit_id, |
|
773 |
'type': |
|
|
883 | 'type': type_str, | |
|
774 | 884 | 'commit_id': commit_id, |
|
775 | 885 | 'idx': 0 |
|
776 | 886 | } |
@@ -781,6 +891,7 b' class GitRemote(RemoteBase):' | |||
|
781 | 891 | def get_refs(self, wire): |
|
782 | 892 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
783 | 893 | region = self._region(wire) |
|
894 | ||
|
784 | 895 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
785 | 896 | def _get_refs(_context_uid, _repo_id): |
|
786 | 897 | |
@@ -788,7 +899,7 b' class GitRemote(RemoteBase):' | |||
|
788 | 899 | with repo_init as repo: |
|
789 | 900 | regex = re.compile('^refs/(heads|tags)/') |
|
790 | 901 | return {x.name: x.target.hex for x in |
|
791 |
|
|
|
902 | [ref for ref in repo.listall_reference_objects() if regex.match(ref.name)]} | |
|
792 | 903 | |
|
793 | 904 | return _get_refs(context_uid, repo_id) |
|
794 | 905 | |
@@ -796,13 +907,14 b' class GitRemote(RemoteBase):' | |||
|
796 | 907 | def get_branch_pointers(self, wire): |
|
797 | 908 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
798 | 909 | region = self._region(wire) |
|
910 | ||
|
799 | 911 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
800 | 912 | def _get_branch_pointers(_context_uid, _repo_id): |
|
801 | 913 | |
|
802 | 914 | repo_init = self._factory.repo_libgit2(wire) |
|
803 | 915 | regex = re.compile('^refs/heads') |
|
804 | 916 | with repo_init as repo: |
|
805 |
branches = |
|
|
917 | branches = [ref for ref in repo.listall_reference_objects() if regex.match(ref.name)] | |
|
806 | 918 | return {x.target.hex: x.shorthand for x in branches} |
|
807 | 919 | |
|
808 | 920 | return _get_branch_pointers(context_uid, repo_id) |
@@ -811,6 +923,7 b' class GitRemote(RemoteBase):' | |||
|
811 | 923 | def head(self, wire, show_exc=True): |
|
812 | 924 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
813 | 925 | region = self._region(wire) |
|
926 | ||
|
814 | 927 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
815 | 928 | def _head(_context_uid, _repo_id, _show_exc): |
|
816 | 929 | repo_init = self._factory.repo_libgit2(wire) |
@@ -824,12 +937,12 b' class GitRemote(RemoteBase):' | |||
|
824 | 937 | |
|
825 | 938 | @reraise_safe_exceptions |
|
826 | 939 | def init(self, wire): |
|
827 |
repo_path = s |
|
|
940 | repo_path = safe_str(wire['path']) | |
|
828 | 941 | self.repo = Repo.init(repo_path) |
|
829 | 942 | |
|
830 | 943 | @reraise_safe_exceptions |
|
831 | 944 | def init_bare(self, wire): |
|
832 |
repo_path = s |
|
|
945 | repo_path = safe_str(wire['path']) | |
|
833 | 946 | self.repo = Repo.init_bare(repo_path) |
|
834 | 947 | |
|
835 | 948 | @reraise_safe_exceptions |
@@ -837,6 +950,7 b' class GitRemote(RemoteBase):' | |||
|
837 | 950 | |
|
838 | 951 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
839 | 952 | region = self._region(wire) |
|
953 | ||
|
840 | 954 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
841 | 955 | def _revision(_context_uid, _repo_id, _rev): |
|
842 | 956 | repo_init = self._factory.repo_libgit2(wire) |
@@ -856,6 +970,7 b' class GitRemote(RemoteBase):' | |||
|
856 | 970 | def date(self, wire, commit_id): |
|
857 | 971 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
858 | 972 | region = self._region(wire) |
|
973 | ||
|
859 | 974 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
860 | 975 | def _date(_repo_id, _commit_id): |
|
861 | 976 | repo_init = self._factory.repo_libgit2(wire) |
@@ -876,6 +991,7 b' class GitRemote(RemoteBase):' | |||
|
876 | 991 | def author(self, wire, commit_id): |
|
877 | 992 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
878 | 993 | region = self._region(wire) |
|
994 | ||
|
879 | 995 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
880 | 996 | def _author(_repo_id, _commit_id): |
|
881 | 997 | repo_init = self._factory.repo_libgit2(wire) |
@@ -888,12 +1004,12 b' class GitRemote(RemoteBase):' | |||
|
888 | 1004 | author = commit.get_object().author |
|
889 | 1005 | |
|
890 | 1006 | if author.email: |
|
891 |
return |
|
|
1007 | return f"{author.name} <{author.email}>" | |
|
892 | 1008 | |
|
893 | 1009 | try: |
|
894 |
return |
|
|
1010 | return f"{author.name}" | |
|
895 | 1011 | except Exception: |
|
896 |
return |
|
|
1012 | return f"{safe_str(author.raw_name)}" | |
|
897 | 1013 | |
|
898 | 1014 | return _author(repo_id, commit_id) |
|
899 | 1015 | |
@@ -901,6 +1017,7 b' class GitRemote(RemoteBase):' | |||
|
901 | 1017 | def message(self, wire, commit_id): |
|
902 | 1018 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
903 | 1019 | region = self._region(wire) |
|
1020 | ||
|
904 | 1021 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
905 | 1022 | def _message(_repo_id, _commit_id): |
|
906 | 1023 | repo_init = self._factory.repo_libgit2(wire) |
@@ -913,6 +1030,7 b' class GitRemote(RemoteBase):' | |||
|
913 | 1030 | def parents(self, wire, commit_id): |
|
914 | 1031 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
915 | 1032 | region = self._region(wire) |
|
1033 | ||
|
916 | 1034 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
917 | 1035 | def _parents(_repo_id, _commit_id): |
|
918 | 1036 | repo_init = self._factory.repo_libgit2(wire) |
@@ -930,17 +1048,23 b' class GitRemote(RemoteBase):' | |||
|
930 | 1048 | def children(self, wire, commit_id): |
|
931 | 1049 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
932 | 1050 | region = self._region(wire) |
|
1051 | ||
|
1052 | head = self.head(wire) | |
|
1053 | ||
|
933 | 1054 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
934 | 1055 | def _children(_repo_id, _commit_id): |
|
1056 | ||
|
935 | 1057 | output, __ = self.run_git_command( |
|
936 | wire, ['rev-list', '--all', '--children']) | |
|
1058 | wire, ['rev-list', '--all', '--children', f'{commit_id}^..{head}']) | |
|
937 | 1059 | |
|
938 | 1060 | child_ids = [] |
|
939 |
pat = re.compile(r'^ |
|
|
940 | for l in output.splitlines(): | |
|
941 |
|
|
|
942 | found_ids = l.split(' ')[1:] | |
|
1061 | pat = re.compile(fr'^{commit_id}') | |
|
1062 | for line in output.splitlines(): | |
|
1063 | line = safe_str(line) | |
|
1064 | if pat.match(line): | |
|
1065 | found_ids = line.split(' ')[1:] | |
|
943 | 1066 | child_ids.extend(found_ids) |
|
1067 | break | |
|
944 | 1068 | |
|
945 | 1069 | return child_ids |
|
946 | 1070 | return _children(repo_id, commit_id) |
@@ -955,7 +1079,11 b' class GitRemote(RemoteBase):' | |||
|
955 | 1079 | def create_branch(self, wire, branch_name, commit_id, force=False): |
|
956 | 1080 | repo_init = self._factory.repo_libgit2(wire) |
|
957 | 1081 | with repo_init as repo: |
|
958 |
|
|
|
1082 | if commit_id: | |
|
1083 | commit = repo[commit_id] | |
|
1084 | else: | |
|
1085 | # if commit is not given just use the HEAD | |
|
1086 | commit = repo.head() | |
|
959 | 1087 | |
|
960 | 1088 | if force: |
|
961 | 1089 | repo.branches.local.create(branch_name, commit, force=force) |
@@ -973,23 +1101,39 b' class GitRemote(RemoteBase):' | |||
|
973 | 1101 | def tag_remove(self, wire, tag_name): |
|
974 | 1102 | repo_init = self._factory.repo_libgit2(wire) |
|
975 | 1103 | with repo_init as repo: |
|
976 |
key = 'refs/tags/{}' |
|
|
1104 | key = f'refs/tags/{tag_name}' | |
|
977 | 1105 | repo.references.delete(key) |
|
978 | 1106 | |
|
979 | 1107 | @reraise_safe_exceptions |
|
980 | 1108 | def tree_changes(self, wire, source_id, target_id): |
|
981 | # TODO(marcink): remove this seems it's only used by tests | |
|
982 | 1109 | repo = self._factory.repo(wire) |
|
1110 | # source can be empty | |
|
1111 | source_id = safe_bytes(source_id if source_id else b'') | |
|
1112 | target_id = safe_bytes(target_id) | |
|
1113 | ||
|
983 | 1114 | source = repo[source_id].tree if source_id else None |
|
984 | 1115 | target = repo[target_id].tree |
|
985 | 1116 | result = repo.object_store.tree_changes(source, target) |
|
986 | return list(result) | |
|
1117 | ||
|
1118 | added = set() | |
|
1119 | modified = set() | |
|
1120 | deleted = set() | |
|
1121 | for (old_path, new_path), (_, _), (_, _) in list(result): | |
|
1122 | if new_path and old_path: | |
|
1123 | modified.add(new_path) | |
|
1124 | elif new_path and not old_path: | |
|
1125 | added.add(new_path) | |
|
1126 | elif not new_path and old_path: | |
|
1127 | deleted.add(old_path) | |
|
1128 | ||
|
1129 | return list(added), list(modified), list(deleted) | |
|
987 | 1130 | |
|
988 | 1131 | @reraise_safe_exceptions |
|
989 | 1132 | def tree_and_type_for_path(self, wire, commit_id, path): |
|
990 | 1133 | |
|
991 | 1134 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
992 | 1135 | region = self._region(wire) |
|
1136 | ||
|
993 | 1137 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
994 | 1138 | def _tree_and_type_for_path(_context_uid, _repo_id, _commit_id, _path): |
|
995 | 1139 | repo_init = self._factory.repo_libgit2(wire) |
@@ -1001,13 +1145,14 b' class GitRemote(RemoteBase):' | |||
|
1001 | 1145 | except KeyError: |
|
1002 | 1146 | return None, None, None |
|
1003 | 1147 | |
|
1004 | return tree.id.hex, tree.type, tree.filemode | |
|
1148 | return tree.id.hex, tree.type_str, tree.filemode | |
|
1005 | 1149 | return _tree_and_type_for_path(context_uid, repo_id, commit_id, path) |
|
1006 | 1150 | |
|
1007 | 1151 | @reraise_safe_exceptions |
|
1008 | 1152 | def tree_items(self, wire, tree_id): |
|
1009 | 1153 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
1010 | 1154 | region = self._region(wire) |
|
1155 | ||
|
1011 | 1156 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
1012 | 1157 | def _tree_items(_repo_id, _tree_id): |
|
1013 | 1158 | |
@@ -1016,13 +1161,13 b' class GitRemote(RemoteBase):' | |||
|
1016 | 1161 | try: |
|
1017 | 1162 | tree = repo[tree_id] |
|
1018 | 1163 | except KeyError: |
|
1019 |
raise ObjectMissing('No tree with id: {}' |
|
|
1164 | raise ObjectMissing(f'No tree with id: {tree_id}') | |
|
1020 | 1165 | |
|
1021 | 1166 | result = [] |
|
1022 | 1167 | for item in tree: |
|
1023 | 1168 | item_sha = item.hex |
|
1024 | 1169 | item_mode = item.filemode |
|
1025 | item_type = item.type | |
|
1170 | item_type = item.type_str | |
|
1026 | 1171 | |
|
1027 | 1172 | if item_type == 'commit': |
|
1028 | 1173 | # NOTE(marcink): submodules we translate to 'link' for backward compat |
@@ -1066,7 +1211,7 b' class GitRemote(RemoteBase):' | |||
|
1066 | 1211 | lines = diff.splitlines() |
|
1067 | 1212 | x = 0 |
|
1068 | 1213 | for line in lines: |
|
1069 | if line.startswith('diff'): | |
|
1214 | if line.startswith(b'diff'): | |
|
1070 | 1215 | break |
|
1071 | 1216 | x += 1 |
|
1072 | 1217 | # Append new line just like 'diff' command do |
@@ -1076,6 +1221,7 b' class GitRemote(RemoteBase):' | |||
|
1076 | 1221 | @reraise_safe_exceptions |
|
1077 | 1222 | def diff(self, wire, commit_id_1, commit_id_2, file_filter, opt_ignorews, context): |
|
1078 | 1223 | repo_init = self._factory.repo_libgit2(wire) |
|
1224 | ||
|
1079 | 1225 | with repo_init as repo: |
|
1080 | 1226 | swap = True |
|
1081 | 1227 | flags = 0 |
@@ -1101,15 +1247,17 b' class GitRemote(RemoteBase):' | |||
|
1101 | 1247 | if file_filter: |
|
1102 | 1248 | for p in diff_obj: |
|
1103 | 1249 | if p.delta.old_file.path == file_filter: |
|
1104 |
return p. |
|
|
1250 | return BytesEnvelope(p.data) or BytesEnvelope(b'') | |
|
1105 | 1251 | # fo matching path == no diff |
|
1106 | return '' | |
|
1107 | return diff_obj.patch or '' | |
|
1252 | return BytesEnvelope(b'') | |
|
1253 | ||
|
1254 | return BytesEnvelope(safe_bytes(diff_obj.patch)) or BytesEnvelope(b'') | |
|
1108 | 1255 | |
|
1109 | 1256 | @reraise_safe_exceptions |
|
1110 | 1257 | def node_history(self, wire, commit_id, path, limit): |
|
1111 | 1258 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
1112 | 1259 | region = self._region(wire) |
|
1260 | ||
|
1113 | 1261 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
1114 | 1262 | def _node_history(_context_uid, _repo_id, _commit_id, _path, _limit): |
|
1115 | 1263 | # optimize for n==1, rev-list is much faster for that use-case |
@@ -1122,14 +1270,14 b' class GitRemote(RemoteBase):' | |||
|
1122 | 1270 | cmd.extend(['--pretty=format: %H', '-s', commit_id, '--', path]) |
|
1123 | 1271 | |
|
1124 | 1272 | output, __ = self.run_git_command(wire, cmd) |
|
1125 | commit_ids = re.findall(r'[0-9a-fA-F]{40}', output) | |
|
1273 | commit_ids = re.findall(rb'[0-9a-fA-F]{40}', output) | |
|
1126 | 1274 | |
|
1127 | 1275 | return [x for x in commit_ids] |
|
1128 | 1276 | return _node_history(context_uid, repo_id, commit_id, path, limit) |
|
1129 | 1277 | |
|
1130 | 1278 | @reraise_safe_exceptions |
|
1131 | def node_annotate(self, wire, commit_id, path): | |
|
1132 | ||
|
1279 | def node_annotate_legacy(self, wire, commit_id, path): | |
|
1280 | # note: replaced by pygit2 implementation | |
|
1133 | 1281 | cmd = ['blame', '-l', '--root', '-r', commit_id, '--', path] |
|
1134 | 1282 | # -l ==> outputs long shas (and we need all 40 characters) |
|
1135 | 1283 | # --root ==> doesn't put '^' character for boundaries |
@@ -1137,13 +1285,31 b' class GitRemote(RemoteBase):' | |||
|
1137 | 1285 | output, __ = self.run_git_command(wire, cmd) |
|
1138 | 1286 | |
|
1139 | 1287 | result = [] |
|
1140 |
for i, blame_line in enumerate(output.split( |
|
|
1288 | for i, blame_line in enumerate(output.splitlines()[:-1]): | |
|
1141 | 1289 | line_no = i + 1 |
|
1142 | commit_id, line = re.split(r' ', blame_line, 1) | |
|
1143 | result.append((line_no, commit_id, line)) | |
|
1290 | blame_commit_id, line = re.split(rb' ', blame_line, 1) | |
|
1291 | result.append((line_no, blame_commit_id, line)) | |
|
1292 | ||
|
1144 | 1293 | return result |
|
1145 | 1294 | |
|
1146 | 1295 | @reraise_safe_exceptions |
|
1296 | def node_annotate(self, wire, commit_id, path): | |
|
1297 | ||
|
1298 | result_libgit = [] | |
|
1299 | repo_init = self._factory.repo_libgit2(wire) | |
|
1300 | with repo_init as repo: | |
|
1301 | commit = repo[commit_id] | |
|
1302 | blame_obj = repo.blame(path, newest_commit=commit_id) | |
|
1303 | for i, line in enumerate(commit.tree[path].data.splitlines()): | |
|
1304 | line_no = i + 1 | |
|
1305 | hunk = blame_obj.for_line(line_no) | |
|
1306 | blame_commit_id = hunk.final_commit_id.hex | |
|
1307 | ||
|
1308 | result_libgit.append((line_no, blame_commit_id, line)) | |
|
1309 | ||
|
1310 | return result_libgit | |
|
1311 | ||
|
1312 | @reraise_safe_exceptions | |
|
1147 | 1313 | def update_server_info(self, wire): |
|
1148 | 1314 | repo = self._factory.repo(wire) |
|
1149 | 1315 | update_server_info(repo) |
@@ -1153,6 +1319,7 b' class GitRemote(RemoteBase):' | |||
|
1153 | 1319 | |
|
1154 | 1320 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
1155 | 1321 | region = self._region(wire) |
|
1322 | ||
|
1156 | 1323 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
1157 | 1324 | def _get_all_commit_ids(_context_uid, _repo_id): |
|
1158 | 1325 | |
@@ -1163,6 +1330,16 b' class GitRemote(RemoteBase):' | |||
|
1163 | 1330 | except Exception: |
|
1164 | 1331 | # Can be raised for empty repositories |
|
1165 | 1332 | return [] |
|
1333 | ||
|
1334 | @region.conditional_cache_on_arguments(condition=cache_on) | |
|
1335 | def _get_all_commit_ids_pygit2(_context_uid, _repo_id): | |
|
1336 | repo_init = self._factory.repo_libgit2(wire) | |
|
1337 | from pygit2 import GIT_SORT_REVERSE, GIT_SORT_TIME, GIT_BRANCH_ALL | |
|
1338 | results = [] | |
|
1339 | with repo_init as repo: | |
|
1340 | for commit in repo.walk(repo.head.target, GIT_SORT_TIME | GIT_BRANCH_ALL | GIT_SORT_REVERSE): | |
|
1341 | results.append(commit.id.hex) | |
|
1342 | ||
|
1166 | 1343 | return _get_all_commit_ids(context_uid, repo_id) |
|
1167 | 1344 | |
|
1168 | 1345 | @reraise_safe_exceptions |
@@ -1176,7 +1353,7 b' class GitRemote(RemoteBase):' | |||
|
1176 | 1353 | _copts = [] |
|
1177 | 1354 | del opts['_bare'] |
|
1178 | 1355 | else: |
|
1179 |
_copts = ['-c', 'core.quotepath=false', |
|
|
1356 | _copts = ['-c', 'core.quotepath=false',] | |
|
1180 | 1357 | safe_call = False |
|
1181 | 1358 | if '_safe' in opts: |
|
1182 | 1359 | # no exc on failure |
@@ -1203,9 +1380,9 b' class GitRemote(RemoteBase):' | |||
|
1203 | 1380 | _opts.update(opts) |
|
1204 | 1381 | proc = subprocessio.SubprocessIOChunker(cmd, **_opts) |
|
1205 | 1382 | |
|
1206 |
return ''.join(proc), ''.join(proc.err |
|
|
1207 |
except |
|
|
1208 | cmd = ' '.join(cmd) # human friendly CMD | |
|
1383 | return b''.join(proc), b''.join(proc.stderr) | |
|
1384 | except OSError as err: | |
|
1385 | cmd = ' '.join(map(safe_str, cmd)) # human friendly CMD | |
|
1209 | 1386 | tb_err = ("Couldn't run git command (%s).\n" |
|
1210 | 1387 | "Original error was:%s\n" |
|
1211 | 1388 | "Call options:%s\n" |
@@ -1224,6 +1401,9 b' class GitRemote(RemoteBase):' | |||
|
1224 | 1401 | from vcsserver.hook_utils import install_git_hooks |
|
1225 | 1402 | bare = self.bare(wire) |
|
1226 | 1403 | path = wire['path'] |
|
1404 | binary_dir = settings.BINARY_DIR | |
|
1405 | if binary_dir: | |
|
1406 | os.path.join(binary_dir, 'python3') | |
|
1227 | 1407 | return install_git_hooks(path, bare, force_create=force) |
|
1228 | 1408 | |
|
1229 | 1409 | @reraise_safe_exceptions |
@@ -1240,13 +1420,15 b' class GitRemote(RemoteBase):' | |||
|
1240 | 1420 | @reraise_safe_exceptions |
|
1241 | 1421 | def set_head_ref(self, wire, head_name): |
|
1242 | 1422 | log.debug('Setting refs/head to `%s`', head_name) |
|
1243 | cmd = ['symbolic-ref', '"HEAD"', '"refs/heads/%s"' % head_name] | |
|
1244 | output, __ = self.run_git_command(wire, cmd) | |
|
1245 | return [head_name] + output.splitlines() | |
|
1423 | repo_init = self._factory.repo_libgit2(wire) | |
|
1424 | with repo_init as repo: | |
|
1425 | repo.set_head(f'refs/heads/{head_name}') | |
|
1426 | ||
|
1427 | return [head_name] + [f'set HEAD to refs/heads/{head_name}'] | |
|
1246 | 1428 | |
|
1247 | 1429 | @reraise_safe_exceptions |
|
1248 |
def archive_repo(self, wire, archive_ |
|
|
1249 | archive_dir_name, commit_id): | |
|
1430 | def archive_repo(self, wire, archive_name_key, kind, mtime, archive_at_path, | |
|
1431 | archive_dir_name, commit_id, cache_config): | |
|
1250 | 1432 | |
|
1251 | 1433 | def file_walker(_commit_id, path): |
|
1252 | 1434 | repo_init = self._factory.repo_libgit2(wire) |
@@ -1262,20 +1444,20 b' class GitRemote(RemoteBase):' | |||
|
1262 | 1444 | try: |
|
1263 | 1445 | tree = repo[tree_id] |
|
1264 | 1446 | except KeyError: |
|
1265 |
raise ObjectMissing('No tree with id: {}' |
|
|
1447 | raise ObjectMissing(f'No tree with id: {tree_id}') | |
|
1266 | 1448 | |
|
1267 | 1449 | index = LibGit2Index.Index() |
|
1268 | 1450 | index.read_tree(tree) |
|
1269 | 1451 | file_iter = index |
|
1270 | 1452 | |
|
1271 | for fn in file_iter: | |
|
1272 | file_path = fn.path | |
|
1273 | mode = fn.mode | |
|
1453 | for file_node in file_iter: | |
|
1454 | file_path = file_node.path | |
|
1455 | mode = file_node.mode | |
|
1274 | 1456 | is_link = stat.S_ISLNK(mode) |
|
1275 | 1457 | if mode == pygit2.GIT_FILEMODE_COMMIT: |
|
1276 | 1458 | log.debug('Skipping path %s as a commit node', file_path) |
|
1277 | 1459 | continue |
|
1278 | yield ArchiveNode(file_path, mode, is_link, repo[fn.hex].read_raw) | |
|
1460 | yield ArchiveNode(file_path, mode, is_link, repo[file_node.hex].read_raw) | |
|
1279 | 1461 | |
|
1280 | return archive_repo(file_walker, archive_dest_path, kind, mtime, archive_at_path, | |
|
1281 | archive_dir_name, commit_id) | |
|
1462 | return store_archive_in_cache( | |
|
1463 | file_walker, archive_name_key, kind, mtime, archive_at_path, archive_dir_name, commit_id, cache_config=cache_config) |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -14,17 +14,17 b'' | |||
|
14 | 14 | # You should have received a copy of the GNU General Public License |
|
15 | 15 | # along with this program; if not, write to the Free Software Foundation, |
|
16 | 16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
17 | import functools | |
|
17 | import binascii | |
|
18 | 18 | import io |
|
19 | 19 | import logging |
|
20 | import os | |
|
21 | 20 | import stat |
|
22 | import urllib | |
|
23 |
import urllib |
|
|
21 | import urllib.request | |
|
22 | import urllib.parse | |
|
24 | 23 | import traceback |
|
24 | import hashlib | |
|
25 | 25 | |
|
26 | 26 | from hgext import largefiles, rebase, purge |
|
27 | from hgext.strip import strip as hgext_strip | |
|
27 | ||
|
28 | 28 | from mercurial import commands |
|
29 | 29 | from mercurial import unionrepo |
|
30 | 30 | from mercurial import verify |
@@ -32,15 +32,19 b' from mercurial import repair' | |||
|
32 | 32 | |
|
33 | 33 | import vcsserver |
|
34 | 34 | from vcsserver import exceptions |
|
35 |
from vcsserver.base import RepoFactory, obfuscate_qs, raise_from_original, archive_ |
|
|
35 | from vcsserver.base import RepoFactory, obfuscate_qs, raise_from_original, store_archive_in_cache, ArchiveNode, BytesEnvelope, \ | |
|
36 | BinaryEnvelope | |
|
36 | 37 | from vcsserver.hgcompat import ( |
|
37 | 38 | archival, bin, clone, config as hgconfig, diffopts, hex, get_ctx, |
|
38 | 39 | hg_url as url_parser, httpbasicauthhandler, httpdigestauthhandler, |
|
39 | 40 | makepeer, instance, match, memctx, exchange, memfilectx, nullrev, hg_merge, |
|
40 | 41 | patch, peer, revrange, ui, hg_tag, Abort, LookupError, RepoError, |
|
41 | 42 | RepoLookupError, InterventionRequired, RequirementError, |
|
42 | alwaysmatcher, patternmatcher, hgutil) | |
|
43 | alwaysmatcher, patternmatcher, hgutil, hgext_strip) | |
|
44 | from vcsserver.str_utils import ascii_bytes, ascii_str, safe_str, safe_bytes | |
|
43 | 45 | from vcsserver.vcs_base import RemoteBase |
|
46 | from vcsserver.config import hooks as hooks_config | |
|
47 | ||
|
44 | 48 | |
|
45 | 49 | log = logging.getLogger(__name__) |
|
46 | 50 | |
@@ -48,25 +52,31 b' log = logging.getLogger(__name__)' | |||
|
48 | 52 | def make_ui_from_config(repo_config): |
|
49 | 53 | |
|
50 | 54 | class LoggingUI(ui.ui): |
|
55 | ||
|
51 | 56 | def status(self, *msg, **opts): |
|
52 | log.info(' '.join(msg).rstrip('\n')) | |
|
53 | super(LoggingUI, self).status(*msg, **opts) | |
|
57 | str_msg = map(safe_str, msg) | |
|
58 | log.info(' '.join(str_msg).rstrip('\n')) | |
|
59 | #super(LoggingUI, self).status(*msg, **opts) | |
|
54 | 60 | |
|
55 | 61 | def warn(self, *msg, **opts): |
|
56 | log.warn(' '.join(msg).rstrip('\n')) | |
|
57 | super(LoggingUI, self).warn(*msg, **opts) | |
|
62 | str_msg = map(safe_str, msg) | |
|
63 | log.warning('ui_logger:'+' '.join(str_msg).rstrip('\n')) | |
|
64 | #super(LoggingUI, self).warn(*msg, **opts) | |
|
58 | 65 | |
|
59 | 66 | def error(self, *msg, **opts): |
|
60 | log.error(' '.join(msg).rstrip('\n')) | |
|
61 | super(LoggingUI, self).error(*msg, **opts) | |
|
67 | str_msg = map(safe_str, msg) | |
|
68 | log.error('ui_logger:'+' '.join(str_msg).rstrip('\n')) | |
|
69 | #super(LoggingUI, self).error(*msg, **opts) | |
|
62 | 70 | |
|
63 | 71 | def note(self, *msg, **opts): |
|
64 | log.info(' '.join(msg).rstrip('\n')) | |
|
65 | super(LoggingUI, self).note(*msg, **opts) | |
|
72 | str_msg = map(safe_str, msg) | |
|
73 | log.info('ui_logger:'+' '.join(str_msg).rstrip('\n')) | |
|
74 | #super(LoggingUI, self).note(*msg, **opts) | |
|
66 | 75 | |
|
67 | 76 | def debug(self, *msg, **opts): |
|
68 | log.debug(' '.join(msg).rstrip('\n')) | |
|
69 | super(LoggingUI, self).debug(*msg, **opts) | |
|
77 | str_msg = map(safe_str, msg) | |
|
78 | log.debug('ui_logger:'+' '.join(str_msg).rstrip('\n')) | |
|
79 | #super(LoggingUI, self).debug(*msg, **opts) | |
|
70 | 80 | |
|
71 | 81 | baseui = LoggingUI() |
|
72 | 82 | |
@@ -76,26 +86,26 b' def make_ui_from_config(repo_config):' | |||
|
76 | 86 | baseui._tcfg = hgconfig.config() |
|
77 | 87 | |
|
78 | 88 | for section, option, value in repo_config: |
|
79 | baseui.setconfig(section, option, value) | |
|
89 | baseui.setconfig(ascii_bytes(section), ascii_bytes(option), ascii_bytes(value)) | |
|
80 | 90 | |
|
81 | 91 | # make our hgweb quiet so it doesn't print output |
|
82 | baseui.setconfig('ui', 'quiet', 'true') | |
|
92 | baseui.setconfig(b'ui', b'quiet', b'true') | |
|
83 | 93 | |
|
84 | baseui.setconfig('ui', 'paginate', 'never') | |
|
94 | baseui.setconfig(b'ui', b'paginate', b'never') | |
|
85 | 95 | # for better Error reporting of Mercurial |
|
86 | baseui.setconfig('ui', 'message-output', 'stderr') | |
|
96 | baseui.setconfig(b'ui', b'message-output', b'stderr') | |
|
87 | 97 | |
|
88 | 98 | # force mercurial to only use 1 thread, otherwise it may try to set a |
|
89 | 99 | # signal in a non-main thread, thus generating a ValueError. |
|
90 | baseui.setconfig('worker', 'numcpus', 1) | |
|
100 | baseui.setconfig(b'worker', b'numcpus', 1) | |
|
91 | 101 | |
|
92 | 102 | # If there is no config for the largefiles extension, we explicitly disable |
|
93 | 103 | # it here. This overrides settings from repositories hgrc file. Recent |
|
94 | 104 | # mercurial versions enable largefiles in hgrc on clone from largefile |
|
95 | 105 | # repo. |
|
96 | if not baseui.hasconfig('extensions', 'largefiles'): | |
|
106 | if not baseui.hasconfig(b'extensions', b'largefiles'): | |
|
97 | 107 | log.debug('Explicitly disable largefiles extension for repo.') |
|
98 | baseui.setconfig('extensions', 'largefiles', '!') | |
|
108 | baseui.setconfig(b'extensions', b'largefiles', b'!') | |
|
99 | 109 | |
|
100 | 110 | return baseui |
|
101 | 111 | |
@@ -107,19 +117,19 b' def reraise_safe_exceptions(func):' | |||
|
107 | 117 | try: |
|
108 | 118 | return func(*args, **kwargs) |
|
109 | 119 | except (Abort, InterventionRequired) as e: |
|
110 | raise_from_original(exceptions.AbortException(e)) | |
|
120 | raise_from_original(exceptions.AbortException(e), e) | |
|
111 | 121 | except RepoLookupError as e: |
|
112 | raise_from_original(exceptions.LookupException(e)) | |
|
122 | raise_from_original(exceptions.LookupException(e), e) | |
|
113 | 123 | except RequirementError as e: |
|
114 | raise_from_original(exceptions.RequirementException(e)) | |
|
124 | raise_from_original(exceptions.RequirementException(e), e) | |
|
115 | 125 | except RepoError as e: |
|
116 | raise_from_original(exceptions.VcsException(e)) | |
|
126 | raise_from_original(exceptions.VcsException(e), e) | |
|
117 | 127 | except LookupError as e: |
|
118 | raise_from_original(exceptions.LookupException(e)) | |
|
128 | raise_from_original(exceptions.LookupException(e), e) | |
|
119 | 129 | except Exception as e: |
|
120 | 130 | if not hasattr(e, '_vcs_kind'): |
|
121 | 131 | log.exception("Unhandled exception in hg remote call") |
|
122 | raise_from_original(exceptions.UnhandledException(e)) | |
|
132 | raise_from_original(exceptions.UnhandledException(e), e) | |
|
123 | 133 | |
|
124 | 134 | raise |
|
125 | 135 | return wrapper |
@@ -130,9 +140,18 b' class MercurialFactory(RepoFactory):' | |||
|
130 | 140 | |
|
131 | 141 | def _create_config(self, config, hooks=True): |
|
132 | 142 | if not hooks: |
|
133 | hooks_to_clean = frozenset(( | |
|
134 | 'changegroup.repo_size', 'preoutgoing.pre_pull', | |
|
135 | 'outgoing.pull_logger', 'prechangegroup.pre_push')) | |
|
143 | ||
|
144 | hooks_to_clean = { | |
|
145 | ||
|
146 | hooks_config.HOOK_REPO_SIZE, | |
|
147 | hooks_config.HOOK_PRE_PULL, | |
|
148 | hooks_config.HOOK_PULL, | |
|
149 | ||
|
150 | hooks_config.HOOK_PRE_PUSH, | |
|
151 | # TODO: what about PRETXT, this was disabled in pre 5.0.0 | |
|
152 | hooks_config.HOOK_PRETX_PUSH, | |
|
153 | ||
|
154 | } | |
|
136 | 155 | new_config = [] |
|
137 | 156 | for section, option, value in config: |
|
138 | 157 | if section == 'hooks' and option in hooks_to_clean: |
@@ -145,7 +164,9 b' class MercurialFactory(RepoFactory):' | |||
|
145 | 164 | |
|
146 | 165 | def _create_repo(self, wire, create): |
|
147 | 166 | baseui = self._create_config(wire["config"]) |
|
148 |
re |
|
|
167 | repo = instance(baseui, safe_bytes(wire["path"]), create) | |
|
168 | log.debug('repository created: got HG object: %s', repo) | |
|
169 | return repo | |
|
149 | 170 | |
|
150 | 171 | def repo(self, wire, create=False): |
|
151 | 172 | """ |
@@ -155,7 +176,7 b' class MercurialFactory(RepoFactory):' | |||
|
155 | 176 | |
|
156 | 177 | |
|
157 | 178 | def patch_ui_message_output(baseui): |
|
158 | baseui.setconfig('ui', 'quiet', 'false') | |
|
179 | baseui.setconfig(b'ui', b'quiet', b'false') | |
|
159 | 180 | output = io.BytesIO() |
|
160 | 181 | |
|
161 | 182 | def write(data, **unused_kwargs): |
@@ -169,6 +190,22 b' def patch_ui_message_output(baseui):' | |||
|
169 | 190 | return baseui, output |
|
170 | 191 | |
|
171 | 192 | |
|
193 | def get_obfuscated_url(url_obj): | |
|
194 | url_obj.passwd = b'*****' if url_obj.passwd else url_obj.passwd | |
|
195 | url_obj.query = obfuscate_qs(url_obj.query) | |
|
196 | obfuscated_uri = str(url_obj) | |
|
197 | return obfuscated_uri | |
|
198 | ||
|
199 | ||
|
200 | def normalize_url_for_hg(url: str): | |
|
201 | _proto = None | |
|
202 | ||
|
203 | if '+' in url[:url.find('://')]: | |
|
204 | _proto = url[0:url.find('+')] | |
|
205 | url = url[url.find('+') + 1:] | |
|
206 | return url, _proto | |
|
207 | ||
|
208 | ||
|
172 | 209 | class HgRemote(RemoteBase): |
|
173 | 210 | |
|
174 | 211 | def __init__(self, factory): |
@@ -187,6 +224,13 b' class HgRemote(RemoteBase):' | |||
|
187 | 224 | "hidden": self.ctx_hidden, |
|
188 | 225 | "_file_paths": self.ctx_list, |
|
189 | 226 | } |
|
227 | self._bulk_file_methods = { | |
|
228 | "size": self.fctx_size, | |
|
229 | "data": self.fctx_node_data, | |
|
230 | "flags": self.fctx_flags, | |
|
231 | "is_binary": self.is_binary, | |
|
232 | "md5": self.md5_hash, | |
|
233 | } | |
|
190 | 234 | |
|
191 | 235 | def _get_ctx(self, repo, ref): |
|
192 | 236 | return get_ctx(repo, ref) |
@@ -194,7 +238,7 b' class HgRemote(RemoteBase):' | |||
|
194 | 238 | @reraise_safe_exceptions |
|
195 | 239 | def discover_hg_version(self): |
|
196 | 240 | from mercurial import util |
|
197 | return util.version() | |
|
241 | return safe_str(util.version()) | |
|
198 | 242 | |
|
199 | 243 | @reraise_safe_exceptions |
|
200 | 244 | def is_empty(self, wire): |
@@ -210,10 +254,11 b' class HgRemote(RemoteBase):' | |||
|
210 | 254 | def bookmarks(self, wire): |
|
211 | 255 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
212 | 256 | region = self._region(wire) |
|
257 | ||
|
213 | 258 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
214 | 259 | def _bookmarks(_context_uid, _repo_id): |
|
215 | 260 | repo = self._factory.repo(wire) |
|
216 | return dict(repo._bookmarks) | |
|
261 | return {safe_str(name): ascii_str(hex(sha)) for name, sha in repo._bookmarks.items()} | |
|
217 | 262 | |
|
218 | 263 | return _bookmarks(context_uid, repo_id) |
|
219 | 264 | |
@@ -221,16 +266,17 b' class HgRemote(RemoteBase):' | |||
|
221 | 266 | def branches(self, wire, normal, closed): |
|
222 | 267 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
223 | 268 | region = self._region(wire) |
|
269 | ||
|
224 | 270 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
225 | 271 | def _branches(_context_uid, _repo_id, _normal, _closed): |
|
226 | 272 | repo = self._factory.repo(wire) |
|
227 | 273 | iter_branches = repo.branchmap().iterbranches() |
|
228 | 274 | bt = {} |
|
229 | for branch_name, _heads, tip, is_closed in iter_branches: | |
|
275 | for branch_name, _heads, tip_node, is_closed in iter_branches: | |
|
230 | 276 | if normal and not is_closed: |
|
231 | bt[branch_name] = tip | |
|
277 | bt[safe_str(branch_name)] = ascii_str(hex(tip_node)) | |
|
232 | 278 | if closed and is_closed: |
|
233 | bt[branch_name] = tip | |
|
279 | bt[safe_str(branch_name)] = ascii_str(hex(tip_node)) | |
|
234 | 280 | |
|
235 | 281 | return bt |
|
236 | 282 | |
@@ -240,12 +286,14 b' class HgRemote(RemoteBase):' | |||
|
240 | 286 | def bulk_request(self, wire, commit_id, pre_load): |
|
241 | 287 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
242 | 288 | region = self._region(wire) |
|
289 | ||
|
243 | 290 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
244 | 291 | def _bulk_request(_repo_id, _commit_id, _pre_load): |
|
245 | 292 | result = {} |
|
246 | 293 | for attr in pre_load: |
|
247 | 294 | try: |
|
248 | 295 | method = self._bulk_methods[attr] |
|
296 | wire.update({'cache': False}) # disable cache for bulk calls so we don't double cache | |
|
249 | 297 | result[attr] = method(wire, commit_id) |
|
250 | 298 | except KeyError as e: |
|
251 | 299 | raise exceptions.VcsException(e)( |
@@ -258,6 +306,7 b' class HgRemote(RemoteBase):' | |||
|
258 | 306 | def ctx_branch(self, wire, commit_id): |
|
259 | 307 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
260 | 308 | region = self._region(wire) |
|
309 | ||
|
261 | 310 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
262 | 311 | def _ctx_branch(_repo_id, _commit_id): |
|
263 | 312 | repo = self._factory.repo(wire) |
@@ -269,6 +318,7 b' class HgRemote(RemoteBase):' | |||
|
269 | 318 | def ctx_date(self, wire, commit_id): |
|
270 | 319 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
271 | 320 | region = self._region(wire) |
|
321 | ||
|
272 | 322 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
273 | 323 | def _ctx_date(_repo_id, _commit_id): |
|
274 | 324 | repo = self._factory.repo(wire) |
@@ -286,6 +336,7 b' class HgRemote(RemoteBase):' | |||
|
286 | 336 | def ctx_files(self, wire, commit_id): |
|
287 | 337 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
288 | 338 | region = self._region(wire) |
|
339 | ||
|
289 | 340 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
290 | 341 | def _ctx_files(_repo_id, _commit_id): |
|
291 | 342 | repo = self._factory.repo(wire) |
@@ -304,6 +355,7 b' class HgRemote(RemoteBase):' | |||
|
304 | 355 | def ctx_parents(self, wire, commit_id): |
|
305 | 356 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
306 | 357 | region = self._region(wire) |
|
358 | ||
|
307 | 359 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
308 | 360 | def _ctx_parents(_repo_id, _commit_id): |
|
309 | 361 | repo = self._factory.repo(wire) |
@@ -317,6 +369,7 b' class HgRemote(RemoteBase):' | |||
|
317 | 369 | def ctx_children(self, wire, commit_id): |
|
318 | 370 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
319 | 371 | region = self._region(wire) |
|
372 | ||
|
320 | 373 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
321 | 374 | def _ctx_children(_repo_id, _commit_id): |
|
322 | 375 | repo = self._factory.repo(wire) |
@@ -330,6 +383,7 b' class HgRemote(RemoteBase):' | |||
|
330 | 383 | def ctx_phase(self, wire, commit_id): |
|
331 | 384 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
332 | 385 | region = self._region(wire) |
|
386 | ||
|
333 | 387 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
334 | 388 | def _ctx_phase(_context_uid, _repo_id, _commit_id): |
|
335 | 389 | repo = self._factory.repo(wire) |
@@ -342,6 +396,7 b' class HgRemote(RemoteBase):' | |||
|
342 | 396 | def ctx_obsolete(self, wire, commit_id): |
|
343 | 397 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
344 | 398 | region = self._region(wire) |
|
399 | ||
|
345 | 400 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
346 | 401 | def _ctx_obsolete(_context_uid, _repo_id, _commit_id): |
|
347 | 402 | repo = self._factory.repo(wire) |
@@ -353,6 +408,7 b' class HgRemote(RemoteBase):' | |||
|
353 | 408 | def ctx_hidden(self, wire, commit_id): |
|
354 | 409 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
355 | 410 | region = self._region(wire) |
|
411 | ||
|
356 | 412 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
357 | 413 | def _ctx_hidden(_context_uid, _repo_id, _commit_id): |
|
358 | 414 | repo = self._factory.repo(wire) |
@@ -384,46 +440,42 b' class HgRemote(RemoteBase):' | |||
|
384 | 440 | |
|
385 | 441 | @reraise_safe_exceptions |
|
386 | 442 | def check_url(self, url, config): |
|
387 | _proto = None | |
|
388 | if '+' in url[:url.find('://')]: | |
|
389 | _proto = url[0:url.find('+')] | |
|
390 | url = url[url.find('+') + 1:] | |
|
443 | url, _proto = normalize_url_for_hg(url) | |
|
444 | url_obj = url_parser(safe_bytes(url)) | |
|
445 | ||
|
446 | test_uri = safe_str(url_obj.authinfo()[0]) | |
|
447 | authinfo = url_obj.authinfo()[1] | |
|
448 | obfuscated_uri = get_obfuscated_url(url_obj) | |
|
449 | log.info("Checking URL for remote cloning/import: %s", obfuscated_uri) | |
|
450 | ||
|
391 | 451 | handlers = [] |
|
392 | url_obj = url_parser(url) | |
|
393 | test_uri, authinfo = url_obj.authinfo() | |
|
394 | url_obj.passwd = '*****' if url_obj.passwd else url_obj.passwd | |
|
395 | url_obj.query = obfuscate_qs(url_obj.query) | |
|
396 | ||
|
397 | cleaned_uri = str(url_obj) | |
|
398 | log.info("Checking URL for remote cloning/import: %s", cleaned_uri) | |
|
399 | ||
|
400 | 452 | if authinfo: |
|
401 | 453 | # create a password manager |
|
402 |
passmgr = urllib |
|
|
454 | passmgr = urllib.request.HTTPPasswordMgrWithDefaultRealm() | |
|
403 | 455 | passmgr.add_password(*authinfo) |
|
404 | 456 | |
|
405 | 457 | handlers.extend((httpbasicauthhandler(passmgr), |
|
406 | 458 | httpdigestauthhandler(passmgr))) |
|
407 | 459 | |
|
408 |
o = urllib |
|
|
460 | o = urllib.request.build_opener(*handlers) | |
|
409 | 461 | o.addheaders = [('Content-Type', 'application/mercurial-0.1'), |
|
410 | 462 | ('Accept', 'application/mercurial-0.1')] |
|
411 | 463 | |
|
412 | 464 | q = {"cmd": 'between'} |
|
413 |
q.update({'pairs': " |
|
|
414 | qs = '?%s' % urllib.urlencode(q) | |
|
415 |
cu = " |
|
|
416 |
req = urllib |
|
|
465 | q.update({'pairs': "{}-{}".format('0' * 40, '0' * 40)}) | |
|
466 | qs = '?%s' % urllib.parse.urlencode(q) | |
|
467 | cu = f"{test_uri}{qs}" | |
|
468 | req = urllib.request.Request(cu, None, {}) | |
|
417 | 469 | |
|
418 | 470 | try: |
|
419 |
log.debug("Trying to open URL %s", |
|
|
471 | log.debug("Trying to open URL %s", obfuscated_uri) | |
|
420 | 472 | resp = o.open(req) |
|
421 | 473 | if resp.code != 200: |
|
422 | 474 | raise exceptions.URLError()('Return Code is not 200') |
|
423 | 475 | except Exception as e: |
|
424 |
log.warning("URL cannot be opened: %s", |
|
|
476 | log.warning("URL cannot be opened: %s", obfuscated_uri, exc_info=True) | |
|
425 | 477 | # means it cannot be cloned |
|
426 |
raise exceptions.URLError(e)("[ |
|
|
478 | raise exceptions.URLError(e)(f"[{obfuscated_uri}] org_exc: {e}") | |
|
427 | 479 | |
|
428 | 480 | # now check if it's a proper hg repo, but don't do it for svn |
|
429 | 481 | try: |
@@ -432,19 +484,18 b' class HgRemote(RemoteBase):' | |||
|
432 | 484 | else: |
|
433 | 485 | # check for pure hg repos |
|
434 | 486 | log.debug( |
|
435 | "Verifying if URL is a Mercurial repository: %s", | |
|
436 | cleaned_uri) | |
|
487 | "Verifying if URL is a Mercurial repository: %s", obfuscated_uri) | |
|
437 | 488 | ui = make_ui_from_config(config) |
|
438 | peer_checker = makepeer(ui, url) | |
|
439 | peer_checker.lookup('tip') | |
|
489 | peer_checker = makepeer(ui, safe_bytes(url)) | |
|
490 | peer_checker.lookup(b'tip') | |
|
440 | 491 | except Exception as e: |
|
441 | 492 | log.warning("URL is not a valid Mercurial repository: %s", |
|
442 |
|
|
|
493 | obfuscated_uri) | |
|
443 | 494 | raise exceptions.URLError(e)( |
|
444 | 495 | "url [%s] does not look like an hg repo org_exc: %s" |
|
445 |
% ( |
|
|
496 | % (obfuscated_uri, e)) | |
|
446 | 497 | |
|
447 |
log.info("URL is a valid Mercurial repository: %s", |
|
|
498 | log.info("URL is a valid Mercurial repository: %s", obfuscated_uri) | |
|
448 | 499 | return True |
|
449 | 500 | |
|
450 | 501 | @reraise_safe_exceptions |
@@ -452,14 +503,17 b' class HgRemote(RemoteBase):' | |||
|
452 | 503 | repo = self._factory.repo(wire) |
|
453 | 504 | |
|
454 | 505 | if file_filter: |
|
455 | match_filter = match(file_filter[0], '', [file_filter[1]]) | |
|
506 | # unpack the file-filter | |
|
507 | repo_path, node_path = file_filter | |
|
508 | match_filter = match(safe_bytes(repo_path), b'', [safe_bytes(node_path)]) | |
|
456 | 509 | else: |
|
457 | 510 | match_filter = file_filter |
|
458 | 511 | opts = diffopts(git=opt_git, ignorews=opt_ignorews, context=context, showfunc=1) |
|
459 | 512 | |
|
460 | 513 | try: |
|
461 |
|
|
|
462 |
repo, node1=commit_id_1, node2=commit_id_2, match=match_filter, opts=opts) |
|
|
514 | diff_iter = patch.diff( | |
|
515 | repo, node1=commit_id_1, node2=commit_id_2, match=match_filter, opts=opts) | |
|
516 | return BytesEnvelope(b"".join(diff_iter)) | |
|
463 | 517 | except RepoLookupError as e: |
|
464 | 518 | raise exceptions.LookupException(e)() |
|
465 | 519 | |
@@ -467,12 +521,13 b' class HgRemote(RemoteBase):' | |||
|
467 | 521 | def node_history(self, wire, revision, path, limit): |
|
468 | 522 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
469 | 523 | region = self._region(wire) |
|
524 | ||
|
470 | 525 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
471 | 526 | def _node_history(_context_uid, _repo_id, _revision, _path, _limit): |
|
472 | 527 | repo = self._factory.repo(wire) |
|
473 | 528 | |
|
474 | 529 | ctx = self._get_ctx(repo, revision) |
|
475 | fctx = ctx.filectx(path) | |
|
530 | fctx = ctx.filectx(safe_bytes(path)) | |
|
476 | 531 | |
|
477 | 532 | def history_iter(): |
|
478 | 533 | limit_rev = fctx.rev() |
@@ -498,11 +553,12 b' class HgRemote(RemoteBase):' | |||
|
498 | 553 | def node_history_untill(self, wire, revision, path, limit): |
|
499 | 554 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
500 | 555 | region = self._region(wire) |
|
556 | ||
|
501 | 557 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
502 | 558 | def _node_history_until(_context_uid, _repo_id): |
|
503 | 559 | repo = self._factory.repo(wire) |
|
504 | 560 | ctx = self._get_ctx(repo, revision) |
|
505 | fctx = ctx.filectx(path) | |
|
561 | fctx = ctx.filectx(safe_bytes(path)) | |
|
506 | 562 | |
|
507 | 563 | file_log = list(fctx.filelog()) |
|
508 | 564 | if limit: |
@@ -513,10 +569,29 b' class HgRemote(RemoteBase):' | |||
|
513 | 569 | return _node_history_until(context_uid, repo_id, revision, path, limit) |
|
514 | 570 | |
|
515 | 571 | @reraise_safe_exceptions |
|
572 | def bulk_file_request(self, wire, commit_id, path, pre_load): | |
|
573 | cache_on, context_uid, repo_id = self._cache_on(wire) | |
|
574 | region = self._region(wire) | |
|
575 | ||
|
576 | @region.conditional_cache_on_arguments(condition=cache_on) | |
|
577 | def _bulk_file_request(_repo_id, _commit_id, _path, _pre_load): | |
|
578 | result = {} | |
|
579 | for attr in pre_load: | |
|
580 | try: | |
|
581 | method = self._bulk_file_methods[attr] | |
|
582 | wire.update({'cache': False}) # disable cache for bulk calls so we don't double cache | |
|
583 | result[attr] = method(wire, _commit_id, _path) | |
|
584 | except KeyError as e: | |
|
585 | raise exceptions.VcsException(e)(f'Unknown bulk attribute: "{attr}"') | |
|
586 | return BinaryEnvelope(result) | |
|
587 | ||
|
588 | return _bulk_file_request(repo_id, commit_id, path, sorted(pre_load)) | |
|
589 | ||
|
590 | @reraise_safe_exceptions | |
|
516 | 591 | def fctx_annotate(self, wire, revision, path): |
|
517 | 592 | repo = self._factory.repo(wire) |
|
518 | 593 | ctx = self._get_ctx(repo, revision) |
|
519 | fctx = ctx.filectx(path) | |
|
594 | fctx = ctx.filectx(safe_bytes(path)) | |
|
520 | 595 | |
|
521 | 596 | result = [] |
|
522 | 597 | for i, annotate_obj in enumerate(fctx.annotate(), 1): |
@@ -530,18 +605,19 b' class HgRemote(RemoteBase):' | |||
|
530 | 605 | def fctx_node_data(self, wire, revision, path): |
|
531 | 606 | repo = self._factory.repo(wire) |
|
532 | 607 | ctx = self._get_ctx(repo, revision) |
|
533 | fctx = ctx.filectx(path) | |
|
534 | return fctx.data() | |
|
608 | fctx = ctx.filectx(safe_bytes(path)) | |
|
609 | return BytesEnvelope(fctx.data()) | |
|
535 | 610 | |
|
536 | 611 | @reraise_safe_exceptions |
|
537 | 612 | def fctx_flags(self, wire, commit_id, path): |
|
538 | 613 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
539 | 614 | region = self._region(wire) |
|
615 | ||
|
540 | 616 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
541 | 617 | def _fctx_flags(_repo_id, _commit_id, _path): |
|
542 | 618 | repo = self._factory.repo(wire) |
|
543 | 619 | ctx = self._get_ctx(repo, commit_id) |
|
544 | fctx = ctx.filectx(path) | |
|
620 | fctx = ctx.filectx(safe_bytes(path)) | |
|
545 | 621 | return fctx.flags() |
|
546 | 622 | |
|
547 | 623 | return _fctx_flags(repo_id, commit_id, path) |
@@ -550,11 +626,12 b' class HgRemote(RemoteBase):' | |||
|
550 | 626 | def fctx_size(self, wire, commit_id, path): |
|
551 | 627 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
552 | 628 | region = self._region(wire) |
|
629 | ||
|
553 | 630 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
554 | 631 | def _fctx_size(_repo_id, _revision, _path): |
|
555 | 632 | repo = self._factory.repo(wire) |
|
556 | 633 | ctx = self._get_ctx(repo, commit_id) |
|
557 | fctx = ctx.filectx(path) | |
|
634 | fctx = ctx.filectx(safe_bytes(path)) | |
|
558 | 635 | return fctx.size() |
|
559 | 636 | return _fctx_size(repo_id, commit_id, path) |
|
560 | 637 | |
@@ -562,44 +639,59 b' class HgRemote(RemoteBase):' | |||
|
562 | 639 | def get_all_commit_ids(self, wire, name): |
|
563 | 640 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
564 | 641 | region = self._region(wire) |
|
642 | ||
|
565 | 643 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
566 | 644 | def _get_all_commit_ids(_context_uid, _repo_id, _name): |
|
567 | 645 | repo = self._factory.repo(wire) |
|
568 | repo = repo.filtered(name) | |
|
569 | revs = map(lambda x: hex(x[7]), repo.changelog.index) | |
|
646 | revs = [ascii_str(repo[x].hex()) for x in repo.filtered(b'visible').changelog.revs()] | |
|
570 | 647 | return revs |
|
571 | 648 | return _get_all_commit_ids(context_uid, repo_id, name) |
|
572 | 649 | |
|
573 | 650 | @reraise_safe_exceptions |
|
574 | 651 | def get_config_value(self, wire, section, name, untrusted=False): |
|
575 | 652 | repo = self._factory.repo(wire) |
|
576 | return repo.ui.config(section, name, untrusted=untrusted) | |
|
653 | return repo.ui.config(ascii_bytes(section), ascii_bytes(name), untrusted=untrusted) | |
|
577 | 654 | |
|
578 | 655 | @reraise_safe_exceptions |
|
579 | 656 | def is_large_file(self, wire, commit_id, path): |
|
580 | 657 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
581 | 658 | region = self._region(wire) |
|
659 | ||
|
582 | 660 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
583 | 661 | def _is_large_file(_context_uid, _repo_id, _commit_id, _path): |
|
584 | return largefiles.lfutil.isstandin(path) | |
|
662 | return largefiles.lfutil.isstandin(safe_bytes(path)) | |
|
585 | 663 | |
|
586 | 664 | return _is_large_file(context_uid, repo_id, commit_id, path) |
|
587 | 665 | |
|
588 | 666 | @reraise_safe_exceptions |
|
589 | 667 | def is_binary(self, wire, revision, path): |
|
590 | 668 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
669 | region = self._region(wire) | |
|
591 | 670 | |
|
592 | region = self._region(wire) | |
|
593 | 671 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
594 | 672 | def _is_binary(_repo_id, _sha, _path): |
|
595 | 673 | repo = self._factory.repo(wire) |
|
596 | 674 | ctx = self._get_ctx(repo, revision) |
|
597 | fctx = ctx.filectx(path) | |
|
675 | fctx = ctx.filectx(safe_bytes(path)) | |
|
598 | 676 | return fctx.isbinary() |
|
599 | 677 | |
|
600 | 678 | return _is_binary(repo_id, revision, path) |
|
601 | 679 | |
|
602 | 680 | @reraise_safe_exceptions |
|
681 | def md5_hash(self, wire, revision, path): | |
|
682 | cache_on, context_uid, repo_id = self._cache_on(wire) | |
|
683 | region = self._region(wire) | |
|
684 | ||
|
685 | @region.conditional_cache_on_arguments(condition=cache_on) | |
|
686 | def _md5_hash(_repo_id, _sha, _path): | |
|
687 | repo = self._factory.repo(wire) | |
|
688 | ctx = self._get_ctx(repo, revision) | |
|
689 | fctx = ctx.filectx(safe_bytes(path)) | |
|
690 | return hashlib.md5(fctx.data()).hexdigest() | |
|
691 | ||
|
692 | return _md5_hash(repo_id, revision, path) | |
|
693 | ||
|
694 | @reraise_safe_exceptions | |
|
603 | 695 | def in_largefiles_store(self, wire, sha): |
|
604 | 696 | repo = self._factory.repo(wire) |
|
605 | 697 | return largefiles.lfutil.instore(repo, sha) |
@@ -627,11 +719,10 b' class HgRemote(RemoteBase):' | |||
|
627 | 719 | @reraise_safe_exceptions |
|
628 | 720 | def lookup(self, wire, revision, both): |
|
629 | 721 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
722 | region = self._region(wire) | |
|
630 | 723 | |
|
631 | region = self._region(wire) | |
|
632 | 724 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
633 | 725 | def _lookup(_context_uid, _repo_id, _revision, _both): |
|
634 | ||
|
635 | 726 | repo = self._factory.repo(wire) |
|
636 | 727 | rev = _revision |
|
637 | 728 | if isinstance(rev, int): |
@@ -644,7 +735,7 b' class HgRemote(RemoteBase):' | |||
|
644 | 735 | rev = rev + -1 |
|
645 | 736 | try: |
|
646 | 737 | ctx = self._get_ctx(repo, rev) |
|
647 | except (TypeError, RepoLookupError) as e: | |
|
738 | except (TypeError, RepoLookupError, binascii.Error) as e: | |
|
648 | 739 | e._org_exc_tb = traceback.format_exc() |
|
649 | 740 | raise exceptions.LookupException(e)(rev) |
|
650 | 741 | except LookupError as e: |
@@ -667,12 +758,12 b' class HgRemote(RemoteBase):' | |||
|
667 | 758 | repo = self._factory.repo(wire) |
|
668 | 759 | |
|
669 | 760 | # Disable any prompts for this repo |
|
670 | repo.ui.setconfig('ui', 'interactive', 'off', '-y') | |
|
761 | repo.ui.setconfig(b'ui', b'interactive', b'off', b'-y') | |
|
671 | 762 | |
|
672 | bookmarks = dict(repo._bookmarks).keys() | |
|
673 | remote = peer(repo, {}, url) | |
|
763 | bookmarks = list(dict(repo._bookmarks).keys()) | |
|
764 | remote = peer(repo, {}, safe_bytes(url)) | |
|
674 | 765 | # Disable any prompts for this remote |
|
675 | remote.ui.setconfig('ui', 'interactive', 'off', '-y') | |
|
766 | remote.ui.setconfig(b'ui', b'interactive', b'off', b'-y') | |
|
676 | 767 | |
|
677 | 768 | return exchange.push( |
|
678 | 769 | repo, remote, newbranch=True, bookmarks=bookmarks).cgresult |
@@ -686,12 +777,15 b' class HgRemote(RemoteBase):' | |||
|
686 | 777 | @reraise_safe_exceptions |
|
687 | 778 | def rev_range(self, wire, commit_filter): |
|
688 | 779 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
780 | region = self._region(wire) | |
|
689 | 781 | |
|
690 | region = self._region(wire) | |
|
691 | 782 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
692 | 783 | def _rev_range(_context_uid, _repo_id, _filter): |
|
693 | 784 | repo = self._factory.repo(wire) |
|
694 | revisions = [rev for rev in revrange(repo, commit_filter)] | |
|
785 | revisions = [ | |
|
786 | ascii_str(repo[rev].hex()) | |
|
787 | for rev in revrange(repo, list(map(ascii_bytes, commit_filter))) | |
|
788 | ] | |
|
695 | 789 | return revisions |
|
696 | 790 | |
|
697 | 791 | return _rev_range(context_uid, repo_id, sorted(commit_filter)) |
@@ -710,17 +804,18 b' class HgRemote(RemoteBase):' | |||
|
710 | 804 | return len(repo) - 1, 0 |
|
711 | 805 | |
|
712 | 806 | stop, start = get_revs(repo, [node + ':']) |
|
713 |
revs = [ |
|
|
807 | revs = [ascii_str(repo[r].hex()) for r in range(start, stop + 1)] | |
|
714 | 808 | return revs |
|
715 | 809 | |
|
716 | 810 | @reraise_safe_exceptions |
|
717 | 811 | def revs_from_revspec(self, wire, rev_spec, *args, **kwargs): |
|
718 | other_path = kwargs.pop('other_path', None) | |
|
812 | org_path = safe_bytes(wire["path"]) | |
|
813 | other_path = safe_bytes(kwargs.pop('other_path', '')) | |
|
719 | 814 | |
|
720 | 815 | # case when we want to compare two independent repositories |
|
721 | 816 | if other_path and other_path != wire["path"]: |
|
722 | 817 | baseui = self._factory._create_config(wire["config"]) |
|
723 |
repo = unionrepo.makeunionrepository(baseui, other_path, |
|
|
818 | repo = unionrepo.makeunionrepository(baseui, other_path, org_path) | |
|
724 | 819 | else: |
|
725 | 820 | repo = self._factory.repo(wire) |
|
726 | 821 | return list(repo.revs(rev_spec, *args)) |
@@ -764,17 +859,20 b' class HgRemote(RemoteBase):' | |||
|
764 | 859 | def tags(self, wire): |
|
765 | 860 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
766 | 861 | region = self._region(wire) |
|
862 | ||
|
767 | 863 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
768 | 864 | def _tags(_context_uid, _repo_id): |
|
769 | 865 | repo = self._factory.repo(wire) |
|
770 | return repo.tags() | |
|
866 | return {safe_str(name): ascii_str(hex(sha)) for name, sha in repo.tags().items()} | |
|
771 | 867 | |
|
772 | 868 | return _tags(context_uid, repo_id) |
|
773 | 869 | |
|
774 | 870 | @reraise_safe_exceptions |
|
775 |
def update(self, wire, node= |
|
|
871 | def update(self, wire, node='', clean=False): | |
|
776 | 872 | repo = self._factory.repo(wire) |
|
777 | 873 | baseui = self._factory._create_config(wire['config']) |
|
874 | node = safe_bytes(node) | |
|
875 | ||
|
778 | 876 | commands.update(baseui, repo, node=node, clean=clean) |
|
779 | 877 | |
|
780 | 878 | @reraise_safe_exceptions |
@@ -800,10 +898,10 b' class HgRemote(RemoteBase):' | |||
|
800 | 898 | |
|
801 | 899 | baseui.write = write |
|
802 | 900 | if branch: |
|
803 | args = [branch] | |
|
901 | args = [safe_bytes(branch)] | |
|
804 | 902 | else: |
|
805 | 903 | args = [] |
|
806 | commands.heads(baseui, repo, template='{node} ', *args) | |
|
904 | commands.heads(baseui, repo, template=b'{node} ', *args) | |
|
807 | 905 | |
|
808 | 906 | return output.getvalue() |
|
809 | 907 | |
@@ -812,63 +910,61 b' class HgRemote(RemoteBase):' | |||
|
812 | 910 | repo = self._factory.repo(wire) |
|
813 | 911 | changelog = repo.changelog |
|
814 | 912 | lookup = repo.lookup |
|
815 | a = changelog.ancestor(lookup(revision1), lookup(revision2)) | |
|
913 | a = changelog.ancestor(lookup(safe_bytes(revision1)), lookup(safe_bytes(revision2))) | |
|
816 | 914 | return hex(a) |
|
817 | 915 | |
|
818 | 916 | @reraise_safe_exceptions |
|
819 | 917 | def clone(self, wire, source, dest, update_after_clone=False, hooks=True): |
|
820 | 918 | baseui = self._factory._create_config(wire["config"], hooks=hooks) |
|
821 | clone(baseui, source, dest, noupdate=not update_after_clone) | |
|
919 | clone(baseui, safe_bytes(source), safe_bytes(dest), noupdate=not update_after_clone) | |
|
822 | 920 | |
|
823 | 921 | @reraise_safe_exceptions |
|
824 | 922 | def commitctx(self, wire, message, parents, commit_time, commit_timezone, user, files, extra, removed, updated): |
|
825 | 923 | |
|
826 | 924 | repo = self._factory.repo(wire) |
|
827 | 925 | baseui = self._factory._create_config(wire['config']) |
|
828 | publishing = baseui.configbool('phases', 'publish') | |
|
829 | if publishing: | |
|
830 | new_commit = 'public' | |
|
831 | else: | |
|
832 | new_commit = 'draft' | |
|
926 | publishing = baseui.configbool(b'phases', b'publish') | |
|
833 | 927 | |
|
834 | def _filectxfn(_repo, ctx, path): | |
|
928 | def _filectxfn(_repo, ctx, path: bytes): | |
|
835 | 929 | """ |
|
836 | 930 | Marks given path as added/changed/removed in a given _repo. This is |
|
837 | 931 | for internal mercurial commit function. |
|
838 | 932 | """ |
|
839 | 933 | |
|
840 | 934 | # check if this path is removed |
|
841 | if path in removed: | |
|
935 | if safe_str(path) in removed: | |
|
842 | 936 | # returning None is a way to mark node for removal |
|
843 | 937 | return None |
|
844 | 938 | |
|
845 | 939 | # check if this path is added |
|
846 | 940 | for node in updated: |
|
847 | if node['path'] == path: | |
|
941 | if safe_bytes(node['path']) == path: | |
|
848 | 942 | return memfilectx( |
|
849 | 943 | _repo, |
|
850 | 944 | changectx=ctx, |
|
851 | path=node['path'], | |
|
852 | data=node['content'], | |
|
945 | path=safe_bytes(node['path']), | |
|
946 | data=safe_bytes(node['content']), | |
|
853 | 947 | islink=False, |
|
854 | 948 | isexec=bool(node['mode'] & stat.S_IXUSR), |
|
855 | 949 | copysource=False) |
|
950 | abort_exc = exceptions.AbortException() | |
|
951 | raise abort_exc(f"Given path haven't been marked as added, changed or removed ({path})") | |
|
856 | 952 | |
|
857 | raise exceptions.AbortException()( | |
|
858 | "Given path haven't been marked as added, " | |
|
859 | "changed or removed (%s)" % path) | |
|
860 | ||
|
861 | with repo.ui.configoverride({('phases', 'new-commit'): new_commit}): | |
|
862 | ||
|
953 | if publishing: | |
|
954 | new_commit_phase = b'public' | |
|
955 | else: | |
|
956 | new_commit_phase = b'draft' | |
|
957 | with repo.ui.configoverride({(b'phases', b'new-commit'): new_commit_phase}): | |
|
958 | kwargs = {safe_bytes(k): safe_bytes(v) for k, v in extra.items()} | |
|
863 | 959 | commit_ctx = memctx( |
|
864 | 960 | repo=repo, |
|
865 | 961 | parents=parents, |
|
866 | text=message, | |
|
867 | files=files, | |
|
962 | text=safe_bytes(message), | |
|
963 | files=[safe_bytes(x) for x in files], | |
|
868 | 964 | filectxfn=_filectxfn, |
|
869 | user=user, | |
|
965 | user=safe_bytes(user), | |
|
870 | 966 | date=(commit_time, commit_timezone), |
|
871 |
extra= |
|
|
967 | extra=kwargs) | |
|
872 | 968 | |
|
873 | 969 | n = repo.commitctx(commit_ctx) |
|
874 | 970 | new_id = hex(n) |
@@ -879,11 +975,11 b' class HgRemote(RemoteBase):' | |||
|
879 | 975 | def pull(self, wire, url, commit_ids=None): |
|
880 | 976 | repo = self._factory.repo(wire) |
|
881 | 977 | # Disable any prompts for this repo |
|
882 | repo.ui.setconfig('ui', 'interactive', 'off', '-y') | |
|
978 | repo.ui.setconfig(b'ui', b'interactive', b'off', b'-y') | |
|
883 | 979 | |
|
884 | remote = peer(repo, {}, url) | |
|
980 | remote = peer(repo, {}, safe_bytes(url)) | |
|
885 | 981 | # Disable any prompts for this remote |
|
886 | remote.ui.setconfig('ui', 'interactive', 'off', '-y') | |
|
982 | remote.ui.setconfig(b'ui', b'interactive', b'off', b'-y') | |
|
887 | 983 | |
|
888 | 984 | if commit_ids: |
|
889 | 985 | commit_ids = [bin(commit_id) for commit_id in commit_ids] |
@@ -892,34 +988,47 b' class HgRemote(RemoteBase):' | |||
|
892 | 988 | repo, remote, heads=commit_ids, force=None).cgresult |
|
893 | 989 | |
|
894 | 990 | @reraise_safe_exceptions |
|
895 |
def pull_cmd(self, wire, source, bookmark= |
|
|
991 | def pull_cmd(self, wire, source, bookmark='', branch='', revision='', hooks=True): | |
|
896 | 992 | repo = self._factory.repo(wire) |
|
897 | 993 | baseui = self._factory._create_config(wire['config'], hooks=hooks) |
|
898 | 994 | |
|
995 | source = safe_bytes(source) | |
|
996 | ||
|
899 | 997 | # Mercurial internally has a lot of logic that checks ONLY if |
|
900 | 998 | # option is defined, we just pass those if they are defined then |
|
901 | 999 | opts = {} |
|
1000 | ||
|
902 | 1001 | if bookmark: |
|
903 | opts['bookmark'] = bookmark | |
|
1002 | opts['bookmark'] = [safe_bytes(x) for x in bookmark] \ | |
|
1003 | if isinstance(bookmark, list) else safe_bytes(bookmark) | |
|
1004 | ||
|
904 | 1005 | if branch: |
|
905 | opts['branch'] = branch | |
|
1006 | opts['branch'] = [safe_bytes(x) for x in branch] \ | |
|
1007 | if isinstance(branch, list) else safe_bytes(branch) | |
|
1008 | ||
|
906 | 1009 | if revision: |
|
907 | opts['rev'] = revision | |
|
1010 | opts['rev'] = [safe_bytes(x) for x in revision] \ | |
|
1011 | if isinstance(revision, list) else safe_bytes(revision) | |
|
908 | 1012 | |
|
909 | 1013 | commands.pull(baseui, repo, source, **opts) |
|
910 | 1014 | |
|
911 | 1015 | @reraise_safe_exceptions |
|
912 | def push(self, wire, revisions, dest_path, hooks=True, push_branches=False): | |
|
1016 | def push(self, wire, revisions, dest_path, hooks: bool = True, push_branches: bool = False): | |
|
913 | 1017 | repo = self._factory.repo(wire) |
|
914 | 1018 | baseui = self._factory._create_config(wire['config'], hooks=hooks) |
|
915 | commands.push(baseui, repo, dest=dest_path, rev=revisions, | |
|
1019 | ||
|
1020 | revisions = [safe_bytes(x) for x in revisions] \ | |
|
1021 | if isinstance(revisions, list) else safe_bytes(revisions) | |
|
1022 | ||
|
1023 | commands.push(baseui, repo, safe_bytes(dest_path), | |
|
1024 | rev=revisions, | |
|
916 | 1025 | new_branch=push_branches) |
|
917 | 1026 | |
|
918 | 1027 | @reraise_safe_exceptions |
|
919 | 1028 | def strip(self, wire, revision, update, backup): |
|
920 | 1029 | repo = self._factory.repo(wire) |
|
921 | 1030 | ctx = self._get_ctx(repo, revision) |
|
922 | hgext_strip( | |
|
1031 | hgext_strip.strip( | |
|
923 | 1032 | repo.baseui, repo, ctx.node(), update=update, backup=backup) |
|
924 | 1033 | |
|
925 | 1034 | @reraise_safe_exceptions |
@@ -943,25 +1052,25 b' class HgRemote(RemoteBase):' | |||
|
943 | 1052 | def merge(self, wire, revision): |
|
944 | 1053 | repo = self._factory.repo(wire) |
|
945 | 1054 | baseui = self._factory._create_config(wire['config']) |
|
946 | repo.ui.setconfig('ui', 'merge', 'internal:dump') | |
|
1055 | repo.ui.setconfig(b'ui', b'merge', b'internal:dump') | |
|
947 | 1056 | |
|
948 | 1057 | # In case of sub repositories are used mercurial prompts the user in |
|
949 | 1058 | # case of merge conflicts or different sub repository sources. By |
|
950 | 1059 | # setting the interactive flag to `False` mercurial doesn't prompt the |
|
951 | 1060 | # used but instead uses a default value. |
|
952 | repo.ui.setconfig('ui', 'interactive', False) | |
|
953 | commands.merge(baseui, repo, rev=revision) | |
|
1061 | repo.ui.setconfig(b'ui', b'interactive', False) | |
|
1062 | commands.merge(baseui, repo, rev=safe_bytes(revision)) | |
|
954 | 1063 | |
|
955 | 1064 | @reraise_safe_exceptions |
|
956 | 1065 | def merge_state(self, wire): |
|
957 | 1066 | repo = self._factory.repo(wire) |
|
958 | repo.ui.setconfig('ui', 'merge', 'internal:dump') | |
|
1067 | repo.ui.setconfig(b'ui', b'merge', b'internal:dump') | |
|
959 | 1068 | |
|
960 | 1069 | # In case of sub repositories are used mercurial prompts the user in |
|
961 | 1070 | # case of merge conflicts or different sub repository sources. By |
|
962 | 1071 | # setting the interactive flag to `False` mercurial doesn't prompt the |
|
963 | 1072 | # used but instead uses a default value. |
|
964 | repo.ui.setconfig('ui', 'interactive', False) | |
|
1073 | repo.ui.setconfig(b'ui', b'interactive', False) | |
|
965 | 1074 | ms = hg_merge.mergestate(repo) |
|
966 | 1075 | return [x for x in ms.unresolved()] |
|
967 | 1076 | |
@@ -969,20 +1078,22 b' class HgRemote(RemoteBase):' | |||
|
969 | 1078 | def commit(self, wire, message, username, close_branch=False): |
|
970 | 1079 | repo = self._factory.repo(wire) |
|
971 | 1080 | baseui = self._factory._create_config(wire['config']) |
|
972 | repo.ui.setconfig('ui', 'username', username) | |
|
973 | commands.commit(baseui, repo, message=message, close_branch=close_branch) | |
|
1081 | repo.ui.setconfig(b'ui', b'username', safe_bytes(username)) | |
|
1082 | commands.commit(baseui, repo, message=safe_bytes(message), close_branch=close_branch) | |
|
974 | 1083 | |
|
975 | 1084 | @reraise_safe_exceptions |
|
976 |
def rebase(self, wire, source= |
|
|
1085 | def rebase(self, wire, source='', dest='', abort=False): | |
|
977 | 1086 | repo = self._factory.repo(wire) |
|
978 | 1087 | baseui = self._factory._create_config(wire['config']) |
|
979 | repo.ui.setconfig('ui', 'merge', 'internal:dump') | |
|
1088 | repo.ui.setconfig(b'ui', b'merge', b'internal:dump') | |
|
980 | 1089 | # In case of sub repositories are used mercurial prompts the user in |
|
981 | 1090 | # case of merge conflicts or different sub repository sources. By |
|
982 | 1091 | # setting the interactive flag to `False` mercurial doesn't prompt the |
|
983 | 1092 | # used but instead uses a default value. |
|
984 | repo.ui.setconfig('ui', 'interactive', False) | |
|
985 | rebase.rebase(baseui, repo, base=source, dest=dest, abort=abort, keep=not abort) | |
|
1093 | repo.ui.setconfig(b'ui', b'interactive', False) | |
|
1094 | ||
|
1095 | rebase.rebase(baseui, repo, base=safe_bytes(source or ''), dest=safe_bytes(dest or ''), | |
|
1096 | abort=abort, keep=not abort) | |
|
986 | 1097 | |
|
987 | 1098 | @reraise_safe_exceptions |
|
988 | 1099 | def tag(self, wire, name, revision, message, local, user, tag_time, tag_timezone): |
@@ -992,17 +1103,18 b' class HgRemote(RemoteBase):' | |||
|
992 | 1103 | |
|
993 | 1104 | date = (tag_time, tag_timezone) |
|
994 | 1105 | try: |
|
995 | hg_tag.tag(repo, name, node, message, local, user, date) | |
|
1106 | hg_tag.tag(repo, safe_bytes(name), node, safe_bytes(message), local, safe_bytes(user), date) | |
|
996 | 1107 | except Abort as e: |
|
997 | 1108 | log.exception("Tag operation aborted") |
|
998 | 1109 | # Exception can contain unicode which we convert |
|
999 | 1110 | raise exceptions.AbortException(e)(repr(e)) |
|
1000 | 1111 | |
|
1001 | 1112 | @reraise_safe_exceptions |
|
1002 |
def bookmark(self, wire, bookmark, revision= |
|
|
1113 | def bookmark(self, wire, bookmark, revision=''): | |
|
1003 | 1114 | repo = self._factory.repo(wire) |
|
1004 | 1115 | baseui = self._factory._create_config(wire['config']) |
|
1005 | commands.bookmark(baseui, repo, bookmark, rev=revision, force=True) | |
|
1116 | revision = revision or '' | |
|
1117 | commands.bookmark(baseui, repo, safe_bytes(bookmark), rev=safe_bytes(revision), force=True) | |
|
1006 | 1118 | |
|
1007 | 1119 | @reraise_safe_exceptions |
|
1008 | 1120 | def install_hooks(self, wire, force=False): |
@@ -1021,8 +1133,8 b' class HgRemote(RemoteBase):' | |||
|
1021 | 1133 | pass |
|
1022 | 1134 | |
|
1023 | 1135 | @reraise_safe_exceptions |
|
1024 |
def archive_repo(self, wire, archive_ |
|
|
1025 | archive_dir_name, commit_id): | |
|
1136 | def archive_repo(self, wire, archive_name_key, kind, mtime, archive_at_path, | |
|
1137 | archive_dir_name, commit_id, cache_config): | |
|
1026 | 1138 | |
|
1027 | 1139 | def file_walker(_commit_id, path): |
|
1028 | 1140 | repo = self._factory.repo(wire) |
@@ -1042,6 +1154,6 b' class HgRemote(RemoteBase):' | |||
|
1042 | 1154 | |
|
1043 | 1155 | yield ArchiveNode(file_path, mode, is_link, ctx[fn].data) |
|
1044 | 1156 | |
|
1045 | return archive_repo(file_walker, archive_dest_path, kind, mtime, archive_at_path, | |
|
1046 | archive_dir_name, commit_id) | |
|
1157 | return store_archive_in_cache( | |
|
1158 | file_walker, archive_name_key, kind, mtime, archive_at_path, archive_dir_name, commit_id, cache_config=cache_config) | |
|
1047 | 1159 |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -15,32 +15,34 b'' | |||
|
15 | 15 | # along with this program; if not, write to the Free Software Foundation, |
|
16 | 16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
17 | 17 | |
|
18 | from __future__ import absolute_import | |
|
19 | 18 | |
|
20 | 19 | import os |
|
21 | 20 | import subprocess |
|
22 | import time | |
|
23 | from urllib2 import URLError | |
|
24 | import urlparse | |
|
21 | from urllib.error import URLError | |
|
22 | import urllib.parse | |
|
25 | 23 | import logging |
|
26 | 24 | import posixpath as vcspath |
|
27 |
import |
|
|
28 | import urllib | |
|
25 | import io | |
|
26 | import urllib.request | |
|
27 | import urllib.parse | |
|
28 | import urllib.error | |
|
29 | 29 | import traceback |
|
30 | 30 | |
|
31 | import svn.client | |
|
32 |
import svn.c |
|
|
33 |
import svn. |
|
|
34 |
import svn.d |
|
|
35 |
import svn. |
|
|
36 |
import svn. |
|
|
31 | ||
|
32 | import svn.client # noqa | |
|
33 | import svn.core # noqa | |
|
34 | import svn.delta # noqa | |
|
35 | import svn.diff # noqa | |
|
36 | import svn.fs # noqa | |
|
37 | import svn.repos # noqa | |
|
37 | 38 | |
|
38 | 39 | from vcsserver import svn_diff, exceptions, subprocessio, settings |
|
39 |
from vcsserver.base import RepoFactory, raise_from_original, ArchiveNode, archive_ |
|
|
40 | from vcsserver.base import RepoFactory, raise_from_original, ArchiveNode, store_archive_in_cache, BytesEnvelope, BinaryEnvelope | |
|
40 | 41 | from vcsserver.exceptions import NoContentException |
|
41 | from vcsserver.utils import safe_str | |
|
42 | from vcsserver.str_utils import safe_str, safe_bytes | |
|
43 | from vcsserver.type_utils import assert_bytes | |
|
42 | 44 | from vcsserver.vcs_base import RemoteBase |
|
43 | ||
|
45 | from vcsserver.lib.svnremoterepo import svnremoterepo | |
|
44 | 46 | log = logging.getLogger(__name__) |
|
45 | 47 | |
|
46 | 48 | |
@@ -52,7 +54,7 b' svn_compatible_versions_map = {' | |||
|
52 | 54 | 'pre-1.9-compatible': '1.8', |
|
53 | 55 | } |
|
54 | 56 | |
|
55 |
current_compatible_version = '1.1 |
|
|
57 | current_compatible_version = '1.14' | |
|
56 | 58 | |
|
57 | 59 | |
|
58 | 60 | def reraise_safe_exceptions(func): |
@@ -63,7 +65,7 b' def reraise_safe_exceptions(func):' | |||
|
63 | 65 | except Exception as e: |
|
64 | 66 | if not hasattr(e, '_vcs_kind'): |
|
65 | 67 | log.exception("Unhandled exception in svn remote call") |
|
66 | raise_from_original(exceptions.UnhandledException(e)) | |
|
68 | raise_from_original(exceptions.UnhandledException(e), e) | |
|
67 | 69 | raise |
|
68 | 70 | return wrapper |
|
69 | 71 | |
@@ -82,12 +84,12 b' class SubversionFactory(RepoFactory):' | |||
|
82 | 84 | or compatible_version |
|
83 | 85 | fs_config['compatible-version'] = compatible_version_string |
|
84 | 86 | |
|
85 |
log.debug('Create SVN repo with config |
|
|
87 | log.debug('Create SVN repo with config `%s`', fs_config) | |
|
86 | 88 | repo = svn.repos.create(path, "", "", None, fs_config) |
|
87 | 89 | else: |
|
88 | 90 | repo = svn.repos.open(path) |
|
89 | 91 | |
|
90 |
log.debug(' |
|
|
92 | log.debug('repository created: got SVN object: %s', repo) | |
|
91 | 93 | return repo |
|
92 | 94 | |
|
93 | 95 | def repo(self, wire, create=False, compatible_version=None): |
@@ -107,9 +109,39 b' class SvnRemote(RemoteBase):' | |||
|
107 | 109 | |
|
108 | 110 | def __init__(self, factory, hg_factory=None): |
|
109 | 111 | self._factory = factory |
|
110 | # TODO: Remove once we do not use internal Mercurial objects anymore | |
|
111 | # for subversion | |
|
112 | self._hg_factory = hg_factory | |
|
112 | ||
|
113 | self._bulk_methods = { | |
|
114 | # NOT supported in SVN ATM... | |
|
115 | } | |
|
116 | self._bulk_file_methods = { | |
|
117 | "size": self.get_file_size, | |
|
118 | "data": self.get_file_content, | |
|
119 | "flags": self.get_node_type, | |
|
120 | "is_binary": self.is_binary, | |
|
121 | "md5": self.md5_hash | |
|
122 | } | |
|
123 | ||
|
124 | @reraise_safe_exceptions | |
|
125 | def bulk_file_request(self, wire, commit_id, path, pre_load): | |
|
126 | cache_on, context_uid, repo_id = self._cache_on(wire) | |
|
127 | region = self._region(wire) | |
|
128 | ||
|
129 | # since we use unified API, we need to cast from str to in for SVN | |
|
130 | commit_id = int(commit_id) | |
|
131 | ||
|
132 | @region.conditional_cache_on_arguments(condition=cache_on) | |
|
133 | def _bulk_file_request(_repo_id, _commit_id, _path, _pre_load): | |
|
134 | result = {} | |
|
135 | for attr in pre_load: | |
|
136 | try: | |
|
137 | method = self._bulk_file_methods[attr] | |
|
138 | wire.update({'cache': False}) # disable cache for bulk calls so we don't double cache | |
|
139 | result[attr] = method(wire, _commit_id, _path) | |
|
140 | except KeyError as e: | |
|
141 | raise exceptions.VcsException(e)(f'Unknown bulk attribute: "{attr}"') | |
|
142 | return BinaryEnvelope(result) | |
|
143 | ||
|
144 | return _bulk_file_request(repo_id, commit_id, path, sorted(pre_load)) | |
|
113 | 145 | |
|
114 | 146 | @reraise_safe_exceptions |
|
115 | 147 | def discover_svn_version(self): |
@@ -118,31 +150,27 b' class SvnRemote(RemoteBase):' | |||
|
118 | 150 | svn_ver = svn.core.SVN_VERSION |
|
119 | 151 | except ImportError: |
|
120 | 152 | svn_ver = None |
|
121 | return svn_ver | |
|
153 | return safe_str(svn_ver) | |
|
122 | 154 | |
|
123 | 155 | @reraise_safe_exceptions |
|
124 | 156 | def is_empty(self, wire): |
|
125 | ||
|
126 | 157 | try: |
|
127 | 158 | return self.lookup(wire, -1) == 0 |
|
128 | 159 | except Exception: |
|
129 | 160 | log.exception("failed to read object_store") |
|
130 | 161 | return False |
|
131 | 162 | |
|
132 |
def check_url(self, url, config |
|
|
133 | # this can throw exception if not installed, but we detect this | |
|
134 | from hgsubversion import svnrepo | |
|
163 | def check_url(self, url, config): | |
|
135 | 164 | |
|
136 | baseui = self._hg_factory._create_config(config_items) | |
|
137 | # uuid function get's only valid UUID from proper repo, else | |
|
165 | # uuid function gets only valid UUID from proper repo, else | |
|
138 | 166 | # throws exception |
|
167 | username, password, src_url = self.get_url_and_credentials(url) | |
|
139 | 168 | try: |
|
140 | svnrepo.svnremoterepo(baseui, url).svn.uuid | |
|
169 | svnremoterepo(safe_bytes(username), safe_bytes(password), safe_bytes(src_url)).svn().uuid | |
|
141 | 170 | except Exception: |
|
142 | 171 | tb = traceback.format_exc() |
|
143 | 172 | log.debug("Invalid Subversion url: `%s`, tb: %s", url, tb) |
|
144 | raise URLError( | |
|
145 | '"%s" is not a valid Subversion source url.' % (url, )) | |
|
173 | raise URLError(f'"{url}" is not a valid Subversion source url.') | |
|
146 | 174 | return True |
|
147 | 175 | |
|
148 | 176 | def is_path_valid_repository(self, wire, path): |
@@ -173,6 +201,7 b' class SvnRemote(RemoteBase):' | |||
|
173 | 201 | stdout, stderr = subprocessio.run_command(cmd) |
|
174 | 202 | return stdout |
|
175 | 203 | |
|
204 | @reraise_safe_exceptions | |
|
176 | 205 | def lookup(self, wire, revision): |
|
177 | 206 | if revision not in [-1, None, 'HEAD']: |
|
178 | 207 | raise NotImplementedError |
@@ -181,6 +210,7 b' class SvnRemote(RemoteBase):' | |||
|
181 | 210 | head = svn.fs.youngest_rev(fs_ptr) |
|
182 | 211 | return head |
|
183 | 212 | |
|
213 | @reraise_safe_exceptions | |
|
184 | 214 | def lookup_interval(self, wire, start_ts, end_ts): |
|
185 | 215 | repo = self._factory.repo(wire) |
|
186 | 216 | fsobj = svn.repos.fs(repo) |
@@ -198,10 +228,12 b' class SvnRemote(RemoteBase):' | |||
|
198 | 228 | end_rev = svn.fs.youngest_rev(fsobj) |
|
199 | 229 | return start_rev, end_rev |
|
200 | 230 | |
|
231 | @reraise_safe_exceptions | |
|
201 | 232 | def revision_properties(self, wire, revision): |
|
202 | 233 | |
|
203 | 234 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
204 | 235 | region = self._region(wire) |
|
236 | ||
|
205 | 237 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
206 | 238 | def _revision_properties(_repo_id, _revision): |
|
207 | 239 | repo = self._factory.repo(wire) |
@@ -228,7 +260,7 b' class SvnRemote(RemoteBase):' | |||
|
228 | 260 | removed = [] |
|
229 | 261 | |
|
230 | 262 | # TODO: CHANGE_ACTION_REPLACE: Figure out where it belongs |
|
231 |
for path, change in editor.changes. |
|
|
263 | for path, change in editor.changes.items(): | |
|
232 | 264 | # TODO: Decide what to do with directory nodes. Subversion can add |
|
233 | 265 | # empty directories. |
|
234 | 266 | |
@@ -243,7 +275,7 b' class SvnRemote(RemoteBase):' | |||
|
243 | 275 | removed.append(path) |
|
244 | 276 | else: |
|
245 | 277 | raise NotImplementedError( |
|
246 |
"Action |
|
|
278 | "Action {} not supported on path {}".format( | |
|
247 | 279 | change.action, path)) |
|
248 | 280 | |
|
249 | 281 | changes = { |
@@ -257,6 +289,7 b' class SvnRemote(RemoteBase):' | |||
|
257 | 289 | def node_history(self, wire, path, revision, limit): |
|
258 | 290 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
259 | 291 | region = self._region(wire) |
|
292 | ||
|
260 | 293 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
261 | 294 | def _assert_correct_path(_context_uid, _repo_id, _path, _revision, _limit): |
|
262 | 295 | cross_copies = False |
@@ -276,9 +309,11 b' class SvnRemote(RemoteBase):' | |||
|
276 | 309 | return history_revisions |
|
277 | 310 | return _assert_correct_path(context_uid, repo_id, path, revision, limit) |
|
278 | 311 | |
|
312 | @reraise_safe_exceptions | |
|
279 | 313 | def node_properties(self, wire, path, revision): |
|
280 | 314 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
281 | 315 | region = self._region(wire) |
|
316 | ||
|
282 | 317 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
283 | 318 | def _node_properties(_repo_id, _path, _revision): |
|
284 | 319 | repo = self._factory.repo(wire) |
@@ -288,7 +323,7 b' class SvnRemote(RemoteBase):' | |||
|
288 | 323 | return _node_properties(repo_id, path, revision) |
|
289 | 324 | |
|
290 | 325 | def file_annotate(self, wire, path, revision): |
|
291 | abs_path = 'file://' + urllib.pathname2url( | |
|
326 | abs_path = 'file://' + urllib.request.pathname2url( | |
|
292 | 327 | vcspath.join(wire['path'], path)) |
|
293 | 328 | file_uri = svn.core.svn_path_canonicalize(abs_path) |
|
294 | 329 | |
@@ -314,12 +349,14 b' class SvnRemote(RemoteBase):' | |||
|
314 | 349 | |
|
315 | 350 | return annotations |
|
316 | 351 | |
|
317 | def get_node_type(self, wire, path, revision=None): | |
|
352 | @reraise_safe_exceptions | |
|
353 | def get_node_type(self, wire, revision=None, path=''): | |
|
318 | 354 | |
|
319 | 355 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
320 | 356 | region = self._region(wire) |
|
357 | ||
|
321 | 358 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
322 |
def _get_node_type(_repo_id, _ |
|
|
359 | def _get_node_type(_repo_id, _revision, _path): | |
|
323 | 360 | repo = self._factory.repo(wire) |
|
324 | 361 | fs_ptr = svn.repos.fs(repo) |
|
325 | 362 | if _revision is None: |
@@ -327,12 +364,14 b' class SvnRemote(RemoteBase):' | |||
|
327 | 364 | root = svn.fs.revision_root(fs_ptr, _revision) |
|
328 | 365 | node = svn.fs.check_path(root, path) |
|
329 | 366 | return NODE_TYPE_MAPPING.get(node, None) |
|
330 |
return _get_node_type(repo_id, |
|
|
367 | return _get_node_type(repo_id, revision, path) | |
|
331 | 368 | |
|
332 | def get_nodes(self, wire, path, revision=None): | |
|
369 | @reraise_safe_exceptions | |
|
370 | def get_nodes(self, wire, revision=None, path=''): | |
|
333 | 371 | |
|
334 | 372 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
335 | 373 | region = self._region(wire) |
|
374 | ||
|
336 | 375 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
337 | 376 | def _get_nodes(_repo_id, _path, _revision): |
|
338 | 377 | repo = self._factory.repo(wire) |
@@ -342,27 +381,32 b' class SvnRemote(RemoteBase):' | |||
|
342 | 381 | root = svn.fs.revision_root(fsobj, _revision) |
|
343 | 382 | entries = svn.fs.dir_entries(root, path) |
|
344 | 383 | result = [] |
|
345 |
for entry_path, entry_info in entries. |
|
|
384 | for entry_path, entry_info in entries.items(): | |
|
346 | 385 | result.append( |
|
347 | 386 | (entry_path, NODE_TYPE_MAPPING.get(entry_info.kind, None))) |
|
348 | 387 | return result |
|
349 | 388 | return _get_nodes(repo_id, path, revision) |
|
350 | 389 | |
|
351 | def get_file_content(self, wire, path, rev=None): | |
|
390 | @reraise_safe_exceptions | |
|
391 | def get_file_content(self, wire, rev=None, path=''): | |
|
352 | 392 | repo = self._factory.repo(wire) |
|
353 | 393 | fsobj = svn.repos.fs(repo) |
|
394 | ||
|
354 | 395 | if rev is None: |
|
355 |
rev = svn.fs.youngest_rev |
|
|
396 | rev = svn.fs.youngest_rev(fsobj) | |
|
397 | ||
|
356 | 398 | root = svn.fs.revision_root(fsobj, rev) |
|
357 | 399 | content = svn.core.Stream(svn.fs.file_contents(root, path)) |
|
358 | return content.read() | |
|
400 | return BytesEnvelope(content.read()) | |
|
359 | 401 | |
|
360 | def get_file_size(self, wire, path, revision=None): | |
|
402 | @reraise_safe_exceptions | |
|
403 | def get_file_size(self, wire, revision=None, path=''): | |
|
361 | 404 | |
|
362 | 405 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
363 | 406 | region = self._region(wire) |
|
407 | ||
|
364 | 408 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
365 |
def _get_file_size(_repo_id, _ |
|
|
409 | def _get_file_size(_repo_id, _revision, _path): | |
|
366 | 410 | repo = self._factory.repo(wire) |
|
367 | 411 | fsobj = svn.repos.fs(repo) |
|
368 | 412 | if _revision is None: |
@@ -370,17 +414,17 b' class SvnRemote(RemoteBase):' | |||
|
370 | 414 | root = svn.fs.revision_root(fsobj, _revision) |
|
371 | 415 | size = svn.fs.file_length(root, path) |
|
372 | 416 | return size |
|
373 |
return _get_file_size(repo_id, |
|
|
417 | return _get_file_size(repo_id, revision, path) | |
|
374 | 418 | |
|
375 | 419 | def create_repository(self, wire, compatible_version=None): |
|
376 | 420 | log.info('Creating Subversion repository in path "%s"', wire['path']) |
|
377 | 421 | self._factory.repo(wire, create=True, |
|
378 | 422 | compatible_version=compatible_version) |
|
379 | 423 | |
|
380 | def get_url_and_credentials(self, src_url): | |
|
381 | obj = urlparse.urlparse(src_url) | |
|
382 |
username = obj.username or |
|
|
383 |
password = obj.password or |
|
|
424 | def get_url_and_credentials(self, src_url) -> tuple[str, str, str]: | |
|
425 | obj = urllib.parse.urlparse(src_url) | |
|
426 | username = obj.username or '' | |
|
427 | password = obj.password or '' | |
|
384 | 428 | return username, password, src_url |
|
385 | 429 | |
|
386 | 430 | def import_remote_repository(self, wire, src_url): |
@@ -411,16 +455,17 b' class SvnRemote(RemoteBase):' | |||
|
411 | 455 | log.debug('Return process ended with code: %s', rdump.returncode) |
|
412 | 456 | if rdump.returncode != 0: |
|
413 | 457 | errors = rdump.stderr.read() |
|
414 | log.error('svnrdump dump failed: statuscode %s: message: %s', | |
|
415 | rdump.returncode, errors) | |
|
458 | log.error('svnrdump dump failed: statuscode %s: message: %s', rdump.returncode, errors) | |
|
459 | ||
|
416 | 460 | reason = 'UNKNOWN' |
|
417 | if 'svnrdump: E230001:' in errors: | |
|
461 | if b'svnrdump: E230001:' in errors: | |
|
418 | 462 | reason = 'INVALID_CERTIFICATE' |
|
419 | 463 | |
|
420 | 464 | if reason == 'UNKNOWN': |
|
421 |
reason = 'UNKNOWN:{ |
|
|
465 | reason = f'UNKNOWN:{safe_str(errors)}' | |
|
466 | ||
|
422 | 467 | raise Exception( |
|
423 |
'Failed to dump the remote repository from |
|
|
468 | 'Failed to dump the remote repository from {}. Reason:{}'.format( | |
|
424 | 469 | src_url, reason)) |
|
425 | 470 | if load.returncode != 0: |
|
426 | 471 | raise Exception( |
@@ -428,8 +473,9 b' class SvnRemote(RemoteBase):' | |||
|
428 | 473 | (src_url, )) |
|
429 | 474 | |
|
430 | 475 | def commit(self, wire, message, author, timestamp, updated, removed): |
|
431 | assert isinstance(message, str) | |
|
432 | assert isinstance(author, str) | |
|
476 | ||
|
477 | message = safe_bytes(message) | |
|
478 | author = safe_bytes(author) | |
|
433 | 479 | |
|
434 | 480 | repo = self._factory.repo(wire) |
|
435 | 481 | fsobj = svn.repos.fs(repo) |
@@ -453,6 +499,7 b' class SvnRemote(RemoteBase):' | |||
|
453 | 499 | log.debug('Committed revision "%s" to "%s".', commit_id, wire['path']) |
|
454 | 500 | return commit_id |
|
455 | 501 | |
|
502 | @reraise_safe_exceptions | |
|
456 | 503 | def diff(self, wire, rev1, rev2, path1=None, path2=None, |
|
457 | 504 | ignore_whitespace=False, context=3): |
|
458 | 505 | |
@@ -461,12 +508,12 b' class SvnRemote(RemoteBase):' | |||
|
461 | 508 | diff_creator = SvnDiffer( |
|
462 | 509 | repo, rev1, path1, rev2, path2, ignore_whitespace, context) |
|
463 | 510 | try: |
|
464 | return diff_creator.generate_diff() | |
|
511 | return BytesEnvelope(diff_creator.generate_diff()) | |
|
465 | 512 | except svn.core.SubversionException as e: |
|
466 | 513 | log.exception( |
|
467 | 514 | "Error during diff operation operation. " |
|
468 |
"Path might not exist %s, %s" |
|
|
469 |
return |
|
|
515 | "Path might not exist %s, %s", path1, path2) | |
|
516 | return BytesEnvelope(b'') | |
|
470 | 517 | |
|
471 | 518 | @reraise_safe_exceptions |
|
472 | 519 | def is_large_file(self, wire, path): |
@@ -475,16 +522,29 b' class SvnRemote(RemoteBase):' | |||
|
475 | 522 | @reraise_safe_exceptions |
|
476 | 523 | def is_binary(self, wire, rev, path): |
|
477 | 524 | cache_on, context_uid, repo_id = self._cache_on(wire) |
|
525 | region = self._region(wire) | |
|
478 | 526 | |
|
479 | region = self._region(wire) | |
|
480 | 527 | @region.conditional_cache_on_arguments(condition=cache_on) |
|
481 | 528 | def _is_binary(_repo_id, _rev, _path): |
|
482 |
raw_bytes = self.get_file_content(wire, |
|
|
483 | return raw_bytes and '\0' in raw_bytes | |
|
529 | raw_bytes = self.get_file_content(wire, rev, path) | |
|
530 | if not raw_bytes: | |
|
531 | return False | |
|
532 | return b'\0' in raw_bytes | |
|
484 | 533 | |
|
485 | 534 | return _is_binary(repo_id, rev, path) |
|
486 | 535 | |
|
487 | 536 | @reraise_safe_exceptions |
|
537 | def md5_hash(self, wire, rev, path): | |
|
538 | cache_on, context_uid, repo_id = self._cache_on(wire) | |
|
539 | region = self._region(wire) | |
|
540 | ||
|
541 | @region.conditional_cache_on_arguments(condition=cache_on) | |
|
542 | def _md5_hash(_repo_id, _rev, _path): | |
|
543 | return '' | |
|
544 | ||
|
545 | return _md5_hash(repo_id, rev, path) | |
|
546 | ||
|
547 | @reraise_safe_exceptions | |
|
488 | 548 | def run_svn_command(self, wire, cmd, **opts): |
|
489 | 549 | path = wire.get('path', None) |
|
490 | 550 | |
@@ -500,14 +560,14 b' class SvnRemote(RemoteBase):' | |||
|
500 | 560 | |
|
501 | 561 | try: |
|
502 | 562 | _opts.update(opts) |
|
503 | p = subprocessio.SubprocessIOChunker(cmd, **_opts) | |
|
563 | proc = subprocessio.SubprocessIOChunker(cmd, **_opts) | |
|
504 | 564 | |
|
505 |
return ''.join(p), ''.join(p.err |
|
|
506 |
except |
|
|
565 | return b''.join(proc), b''.join(proc.stderr) | |
|
566 | except OSError as err: | |
|
507 | 567 | if safe_call: |
|
508 | 568 | return '', safe_str(err).strip() |
|
509 | 569 | else: |
|
510 | cmd = ' '.join(cmd) # human friendly CMD | |
|
570 | cmd = ' '.join(map(safe_str, cmd)) # human friendly CMD | |
|
511 | 571 | tb_err = ("Couldn't run svn command (%s).\n" |
|
512 | 572 | "Original error was:%s\n" |
|
513 | 573 | "Call options:%s\n" |
@@ -522,9 +582,8 b' class SvnRemote(RemoteBase):' | |||
|
522 | 582 | binary_dir = settings.BINARY_DIR |
|
523 | 583 | executable = None |
|
524 | 584 | if binary_dir: |
|
525 | executable = os.path.join(binary_dir, 'python') | |
|
526 | return install_svn_hooks( | |
|
527 | repo_path, executable=executable, force_create=force) | |
|
585 | executable = os.path.join(binary_dir, 'python3') | |
|
586 | return install_svn_hooks(repo_path, force_create=force) | |
|
528 | 587 | |
|
529 | 588 | @reraise_safe_exceptions |
|
530 | 589 | def get_hooks_info(self, wire): |
@@ -541,13 +600,14 b' class SvnRemote(RemoteBase):' | |||
|
541 | 600 | pass |
|
542 | 601 | |
|
543 | 602 | @reraise_safe_exceptions |
|
544 |
def archive_repo(self, wire, archive_ |
|
|
545 | archive_dir_name, commit_id): | |
|
603 | def archive_repo(self, wire, archive_name_key, kind, mtime, archive_at_path, | |
|
604 | archive_dir_name, commit_id, cache_config): | |
|
546 | 605 | |
|
547 | 606 | def walk_tree(root, root_dir, _commit_id): |
|
548 | 607 | """ |
|
549 | 608 | Special recursive svn repo walker |
|
550 | 609 | """ |
|
610 | root_dir = safe_bytes(root_dir) | |
|
551 | 611 | |
|
552 | 612 | filemode_default = 0o100644 |
|
553 | 613 | filemode_executable = 0o100755 |
@@ -560,10 +620,10 b' class SvnRemote(RemoteBase):' | |||
|
560 | 620 | # return only DIR, and then all entries in that dir |
|
561 | 621 | yield os.path.join(root_dir, f_name), {'mode': filemode_default}, f_type |
|
562 | 622 | new_root = os.path.join(root_dir, f_name) |
|
563 |
|
|
|
564 | yield _f_name, _f_data, _f_type | |
|
623 | yield from walk_tree(root, new_root, _commit_id) | |
|
565 | 624 | else: |
|
566 | f_path = os.path.join(root_dir, f_name).rstrip('/') | |
|
625 | ||
|
626 | f_path = os.path.join(root_dir, f_name).rstrip(b'/') | |
|
567 | 627 | prop_list = svn.fs.node_proplist(root, f_path) |
|
568 | 628 | |
|
569 | 629 | f_mode = filemode_default |
@@ -601,8 +661,8 b' class SvnRemote(RemoteBase):' | |||
|
601 | 661 | data_stream = f_data['content_stream'] |
|
602 | 662 | yield ArchiveNode(file_path, mode, is_link, data_stream) |
|
603 | 663 | |
|
604 | return archive_repo(file_walker, archive_dest_path, kind, mtime, archive_at_path, | |
|
605 | archive_dir_name, commit_id) | |
|
664 | return store_archive_in_cache( | |
|
665 | file_walker, archive_name_key, kind, mtime, archive_at_path, archive_dir_name, commit_id, cache_config=cache_config) | |
|
606 | 666 | |
|
607 | 667 | |
|
608 | 668 | class SvnDiffer(object): |
@@ -643,15 +703,15 b' class SvnDiffer(object):' | |||
|
643 | 703 | "Source type: %s, target type: %s" % |
|
644 | 704 | (self.src_kind, self.tgt_kind)) |
|
645 | 705 | |
|
646 | def generate_diff(self): | |
|
647 |
buf = |
|
|
706 | def generate_diff(self) -> bytes: | |
|
707 | buf = io.BytesIO() | |
|
648 | 708 | if self.tgt_kind == svn.core.svn_node_dir: |
|
649 | 709 | self._generate_dir_diff(buf) |
|
650 | 710 | else: |
|
651 | 711 | self._generate_file_diff(buf) |
|
652 | 712 | return buf.getvalue() |
|
653 | 713 | |
|
654 | def _generate_dir_diff(self, buf): | |
|
714 | def _generate_dir_diff(self, buf: io.BytesIO): | |
|
655 | 715 | editor = DiffChangeEditor() |
|
656 | 716 | editor_ptr, editor_baton = svn.delta.make_editor(editor) |
|
657 | 717 | svn.repos.dir_delta2( |
@@ -672,7 +732,7 b' class SvnDiffer(object):' | |||
|
672 | 732 | self._generate_node_diff( |
|
673 | 733 | buf, change, path, self.tgt_path, path, self.src_path) |
|
674 | 734 | |
|
675 | def _generate_file_diff(self, buf): | |
|
735 | def _generate_file_diff(self, buf: io.BytesIO): | |
|
676 | 736 | change = None |
|
677 | 737 | if self.src_kind == svn.core.svn_node_none: |
|
678 | 738 | change = "add" |
@@ -684,7 +744,13 b' class SvnDiffer(object):' | |||
|
684 | 744 | buf, change, tgt_path, tgt_base, src_path, src_base) |
|
685 | 745 | |
|
686 | 746 | def _generate_node_diff( |
|
687 | self, buf, change, tgt_path, tgt_base, src_path, src_base): | |
|
747 | self, buf: io.BytesIO, change, tgt_path, tgt_base, src_path, src_base): | |
|
748 | ||
|
749 | tgt_path_bytes = safe_bytes(tgt_path) | |
|
750 | tgt_path = safe_str(tgt_path) | |
|
751 | ||
|
752 | src_path_bytes = safe_bytes(src_path) | |
|
753 | src_path = safe_str(src_path) | |
|
688 | 754 | |
|
689 | 755 | if self.src_rev == self.tgt_rev and tgt_base == src_base: |
|
690 | 756 | # makes consistent behaviour with git/hg to return empty diff if |
@@ -697,55 +763,55 b' class SvnDiffer(object):' | |||
|
697 | 763 | self.binary_content = False |
|
698 | 764 | mime_type = self._get_mime_type(tgt_full_path) |
|
699 | 765 | |
|
700 | if mime_type and not mime_type.startswith('text'): | |
|
766 | if mime_type and not mime_type.startswith(b'text'): | |
|
701 | 767 | self.binary_content = True |
|
702 | buf.write("=" * 67 + '\n') | |
|
703 | buf.write("Cannot display: file marked as a binary type.\n") | |
|
704 | buf.write("svn:mime-type = %s\n" % mime_type) | |
|
705 |
buf.write("Index: % |
|
|
706 | buf.write("=" * 67 + '\n') | |
|
707 |
buf.write("diff --git a/% |
|
|
708 | 'tgt_path': tgt_path}) | |
|
768 | buf.write(b"=" * 67 + b'\n') | |
|
769 | buf.write(b"Cannot display: file marked as a binary type.\n") | |
|
770 | buf.write(b"svn:mime-type = %s\n" % mime_type) | |
|
771 | buf.write(b"Index: %b\n" % tgt_path_bytes) | |
|
772 | buf.write(b"=" * 67 + b'\n') | |
|
773 | buf.write(b"diff --git a/%b b/%b\n" % (tgt_path_bytes, tgt_path_bytes)) | |
|
709 | 774 | |
|
710 | 775 | if change == 'add': |
|
711 | 776 | # TODO: johbo: SVN is missing a zero here compared to git |
|
712 | buf.write("new file mode 10644\n") | |
|
777 | buf.write(b"new file mode 10644\n") | |
|
778 | ||
|
779 | # TODO(marcink): intro to binary detection of svn patches | |
|
780 | # if self.binary_content: | |
|
781 | # buf.write(b'GIT binary patch\n') | |
|
713 | 782 | |
|
714 | #TODO(marcink): intro to binary detection of svn patches | |
|
783 | buf.write(b"--- /dev/null\t(revision 0)\n") | |
|
784 | src_lines = [] | |
|
785 | else: | |
|
786 | if change == 'delete': | |
|
787 | buf.write(b"deleted file mode 10644\n") | |
|
788 | ||
|
789 | # TODO(marcink): intro to binary detection of svn patches | |
|
715 | 790 | # if self.binary_content: |
|
716 | 791 | # buf.write('GIT binary patch\n') |
|
717 | 792 | |
|
718 |
buf.write("--- |
|
|
719 | src_lines = [] | |
|
720 | else: | |
|
721 | if change == 'delete': | |
|
722 | buf.write("deleted file mode 10644\n") | |
|
723 | ||
|
724 | #TODO(marcink): intro to binary detection of svn patches | |
|
725 | # if self.binary_content: | |
|
726 | # buf.write('GIT binary patch\n') | |
|
727 | ||
|
728 | buf.write("--- a/%s\t(revision %s)\n" % ( | |
|
729 | src_path, self.src_rev)) | |
|
793 | buf.write(b"--- a/%b\t(revision %d)\n" % (src_path_bytes, self.src_rev)) | |
|
730 | 794 | src_lines = self._svn_readlines(self.src_root, src_full_path) |
|
731 | 795 | |
|
732 | 796 | if change == 'delete': |
|
733 |
buf.write("+++ /dev/null\t(revision % |
|
|
797 | buf.write(b"+++ /dev/null\t(revision %d)\n" % self.tgt_rev) | |
|
734 | 798 | tgt_lines = [] |
|
735 | 799 | else: |
|
736 |
buf.write("+++ b/% |
|
|
737 | tgt_path, self.tgt_rev)) | |
|
800 | buf.write(b"+++ b/%b\t(revision %d)\n" % (tgt_path_bytes, self.tgt_rev)) | |
|
738 | 801 | tgt_lines = self._svn_readlines(self.tgt_root, tgt_full_path) |
|
739 | 802 | |
|
803 | # we made our diff header, time to generate the diff content into our buffer | |
|
804 | ||
|
740 | 805 | if not self.binary_content: |
|
741 | 806 | udiff = svn_diff.unified_diff( |
|
742 | 807 | src_lines, tgt_lines, context=self.context, |
|
743 | 808 | ignore_blank_lines=self.ignore_whitespace, |
|
744 | 809 | ignore_case=False, |
|
745 | 810 | ignore_space_changes=self.ignore_whitespace) |
|
811 | ||
|
746 | 812 | buf.writelines(udiff) |
|
747 | 813 | |
|
748 | def _get_mime_type(self, path): | |
|
814 | def _get_mime_type(self, path) -> bytes: | |
|
749 | 815 | try: |
|
750 | 816 | mime_type = svn.fs.node_prop( |
|
751 | 817 | self.tgt_root, path, svn.core.SVN_PROP_MIME_TYPE) |
@@ -761,7 +827,9 b' class SvnDiffer(object):' | |||
|
761 | 827 | if node_kind not in ( |
|
762 | 828 | svn.core.svn_node_file, svn.core.svn_node_symlink): |
|
763 | 829 | return [] |
|
764 | content = svn.core.Stream(svn.fs.file_contents(fs_root, node_path)).read() | |
|
830 | content = svn.core.Stream( | |
|
831 | svn.fs.file_contents(fs_root, node_path)).read() | |
|
832 | ||
|
765 | 833 | return content.splitlines(True) |
|
766 | 834 | |
|
767 | 835 | |
@@ -799,7 +867,7 b' class TxnNodeProcessor(object):' | |||
|
799 | 867 | """ |
|
800 | 868 | |
|
801 | 869 | def __init__(self, node, txn_root): |
|
802 |
assert |
|
|
870 | assert_bytes(node['path']) | |
|
803 | 871 | |
|
804 | 872 | self.node = node |
|
805 | 873 | self.txn_root = txn_root |
@@ -835,23 +903,24 b' class TxnNodeProcessor(object):' | |||
|
835 | 903 | svn.fs.make_file(self.txn_root, self.node['path']) |
|
836 | 904 | |
|
837 | 905 | def _update_file_content(self): |
|
838 |
assert |
|
|
906 | assert_bytes(self.node['content']) | |
|
907 | ||
|
839 | 908 | handler, baton = svn.fs.apply_textdelta( |
|
840 | 909 | self.txn_root, self.node['path'], None, None) |
|
841 | 910 | svn.delta.svn_txdelta_send_string(self.node['content'], handler, baton) |
|
842 | 911 | |
|
843 | 912 | def _update_file_properties(self): |
|
844 | 913 | properties = self.node.get('properties', {}) |
|
845 |
for key, value in properties. |
|
|
914 | for key, value in properties.items(): | |
|
846 | 915 | svn.fs.change_node_prop( |
|
847 | self.txn_root, self.node['path'], key, value) | |
|
916 | self.txn_root, self.node['path'], safe_bytes(key), safe_bytes(value)) | |
|
848 | 917 | |
|
849 | 918 | |
|
850 | 919 | def apr_time_t(timestamp): |
|
851 | 920 | """ |
|
852 | 921 | Convert a Python timestamp into APR timestamp type apr_time_t |
|
853 | 922 | """ |
|
854 | return timestamp * 1E6 | |
|
923 | return int(timestamp * 1E6) | |
|
855 | 924 | |
|
856 | 925 | |
|
857 | 926 | def svn_opt_revision_value_t(num): |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -27,7 +27,7 b' import mercurial.hgweb.hgweb_mod' | |||
|
27 | 27 | import webob.exc |
|
28 | 28 | |
|
29 | 29 | from vcsserver import pygrack, exceptions, settings, git_lfs |
|
30 | ||
|
30 | from vcsserver.str_utils import ascii_bytes, safe_bytes | |
|
31 | 31 | |
|
32 | 32 | log = logging.getLogger(__name__) |
|
33 | 33 | |
@@ -81,7 +81,7 b' class HgWeb(mercurial.hgweb.hgweb_mod.hg' | |||
|
81 | 81 | first_chunk = None |
|
82 | 82 | |
|
83 | 83 | try: |
|
84 |
data = |
|
|
84 | data = next(gen) | |
|
85 | 85 | |
|
86 | 86 | def first_chunk(): |
|
87 | 87 | yield data |
@@ -94,16 +94,17 b' class HgWeb(mercurial.hgweb.hgweb_mod.hg' | |||
|
94 | 94 | |
|
95 | 95 | def _runwsgi(self, req, res, repo): |
|
96 | 96 | |
|
97 | cmd = req.qsparams.get('cmd', '') | |
|
97 | cmd = req.qsparams.get(b'cmd', '') | |
|
98 | 98 | if not mercurial.wireprotoserver.iscmd(cmd): |
|
99 | 99 | # NOTE(marcink): for unsupported commands, we return bad request |
|
100 | 100 | # internally from HG |
|
101 | log.warning('cmd: `%s` is not supported by the mercurial wireprotocol v1', cmd) | |
|
101 | 102 | from mercurial.hgweb.common import statusmessage |
|
102 | 103 | res.status = statusmessage(mercurial.hgweb.common.HTTP_BAD_REQUEST) |
|
103 | res.setbodybytes('') | |
|
104 | res.setbodybytes(b'') | |
|
104 | 105 | return res.sendresponse() |
|
105 | 106 | |
|
106 |
return super( |
|
|
107 | return super()._runwsgi(req, res, repo) | |
|
107 | 108 | |
|
108 | 109 | |
|
109 | 110 | def make_hg_ui_from_config(repo_config): |
@@ -115,10 +116,13 b' def make_hg_ui_from_config(repo_config):' | |||
|
115 | 116 | baseui._tcfg = mercurial.config.config() |
|
116 | 117 | |
|
117 | 118 | for section, option, value in repo_config: |
|
118 |
baseui.setconfig( |
|
|
119 | baseui.setconfig( | |
|
120 | ascii_bytes(section, allow_bytes=True), | |
|
121 | ascii_bytes(option, allow_bytes=True), | |
|
122 | ascii_bytes(value, allow_bytes=True)) | |
|
119 | 123 | |
|
120 | 124 | # make our hgweb quiet so it doesn't print output |
|
121 | baseui.setconfig('ui', 'quiet', 'true') | |
|
125 | baseui.setconfig(b'ui', b'quiet', b'true') | |
|
122 | 126 | |
|
123 | 127 | return baseui |
|
124 | 128 | |
@@ -131,11 +135,14 b' def update_hg_ui_from_hgrc(baseui, repo_' | |||
|
131 | 135 | return |
|
132 | 136 | log.debug('reading hgrc from %s', path) |
|
133 | 137 | cfg = mercurial.config.config() |
|
134 | cfg.read(path) | |
|
138 | cfg.read(ascii_bytes(path)) | |
|
135 | 139 | for section in HG_UI_SECTIONS: |
|
136 | 140 | for k, v in cfg.items(section): |
|
137 | 141 | log.debug('settings ui from file: [%s] %s=%s', section, k, v) |
|
138 |
baseui.setconfig( |
|
|
142 | baseui.setconfig( | |
|
143 | ascii_bytes(section, allow_bytes=True), | |
|
144 | ascii_bytes(k, allow_bytes=True), | |
|
145 | ascii_bytes(v, allow_bytes=True)) | |
|
139 | 146 | |
|
140 | 147 | |
|
141 | 148 | def create_hg_wsgi_app(repo_path, repo_name, config): |
@@ -151,7 +158,7 b' def create_hg_wsgi_app(repo_path, repo_n' | |||
|
151 | 158 | update_hg_ui_from_hgrc(baseui, repo_path) |
|
152 | 159 | |
|
153 | 160 | try: |
|
154 | return HgWeb(repo_path, name=repo_name, baseui=baseui) | |
|
161 | return HgWeb(safe_bytes(repo_path), name=safe_bytes(repo_name), baseui=baseui) | |
|
155 | 162 | except mercurial.error.RequirementError as e: |
|
156 | 163 | raise exceptions.RequirementException(e)(e) |
|
157 | 164 | |
@@ -225,10 +232,10 b' class GitLFSHandler(object):' | |||
|
225 | 232 | |
|
226 | 233 | def create_git_lfs_wsgi_app(repo_path, repo_name, config): |
|
227 | 234 | git_path = settings.GIT_EXECUTABLE |
|
228 | update_server_info = config.pop('git_update_server_info') | |
|
229 | git_lfs_enabled = config.pop('git_lfs_enabled') | |
|
230 | git_lfs_store_path = config.pop('git_lfs_store_path') | |
|
231 | git_lfs_http_scheme = config.pop('git_lfs_http_scheme', 'http') | |
|
235 | update_server_info = config.pop(b'git_update_server_info') | |
|
236 | git_lfs_enabled = config.pop(b'git_lfs_enabled') | |
|
237 | git_lfs_store_path = config.pop(b'git_lfs_store_path') | |
|
238 | git_lfs_http_scheme = config.pop(b'git_lfs_http_scheme', 'http') | |
|
232 | 239 | app = GitLFSHandler( |
|
233 | 240 | repo_path, repo_name, git_path, update_server_info, config) |
|
234 | 241 |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -23,15 +23,17 b' along with git_http_backend.py Project.' | |||
|
23 | 23 | If not, see <http://www.gnu.org/licenses/>. |
|
24 | 24 | """ |
|
25 | 25 | import os |
|
26 | import collections | |
|
26 | 27 | import logging |
|
27 |
import |
|
|
28 | from collections import deque | |
|
29 | from threading import Event, Thread | |
|
28 | import subprocess | |
|
29 | import threading | |
|
30 | ||
|
31 | from vcsserver.str_utils import safe_str | |
|
30 | 32 | |
|
31 | 33 | log = logging.getLogger(__name__) |
|
32 | 34 | |
|
33 | 35 | |
|
34 | class StreamFeeder(Thread): | |
|
36 | class StreamFeeder(threading.Thread): | |
|
35 | 37 | """ |
|
36 | 38 | Normal writing into pipe-like is blocking once the buffer is filled. |
|
37 | 39 | This thread allows a thread to seep data from a file-like into a pipe |
@@ -40,24 +42,18 b' class StreamFeeder(Thread):' | |||
|
40 | 42 | """ |
|
41 | 43 | |
|
42 | 44 | def __init__(self, source): |
|
43 |
super( |
|
|
45 | super().__init__() | |
|
44 | 46 | self.daemon = True |
|
45 | 47 | filelike = False |
|
46 |
self.bytes = b |
|
|
47 |
if type(source) in ( |
|
|
48 | self.bytes = b'' | |
|
49 | if type(source) in (str, bytes, bytearray): # string-like | |
|
48 | 50 | self.bytes = bytes(source) |
|
49 | 51 | else: # can be either file pointer or file-like |
|
50 |
if |
|
|
52 | if isinstance(source, int): # file pointer it is | |
|
51 | 53 | # converting file descriptor (int) stdin into file-like |
|
52 | try: | |
|
53 | source = os.fdopen(source, 'rb', 16384) | |
|
54 | except Exception: | |
|
55 | pass | |
|
54 | source = os.fdopen(source, 'rb', 16384) | |
|
56 | 55 | # let's see if source is file-like by now |
|
57 | try: | |
|
58 | filelike = source.read | |
|
59 | except Exception: | |
|
60 | pass | |
|
56 | filelike = hasattr(source, 'read') | |
|
61 | 57 | if not filelike and not self.bytes: |
|
62 | 58 | raise TypeError("StreamFeeder's source object must be a readable " |
|
63 | 59 | "file-like, a file descriptor, or a string-like.") |
@@ -65,28 +61,31 b' class StreamFeeder(Thread):' | |||
|
65 | 61 | self.readiface, self.writeiface = os.pipe() |
|
66 | 62 | |
|
67 | 63 | def run(self): |
|
68 | t = self.writeiface | |
|
64 | writer = self.writeiface | |
|
69 | 65 | try: |
|
70 | 66 | if self.bytes: |
|
71 | os.write(t, self.bytes) | |
|
67 | os.write(writer, self.bytes) | |
|
72 | 68 | else: |
|
73 | 69 | s = self.source |
|
74 | b = s.read(4096) | |
|
75 |
while |
|
|
76 |
|
|
|
77 |
|
|
|
70 | ||
|
71 | while 1: | |
|
72 | _bytes = s.read(4096) | |
|
73 | if not _bytes: | |
|
74 | break | |
|
75 | os.write(writer, _bytes) | |
|
76 | ||
|
78 | 77 | finally: |
|
79 | os.close(t) | |
|
78 | os.close(writer) | |
|
80 | 79 | |
|
81 | 80 | @property |
|
82 | 81 | def output(self): |
|
83 | 82 | return self.readiface |
|
84 | 83 | |
|
85 | 84 | |
|
86 | class InputStreamChunker(Thread): | |
|
85 | class InputStreamChunker(threading.Thread): | |
|
87 | 86 | def __init__(self, source, target, buffer_size, chunk_size): |
|
88 | 87 | |
|
89 |
super( |
|
|
88 | super().__init__() | |
|
90 | 89 | |
|
91 | 90 | self.daemon = True # die die die. |
|
92 | 91 | |
@@ -95,16 +94,16 b' class InputStreamChunker(Thread):' | |||
|
95 | 94 | self.chunk_count_max = int(buffer_size / chunk_size) + 1 |
|
96 | 95 | self.chunk_size = chunk_size |
|
97 | 96 | |
|
98 | self.data_added = Event() | |
|
97 | self.data_added = threading.Event() | |
|
99 | 98 | self.data_added.clear() |
|
100 | 99 | |
|
101 | self.keep_reading = Event() | |
|
100 | self.keep_reading = threading.Event() | |
|
102 | 101 | self.keep_reading.set() |
|
103 | 102 | |
|
104 | self.EOF = Event() | |
|
103 | self.EOF = threading.Event() | |
|
105 | 104 | self.EOF.clear() |
|
106 | 105 | |
|
107 | self.go = Event() | |
|
106 | self.go = threading.Event() | |
|
108 | 107 | self.go.set() |
|
109 | 108 | |
|
110 | 109 | def stop(self): |
@@ -115,7 +114,7 b' class InputStreamChunker(Thread):' | |||
|
115 | 114 | # go of the input because, if successful, .close() will send EOF |
|
116 | 115 | # down the pipe. |
|
117 | 116 | self.source.close() |
|
118 | except: | |
|
117 | except Exception: | |
|
119 | 118 | pass |
|
120 | 119 | |
|
121 | 120 | def run(self): |
@@ -146,7 +145,7 b' class InputStreamChunker(Thread):' | |||
|
146 | 145 | |
|
147 | 146 | try: |
|
148 | 147 | b = s.read(cs) |
|
149 | except ValueError: | |
|
148 | except ValueError: # probably "I/O operation on closed file" | |
|
150 | 149 | b = '' |
|
151 | 150 | |
|
152 | 151 | self.EOF.set() |
@@ -166,18 +165,20 b' class BufferedGenerator(object):' | |||
|
166 | 165 | StopIteration after the last chunk of data is yielded. |
|
167 | 166 | """ |
|
168 | 167 | |
|
169 | def __init__(self, source, buffer_size=65536, chunk_size=4096, | |
|
168 | def __init__(self, name, source, buffer_size=65536, chunk_size=4096, | |
|
170 | 169 | starting_values=None, bottomless=False): |
|
171 | 170 | starting_values = starting_values or [] |
|
171 | self.name = name | |
|
172 | self.buffer_size = buffer_size | |
|
173 | self.chunk_size = chunk_size | |
|
172 | 174 | |
|
173 | 175 | if bottomless: |
|
174 | 176 | maxlen = int(buffer_size / chunk_size) |
|
175 | 177 | else: |
|
176 | 178 | maxlen = None |
|
177 | 179 | |
|
178 | self.data = deque(starting_values, maxlen) | |
|
179 | self.worker = InputStreamChunker(source, self.data, buffer_size, | |
|
180 | chunk_size) | |
|
180 | self.data_queue = collections.deque(starting_values, maxlen) | |
|
181 | self.worker = InputStreamChunker(source, self.data_queue, buffer_size, chunk_size) | |
|
181 | 182 | if starting_values: |
|
182 | 183 | self.worker.data_added.set() |
|
183 | 184 | self.worker.start() |
@@ -185,17 +186,21 b' class BufferedGenerator(object):' | |||
|
185 | 186 | #################### |
|
186 | 187 | # Generator's methods |
|
187 | 188 | #################### |
|
189 | def __str__(self): | |
|
190 | return f'BufferedGenerator(name={self.name} chunk: {self.chunk_size} on buffer: {self.buffer_size})' | |
|
188 | 191 | |
|
189 | 192 | def __iter__(self): |
|
190 | 193 | return self |
|
191 | 194 | |
|
192 | def next(self): | |
|
193 | while not len(self.data) and not self.worker.EOF.is_set(): | |
|
195 | def __next__(self): | |
|
196 | ||
|
197 | while not self.length and not self.worker.EOF.is_set(): | |
|
194 | 198 | self.worker.data_added.clear() |
|
195 | 199 | self.worker.data_added.wait(0.2) |
|
196 | if len(self.data): | |
|
200 | ||
|
201 | if self.length: | |
|
197 | 202 | self.worker.keep_reading.set() |
|
198 | return bytes(self.data.popleft()) | |
|
203 | return bytes(self.data_queue.popleft()) | |
|
199 | 204 | elif self.worker.EOF.is_set(): |
|
200 | 205 | raise StopIteration |
|
201 | 206 | |
@@ -249,7 +254,7 b' class BufferedGenerator(object):' | |||
|
249 | 254 | @property |
|
250 | 255 | def done_reading(self): |
|
251 | 256 | """ |
|
252 | Done_reding does not mean that the iterator's buffer is empty. | |
|
257 | Done_reading does not mean that the iterator's buffer is empty. | |
|
253 | 258 | Iterator might have done reading from underlying source, but the read |
|
254 | 259 | chunks might still be available for serving through .next() method. |
|
255 | 260 | |
@@ -262,31 +267,31 b' class BufferedGenerator(object):' | |||
|
262 | 267 | """ |
|
263 | 268 | returns int. |
|
264 | 269 | |
|
265 |
This is the leng |
|
|
270 | This is the length of the queue of chunks, not the length of | |
|
266 | 271 | the combined contents in those chunks. |
|
267 | 272 | |
|
268 | 273 | __len__() cannot be meaningfully implemented because this |
|
269 | reader is just flying throuh a bottomless pit content and | |
|
270 |
can only know the leng |
|
|
274 | reader is just flying through a bottomless pit content and | |
|
275 | can only know the length of what it already saw. | |
|
271 | 276 | |
|
272 | 277 | If __len__() on WSGI server per PEP 3333 returns a value, |
|
273 |
the respon |
|
|
278 | the response's length will be set to that. In order not to | |
|
274 | 279 | confuse WSGI PEP3333 servers, we will not implement __len__ |
|
275 | 280 | at all. |
|
276 | 281 | """ |
|
277 | return len(self.data) | |
|
282 | return len(self.data_queue) | |
|
278 | 283 | |
|
279 | 284 | def prepend(self, x): |
|
280 | self.data.appendleft(x) | |
|
285 | self.data_queue.appendleft(x) | |
|
281 | 286 | |
|
282 | 287 | def append(self, x): |
|
283 | self.data.append(x) | |
|
288 | self.data_queue.append(x) | |
|
284 | 289 | |
|
285 | 290 | def extend(self, o): |
|
286 | self.data.extend(o) | |
|
291 | self.data_queue.extend(o) | |
|
287 | 292 | |
|
288 | 293 | def __getitem__(self, i): |
|
289 | return self.data[i] | |
|
294 | return self.data_queue[i] | |
|
290 | 295 | |
|
291 | 296 | |
|
292 | 297 | class SubprocessIOChunker(object): |
@@ -314,7 +319,7 b' class SubprocessIOChunker(object):' | |||
|
314 | 319 | |
|
315 | 320 | - We are multithreaded. Writing in and reading out, err are all sep threads. |
|
316 | 321 | - We support concurrent (in and out) stream processing. |
|
317 |
- The output is not a stream. It's a queue of read string (bytes, not |
|
|
322 | - The output is not a stream. It's a queue of read string (bytes, not str) | |
|
318 | 323 | chunks. The object behaves as an iterable. You can "for chunk in obj:" us. |
|
319 | 324 | - We are non-blocking in more respects than communicate() |
|
320 | 325 | (reading from subprocess out pauses when internal buffer is full, but |
@@ -323,16 +328,16 b' class SubprocessIOChunker(object):' | |||
|
323 | 328 | does not block the parallel inpipe reading occurring parallel thread.) |
|
324 | 329 | |
|
325 | 330 | The purpose of the object is to allow us to wrap subprocess interactions into |
|
326 |
an |
|
|
331 | an iterable that can be passed to a WSGI server as the application's return | |
|
327 | 332 | value. Because of stream-processing-ability, WSGI does not have to read ALL |
|
328 | 333 | of the subprocess's output and buffer it, before handing it to WSGI server for |
|
329 | 334 | HTTP response. Instead, the class initializer reads just a bit of the stream |
|
330 | to figure out if error ocurred or likely to occur and if not, just hands the | |
|
335 | to figure out if error occurred or likely to occur and if not, just hands the | |
|
331 | 336 | further iteration over subprocess output to the server for completion of HTTP |
|
332 | 337 | response. |
|
333 | 338 | |
|
334 | 339 | The real or perceived subprocess error is trapped and raised as one of |
|
335 |
|
|
|
340 | OSError family of exceptions | |
|
336 | 341 | |
|
337 | 342 | Example usage: |
|
338 | 343 | # try: |
@@ -342,7 +347,7 b' class SubprocessIOChunker(object):' | |||
|
342 | 347 | # buffer_size = 65536, |
|
343 | 348 | # chunk_size = 4096 |
|
344 | 349 | # ) |
|
345 |
# except ( |
|
|
350 | # except (OSError) as e: | |
|
346 | 351 | # print str(e) |
|
347 | 352 | # raise e |
|
348 | 353 | # |
@@ -358,15 +363,17 b' class SubprocessIOChunker(object):' | |||
|
358 | 363 | _close_input_fd = None |
|
359 | 364 | |
|
360 | 365 | _closed = False |
|
366 | _stdout = None | |
|
367 | _stderr = None | |
|
361 | 368 | |
|
362 | def __init__(self, cmd, inputstream=None, buffer_size=65536, | |
|
369 | def __init__(self, cmd, input_stream=None, buffer_size=65536, | |
|
363 | 370 | chunk_size=4096, starting_values=None, fail_on_stderr=True, |
|
364 | 371 | fail_on_return_code=True, **kwargs): |
|
365 | 372 | """ |
|
366 | 373 | Initializes SubprocessIOChunker |
|
367 | 374 | |
|
368 | 375 | :param cmd: A Subprocess.Popen style "cmd". Can be string or array of strings |
|
369 | :param inputstream: (Default: None) A file-like, string, or file pointer. | |
|
376 | :param input_stream: (Default: None) A file-like, string, or file pointer. | |
|
370 | 377 | :param buffer_size: (Default: 65536) A size of total buffer per stream in bytes. |
|
371 | 378 | :param chunk_size: (Default: 4096) A max size of a chunk. Actual chunk may be smaller. |
|
372 | 379 | :param starting_values: (Default: []) An array of strings to put in front of output que. |
@@ -376,71 +383,86 b' class SubprocessIOChunker(object):' | |||
|
376 | 383 | exception if the return code is not 0. |
|
377 | 384 | """ |
|
378 | 385 | |
|
386 | kwargs['shell'] = kwargs.get('shell', True) | |
|
387 | ||
|
379 | 388 | starting_values = starting_values or [] |
|
380 | if inputstream: | |
|
381 | input_streamer = StreamFeeder(inputstream) | |
|
389 | if input_stream: | |
|
390 | input_streamer = StreamFeeder(input_stream) | |
|
382 | 391 | input_streamer.start() |
|
383 | inputstream = input_streamer.output | |
|
384 | self._close_input_fd = inputstream | |
|
392 | input_stream = input_streamer.output | |
|
393 | self._close_input_fd = input_stream | |
|
385 | 394 | |
|
386 | 395 | self._fail_on_stderr = fail_on_stderr |
|
387 | 396 | self._fail_on_return_code = fail_on_return_code |
|
388 | ||
|
389 | _shell = kwargs.get('shell', True) | |
|
390 | kwargs['shell'] = _shell | |
|
397 | self.cmd = cmd | |
|
391 | 398 | |
|
392 | _p = subprocess.Popen(cmd, bufsize=-1, | |
|
393 | stdin=inputstream, | |
|
394 | stdout=subprocess.PIPE, | |
|
395 | stderr=subprocess.PIPE, | |
|
399 | _p = subprocess.Popen(cmd, bufsize=-1, stdin=input_stream, stdout=subprocess.PIPE, stderr=subprocess.PIPE, | |
|
396 | 400 | **kwargs) |
|
401 | self.process = _p | |
|
397 | 402 | |
|
398 | bg_out = BufferedGenerator(_p.stdout, buffer_size, chunk_size, | |
|
399 | starting_values) | |
|
400 | bg_err = BufferedGenerator(_p.stderr, 16000, 1, bottomless=True) | |
|
403 | bg_out = BufferedGenerator('stdout', _p.stdout, buffer_size, chunk_size, starting_values) | |
|
404 | bg_err = BufferedGenerator('stderr', _p.stderr, 10240, 1, bottomless=True) | |
|
401 | 405 | |
|
402 | 406 | while not bg_out.done_reading and not bg_out.reading_paused and not bg_err.length: |
|
403 | 407 | # doing this until we reach either end of file, or end of buffer. |
|
404 |
bg_out.data_added_event.wait( |
|
|
408 | bg_out.data_added_event.wait(0.2) | |
|
405 | 409 | bg_out.data_added_event.clear() |
|
406 | 410 | |
|
407 | 411 | # at this point it's still ambiguous if we are done reading or just full buffer. |
|
408 | 412 | # Either way, if error (returned by ended process, or implied based on |
|
409 | 413 | # presence of stuff in stderr output) we error out. |
|
410 | 414 | # Else, we are happy. |
|
411 |
|
|
|
415 | return_code = _p.poll() | |
|
416 | ret_code_ok = return_code in [None, 0] | |
|
417 | ret_code_fail = return_code is not None and return_code != 0 | |
|
418 | if ( | |
|
419 | (ret_code_fail and fail_on_return_code) or | |
|
420 | (ret_code_ok and fail_on_stderr and bg_err.length) | |
|
421 | ): | |
|
412 | 422 | |
|
413 | if ((_returncode and fail_on_return_code) or | |
|
414 | (fail_on_stderr and _returncode is None and bg_err.length)): | |
|
415 | 423 | try: |
|
416 | 424 | _p.terminate() |
|
417 | 425 | except Exception: |
|
418 | 426 | pass |
|
427 | ||
|
419 | 428 | bg_out.stop() |
|
429 | out = b''.join(bg_out) | |
|
430 | self._stdout = out | |
|
431 | ||
|
420 | 432 | bg_err.stop() |
|
421 | if fail_on_stderr: | |
|
422 |
|
|
|
423 | raise EnvironmentError( | |
|
424 | "Subprocess exited due to an error:\n" + err) | |
|
425 | if _returncode and fail_on_return_code: | |
|
426 |
|
|
|
433 | err = b''.join(bg_err) | |
|
434 | self._stderr = err | |
|
435 | ||
|
436 | # code from https://github.com/schacon/grack/pull/7 | |
|
437 | if err.strip() == b'fatal: The remote end hung up unexpectedly' and out.startswith(b'0034shallow '): | |
|
438 | bg_out = iter([out]) | |
|
439 | _p = None | |
|
440 | elif err and fail_on_stderr: | |
|
441 | text_err = err.decode() | |
|
442 | raise OSError( | |
|
443 | f"Subprocess exited due to an error:\n{text_err}") | |
|
444 | ||
|
445 | if ret_code_fail and fail_on_return_code: | |
|
446 | text_err = err.decode() | |
|
427 | 447 | if not err: |
|
428 | 448 | # maybe get empty stderr, try stdout instead |
|
429 | 449 | # in many cases git reports the errors on stdout too |
|
430 |
err = |
|
|
431 |
raise |
|
|
432 |
"Subprocess exited with non 0 ret code: |
|
|
433 | _returncode, err)) | |
|
450 | text_err = out.decode() | |
|
451 | raise OSError( | |
|
452 | f"Subprocess exited with non 0 ret code:{return_code}: stderr:{text_err}") | |
|
434 | 453 | |
|
435 |
self. |
|
|
436 |
self. |
|
|
437 | self.error = bg_err | |
|
438 | self.inputstream = inputstream | |
|
454 | self.stdout = bg_out | |
|
455 | self.stderr = bg_err | |
|
456 | self.inputstream = input_stream | |
|
457 | ||
|
458 | def __str__(self): | |
|
459 | proc = getattr(self, 'process', 'NO_PROCESS') | |
|
460 | return f'SubprocessIOChunker: {proc}' | |
|
439 | 461 | |
|
440 | 462 | def __iter__(self): |
|
441 | 463 | return self |
|
442 | 464 | |
|
443 | def next(self): | |
|
465 | def __next__(self): | |
|
444 | 466 | # Note: mikhail: We need to be sure that we are checking the return |
|
445 | 467 | # code after the stdout stream is closed. Some processes, e.g. git |
|
446 | 468 | # are doing some magic in between closing stdout and terminating the |
@@ -449,27 +471,31 b' class SubprocessIOChunker(object):' | |||
|
449 | 471 | result = None |
|
450 | 472 | stop_iteration = None |
|
451 | 473 | try: |
|
452 |
result = self. |
|
|
474 | result = next(self.stdout) | |
|
453 | 475 | except StopIteration as e: |
|
454 | 476 | stop_iteration = e |
|
455 | 477 | |
|
456 | if self.process.poll() and self._fail_on_return_code: | |
|
457 | err = '%s' % ''.join(self.error) | |
|
458 | raise EnvironmentError( | |
|
459 | "Subprocess exited due to an error:\n" + err) | |
|
478 | if self.process: | |
|
479 | return_code = self.process.poll() | |
|
480 | ret_code_fail = return_code is not None and return_code != 0 | |
|
481 | if ret_code_fail and self._fail_on_return_code: | |
|
482 | self.stop_streams() | |
|
483 | err = self.get_stderr() | |
|
484 | raise OSError( | |
|
485 | f"Subprocess exited (exit_code:{return_code}) due to an error during iteration:\n{err}") | |
|
460 | 486 | |
|
461 | 487 | if stop_iteration: |
|
462 | 488 | raise stop_iteration |
|
463 | 489 | return result |
|
464 | 490 | |
|
465 | def throw(self, type, value=None, traceback=None): | |
|
466 |
if self. |
|
|
467 | raise type(value) | |
|
491 | def throw(self, exc_type, value=None, traceback=None): | |
|
492 | if self.stdout.length or not self.stdout.done_reading: | |
|
493 | raise exc_type(value) | |
|
468 | 494 | |
|
469 | 495 | def close(self): |
|
470 | 496 | if self._closed: |
|
471 | 497 | return |
|
472 | self._closed = True | |
|
498 | ||
|
473 | 499 | try: |
|
474 | 500 | self.process.terminate() |
|
475 | 501 | except Exception: |
@@ -477,11 +503,11 b' class SubprocessIOChunker(object):' | |||
|
477 | 503 | if self._close_input_fd: |
|
478 | 504 | os.close(self._close_input_fd) |
|
479 | 505 | try: |
|
480 |
self. |
|
|
506 | self.stdout.close() | |
|
481 | 507 | except Exception: |
|
482 | 508 | pass |
|
483 | 509 | try: |
|
484 |
self.err |
|
|
510 | self.stderr.close() | |
|
485 | 511 | except Exception: |
|
486 | 512 | pass |
|
487 | 513 | try: |
@@ -489,6 +515,24 b' class SubprocessIOChunker(object):' | |||
|
489 | 515 | except Exception: |
|
490 | 516 | pass |
|
491 | 517 | |
|
518 | self._closed = True | |
|
519 | ||
|
520 | def stop_streams(self): | |
|
521 | getattr(self.stdout, 'stop', lambda: None)() | |
|
522 | getattr(self.stderr, 'stop', lambda: None)() | |
|
523 | ||
|
524 | def get_stdout(self): | |
|
525 | if self._stdout: | |
|
526 | return self._stdout | |
|
527 | else: | |
|
528 | return b''.join(self.stdout) | |
|
529 | ||
|
530 | def get_stderr(self): | |
|
531 | if self._stderr: | |
|
532 | return self._stderr | |
|
533 | else: | |
|
534 | return b''.join(self.stderr) | |
|
535 | ||
|
492 | 536 | |
|
493 | 537 | def run_command(arguments, env=None): |
|
494 | 538 | """ |
@@ -506,9 +550,9 b' def run_command(arguments, env=None):' | |||
|
506 | 550 | if env: |
|
507 | 551 | _opts.update({'env': env}) |
|
508 | 552 | proc = SubprocessIOChunker(cmd, **_opts) |
|
509 |
return ''.join(proc), ''.join(proc.err |
|
|
510 |
except |
|
|
511 |
cmd = ' '.join(cmd) |
|
|
553 | return b''.join(proc), b''.join(proc.stderr) | |
|
554 | except OSError as err: | |
|
555 | cmd = ' '.join(map(safe_str, cmd)) # human friendly CMD | |
|
512 | 556 | tb_err = ("Couldn't run subprocessio command (%s).\n" |
|
513 | 557 | "Original error was:%s\n" % (cmd, err)) |
|
514 | 558 | log.exception(tb_err) |
@@ -1,7 +1,7 b'' | |||
|
1 | # -*- coding: utf-8 -*- | |
|
2 | 1 |
|
|
3 | 2 | # Copyright (C) 2004-2009 Edgewall Software |
|
4 | 3 | # Copyright (C) 2004-2006 Christopher Lenz <cmlenz@gmx.de> |
|
4 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
5 | 5 | # All rights reserved. |
|
6 | 6 | # |
|
7 | 7 | # This software is licensed as described in the file COPYING, which |
@@ -17,15 +17,15 b'' | |||
|
17 | 17 | import difflib |
|
18 | 18 | |
|
19 | 19 | |
|
20 | def get_filtered_hunks(fromlines, tolines, context=None, | |
|
21 | ignore_blank_lines=False, ignore_case=False, | |
|
22 | ignore_space_changes=False): | |
|
20 | def get_filtered_hunks(from_lines, to_lines, context=None, | |
|
21 | ignore_blank_lines: bool = False, ignore_case: bool = False, | |
|
22 | ignore_space_changes: bool = False): | |
|
23 | 23 | """Retrieve differences in the form of `difflib.SequenceMatcher` |
|
24 | 24 | opcodes, grouped according to the ``context`` and ``ignore_*`` |
|
25 | 25 | parameters. |
|
26 | 26 | |
|
27 | :param fromlines: list of lines corresponding to the old content | |
|
28 | :param tolines: list of lines corresponding to the new content | |
|
27 | :param from_lines: list of lines corresponding to the old content | |
|
28 | :param to_lines: list of lines corresponding to the new content | |
|
29 | 29 | :param ignore_blank_lines: differences about empty lines only are ignored |
|
30 | 30 | :param ignore_case: upper case / lower case only differences are ignored |
|
31 | 31 | :param ignore_space_changes: differences in amount of spaces are ignored |
@@ -37,27 +37,27 b' def get_filtered_hunks(fromlines, toline' | |||
|
37 | 37 | to filter out the results will come straight from the |
|
38 | 38 | SequenceMatcher. |
|
39 | 39 | """ |
|
40 | hunks = get_hunks(fromlines, tolines, context) | |
|
40 | hunks = get_hunks(from_lines, to_lines, context) | |
|
41 | 41 | if ignore_space_changes or ignore_case or ignore_blank_lines: |
|
42 | hunks = filter_ignorable_lines(hunks, fromlines, tolines, context, | |
|
42 | hunks = filter_ignorable_lines(hunks, from_lines, to_lines, context, | |
|
43 | 43 | ignore_blank_lines, ignore_case, |
|
44 | 44 | ignore_space_changes) |
|
45 | 45 | return hunks |
|
46 | 46 | |
|
47 | 47 | |
|
48 | def get_hunks(fromlines, tolines, context=None): | |
|
48 | def get_hunks(from_lines, to_lines, context=None): | |
|
49 | 49 | """Generator yielding grouped opcodes describing differences . |
|
50 | 50 | |
|
51 | 51 | See `get_filtered_hunks` for the parameter descriptions. |
|
52 | 52 | """ |
|
53 | matcher = difflib.SequenceMatcher(None, fromlines, tolines) | |
|
53 | matcher = difflib.SequenceMatcher(None, from_lines, to_lines) | |
|
54 | 54 | if context is None: |
|
55 | 55 | return (hunk for hunk in [matcher.get_opcodes()]) |
|
56 | 56 | else: |
|
57 | 57 | return matcher.get_grouped_opcodes(context) |
|
58 | 58 | |
|
59 | 59 | |
|
60 | def filter_ignorable_lines(hunks, fromlines, tolines, context, | |
|
60 | def filter_ignorable_lines(hunks, from_lines, to_lines, context, | |
|
61 | 61 | ignore_blank_lines, ignore_case, |
|
62 | 62 | ignore_space_changes): |
|
63 | 63 | """Detect line changes that should be ignored and emits them as |
@@ -67,11 +67,12 b' def filter_ignorable_lines(hunks, fromli' | |||
|
67 | 67 | See `get_filtered_hunks` for the parameter descriptions. |
|
68 | 68 | """ |
|
69 | 69 | def is_ignorable(tag, fromlines, tolines): |
|
70 | ||
|
70 | 71 | if tag == 'delete' and ignore_blank_lines: |
|
71 | if ''.join(fromlines) == '': | |
|
72 | if b''.join(fromlines) == b'': | |
|
72 | 73 | return True |
|
73 | 74 | elif tag == 'insert' and ignore_blank_lines: |
|
74 | if ''.join(tolines) == '': | |
|
75 | if b''.join(tolines) == b'': | |
|
75 | 76 | return True |
|
76 | 77 | elif tag == 'replace' and (ignore_case or ignore_space_changes): |
|
77 | 78 | if len(fromlines) != len(tolines): |
@@ -81,7 +82,7 b' def filter_ignorable_lines(hunks, fromli' | |||
|
81 | 82 | if ignore_case: |
|
82 | 83 | input_str = input_str.lower() |
|
83 | 84 | if ignore_space_changes: |
|
84 | input_str = ' '.join(input_str.split()) | |
|
85 | input_str = b' '.join(input_str.split()) | |
|
85 | 86 | return input_str |
|
86 | 87 | |
|
87 | 88 | for i in range(len(fromlines)): |
@@ -101,7 +102,7 b' def filter_ignorable_lines(hunks, fromli' | |||
|
101 | 102 | else: |
|
102 | 103 | prev = (tag, i1, i2, j1, j2) |
|
103 | 104 | else: |
|
104 | if is_ignorable(tag, fromlines[i1:i2], tolines[j1:j2]): | |
|
105 | if is_ignorable(tag, from_lines[i1:i2], to_lines[j1:j2]): | |
|
105 | 106 | ignored_lines = True |
|
106 | 107 | if prev: |
|
107 | 108 | prev = 'equal', prev[1], i2, prev[3], j2 |
@@ -125,10 +126,11 b' def filter_ignorable_lines(hunks, fromli' | |||
|
125 | 126 | nn = n + n |
|
126 | 127 | |
|
127 | 128 | group = [] |
|
129 | ||
|
128 | 130 | def all_equal(): |
|
129 | 131 | all(op[0] == 'equal' for op in group) |
|
130 | 132 | for idx, (tag, i1, i2, j1, j2) in enumerate(opcodes): |
|
131 | if idx == 0 and tag == 'equal': # Fixup leading unchanged block | |
|
133 | if idx == 0 and tag == 'equal': # Fixup leading unchanged block | |
|
132 | 134 | i1, j1 = max(i1, i2 - n), max(j1, j2 - n) |
|
133 | 135 | elif tag == 'equal' and i2 - i1 > nn: |
|
134 | 136 | group.append((tag, i1, min(i2, i1 + n), j1, |
@@ -140,7 +142,7 b' def filter_ignorable_lines(hunks, fromli' | |||
|
140 | 142 | group.append((tag, i1, i2, j1, j2)) |
|
141 | 143 | |
|
142 | 144 | if group and not (len(group) == 1 and group[0][0] == 'equal'): |
|
143 | if group[-1][0] == 'equal': # Fixup trailing unchanged block | |
|
145 | if group[-1][0] == 'equal': # Fixup trailing unchanged block | |
|
144 | 146 | tag, i1, i2, j1, j2 = group[-1] |
|
145 | 147 | group[-1] = tag, i1, min(i2, i1 + n), j1, min(j2, j1 + n) |
|
146 | 148 | if not all_equal(): |
@@ -150,22 +152,30 b' def filter_ignorable_lines(hunks, fromli' | |||
|
150 | 152 | yield hunk |
|
151 | 153 | |
|
152 | 154 | |
|
153 | NO_NEWLINE_AT_END = '\\ No newline at end of file' | |
|
155 | NO_NEWLINE_AT_END = b'\\ No newline at end of file' | |
|
156 | LINE_TERM = b'\n' | |
|
154 | 157 | |
|
155 | 158 | |
|
156 |
def unified_diff(fromlines, tolines, context=None, ignore_blank_lines= |
|
|
157 |
ignore_case= |
|
|
159 | def unified_diff(from_lines, to_lines, context=None, ignore_blank_lines: bool = False, | |
|
160 | ignore_case: bool = False, ignore_space_changes: bool = False, lineterm=LINE_TERM) -> bytes: | |
|
158 | 161 | """ |
|
159 | 162 | Generator producing lines corresponding to a textual diff. |
|
160 | 163 | |
|
161 | 164 | See `get_filtered_hunks` for the parameter descriptions. |
|
162 | 165 | """ |
|
163 | 166 | # TODO: johbo: Check if this can be nicely integrated into the matching |
|
167 | ||
|
164 | 168 | if ignore_space_changes: |
|
165 | fromlines = [l.strip() for l in fromlines] | |
|
166 | tolines = [l.strip() for l in tolines] | |
|
169 | from_lines = [l.strip() for l in from_lines] | |
|
170 | to_lines = [l.strip() for l in to_lines] | |
|
167 | 171 | |
|
168 | for group in get_filtered_hunks(fromlines, tolines, context, | |
|
172 | def _hunk_range(start, length) -> bytes: | |
|
173 | if length != 1: | |
|
174 | return b'%d,%d' % (start, length) | |
|
175 | else: | |
|
176 | return b'%d' % (start,) | |
|
177 | ||
|
178 | for group in get_filtered_hunks(from_lines, to_lines, context, | |
|
169 | 179 | ignore_blank_lines, ignore_case, |
|
170 | 180 | ignore_space_changes): |
|
171 | 181 | i1, i2, j1, j2 = group[0][1], group[-1][2], group[0][3], group[-1][4] |
@@ -173,37 +183,30 b' def unified_diff(fromlines, tolines, con' | |||
|
173 | 183 | i1, i2 = -1, -1 # support for Add changes |
|
174 | 184 | if j1 == 0 and j2 == 0: |
|
175 | 185 | j1, j2 = -1, -1 # support for Delete changes |
|
176 |
yield '@@ -% |
|
|
186 | yield b'@@ -%b +%b @@%b' % ( | |
|
177 | 187 | _hunk_range(i1 + 1, i2 - i1), |
|
178 | 188 | _hunk_range(j1 + 1, j2 - j1), |
|
179 | 189 | lineterm) |
|
180 | 190 | for tag, i1, i2, j1, j2 in group: |
|
181 | 191 | if tag == 'equal': |
|
182 | for line in fromlines[i1:i2]: | |
|
192 | for line in from_lines[i1:i2]: | |
|
183 | 193 | if not line.endswith(lineterm): |
|
184 | yield ' ' + line + lineterm | |
|
194 | yield b' ' + line + lineterm | |
|
185 | 195 | yield NO_NEWLINE_AT_END + lineterm |
|
186 | 196 | else: |
|
187 | yield ' ' + line | |
|
197 | yield b' ' + line | |
|
188 | 198 | else: |
|
189 | 199 | if tag in ('replace', 'delete'): |
|
190 | for line in fromlines[i1:i2]: | |
|
200 | for line in from_lines[i1:i2]: | |
|
191 | 201 | if not line.endswith(lineterm): |
|
192 | yield '-' + line + lineterm | |
|
202 | yield b'-' + line + lineterm | |
|
193 | 203 | yield NO_NEWLINE_AT_END + lineterm |
|
194 | 204 | else: |
|
195 | yield '-' + line | |
|
205 | yield b'-' + line | |
|
196 | 206 | if tag in ('replace', 'insert'): |
|
197 | for line in tolines[j1:j2]: | |
|
207 | for line in to_lines[j1:j2]: | |
|
198 | 208 | if not line.endswith(lineterm): |
|
199 | yield '+' + line + lineterm | |
|
209 | yield b'+' + line + lineterm | |
|
200 | 210 | yield NO_NEWLINE_AT_END + lineterm |
|
201 | 211 | else: |
|
202 | yield '+' + line | |
|
203 | ||
|
204 | ||
|
205 | def _hunk_range(start, length): | |
|
206 | if length != 1: | |
|
207 | return '%d,%d' % (start, length) | |
|
208 | else: | |
|
209 | return '%d' % (start, ) | |
|
212 | yield b'+' + line |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -18,8 +18,7 b'' | |||
|
18 | 18 | import os |
|
19 | 19 | import shutil |
|
20 | 20 | import tempfile |
|
21 | ||
|
22 | import configobj | |
|
21 | import configparser | |
|
23 | 22 | |
|
24 | 23 | |
|
25 | 24 | class ContextINI(object): |
@@ -53,17 +52,17 b' class ContextINI(object):' | |||
|
53 | 52 | with open(self.new_path, 'wb'): |
|
54 | 53 | pass |
|
55 | 54 | |
|
56 |
|
|
|
57 | self.new_path, file_error=True, write_empty_values=True) | |
|
55 | parser = configparser.ConfigParser() | |
|
56 | parser.read(self.ini_file_path) | |
|
58 | 57 | |
|
59 | 58 | for data in self.ini_params: |
|
60 | section, ini_params = data.items()[0] | |
|
61 | key, val = ini_params.items()[0] | |
|
62 |
if section not in |
|
|
63 |
|
|
|
64 |
|
|
|
65 | ||
|
66 |
|
|
|
59 | section, ini_params = list(data.items())[0] | |
|
60 | key, val = list(ini_params.items())[0] | |
|
61 | if section not in parser: | |
|
62 | parser[section] = {} | |
|
63 | parser[section][key] = val | |
|
64 | with open(self.ini_file_path, 'w') as f: | |
|
65 | parser.write(f) | |
|
67 | 66 | return self.new_path |
|
68 | 67 | |
|
69 | 68 | def __exit__(self, exc_type, exc_val, exc_tb): |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -21,8 +21,7 b' import pytest' | |||
|
21 | 21 | import dulwich.errors |
|
22 | 22 | from mock import Mock, patch |
|
23 | 23 | |
|
24 | from vcsserver import git | |
|
25 | ||
|
24 | from vcsserver.remote import git | |
|
26 | 25 | |
|
27 | 26 | SAMPLE_REFS = { |
|
28 | 27 | 'HEAD': 'fd627b9e0dd80b47be81af07c4a98518244ed2f7', |
@@ -49,7 +48,7 b' def test_discover_git_version(git_remote' | |||
|
49 | 48 | |
|
50 | 49 | |
|
51 | 50 | class TestGitFetch(object): |
|
52 | def setup(self): | |
|
51 | def setup_method(self): | |
|
53 | 52 | self.mock_repo = Mock() |
|
54 | 53 | factory = Mock() |
|
55 | 54 | factory.repo = Mock(return_value=self.mock_repo) |
@@ -67,8 +66,8 b' class TestGitFetch(object):' | |||
|
67 | 66 | |
|
68 | 67 | def test_fetches_specified_commits(self): |
|
69 | 68 | selected_refs = { |
|
70 | 'refs/tags/v0.1.8': '74ebce002c088b8a5ecf40073db09375515ecd68', | |
|
71 | 'refs/tags/v0.1.3': '5a3a8fb005554692b16e21dee62bf02667d8dc3e', | |
|
69 | 'refs/tags/v0.1.8': b'74ebce002c088b8a5ecf40073db09375515ecd68', | |
|
70 | 'refs/tags/v0.1.3': b'5a3a8fb005554692b16e21dee62bf02667d8dc3e', | |
|
72 | 71 | } |
|
73 | 72 | |
|
74 | 73 | def side_effect(determine_wants, *args, **kwargs): |
@@ -80,7 +79,7 b' class TestGitFetch(object):' | |||
|
80 | 79 | mock_fetch.side_effect = side_effect |
|
81 | 80 | self.remote_git.pull( |
|
82 | 81 | wire={}, url='/tmp/', apply_refs=False, |
|
83 | refs=selected_refs.keys()) | |
|
82 | refs=list(selected_refs.keys())) | |
|
84 | 83 | determine_wants = self.mock_repo.object_store.determine_wants_all |
|
85 | 84 | assert determine_wants.call_count == 0 |
|
86 | 85 | |
@@ -93,7 +92,7 b' class TestGitFetch(object):' | |||
|
93 | 92 | 'refs/tags/v0.1.3': '5a3a8fb005554692b16e21dee62bf02667d8dc3e', |
|
94 | 93 | } |
|
95 | 94 | |
|
96 | with patch('vcsserver.git.Repo', create=False) as mock_repo: | |
|
95 | with patch('vcsserver.remote.git.Repo', create=False) as mock_repo: | |
|
97 | 96 | mock_repo().get_refs.return_value = sample_refs |
|
98 | 97 | remote_refs = remote_git.get_remote_refs(wire={}, url=url) |
|
99 | 98 | mock_repo().get_refs.assert_called_once_with() |
@@ -114,7 +113,7 b' class TestReraiseSafeExceptions(object):' | |||
|
114 | 113 | methods = inspect.getmembers(git_remote, predicate=inspect.ismethod) |
|
115 | 114 | for method_name, method in methods: |
|
116 | 115 | if not method_name.startswith('_') and method_name not in ['vcsserver_invalidate_cache']: |
|
117 |
assert method. |
|
|
116 | assert method.__func__.__code__ == decorator.__code__ | |
|
118 | 117 | |
|
119 | 118 | @pytest.mark.parametrize('side_effect, expected_type', [ |
|
120 | 119 | (dulwich.errors.ChecksumMismatch('0000000', 'deadbeef'), 'lookup'), |
@@ -138,11 +137,14 b' class TestReraiseSafeExceptions(object):' | |||
|
138 | 137 | class TestDulwichRepoWrapper(object): |
|
139 | 138 | def test_calls_close_on_delete(self): |
|
140 | 139 | isdir_patcher = patch('dulwich.repo.os.path.isdir', return_value=True) |
|
141 | with isdir_patcher: | |
|
142 | repo = git.Repo('/tmp/abcde') | |
|
143 | with patch.object(git.DulwichRepo, 'close') as close_mock: | |
|
144 |
|
|
|
145 | close_mock.assert_called_once_with() | |
|
140 | with patch.object(git.Repo, 'close') as close_mock: | |
|
141 | with isdir_patcher: | |
|
142 | repo = git.Repo('/tmp/abcde') | |
|
143 | assert repo is not None | |
|
144 | repo.__del__() | |
|
145 | # can't use del repo as in python3 this isn't always calling .__del__() | |
|
146 | ||
|
147 | close_mock.assert_called_once_with() | |
|
146 | 148 | |
|
147 | 149 | |
|
148 | 150 | class TestGitFactory(object): |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -21,9 +21,10 b' import traceback' | |||
|
21 | 21 | |
|
22 | 22 | import pytest |
|
23 | 23 | from mercurial.error import LookupError |
|
24 |
from mock import Mock, |
|
|
24 | from mock import Mock, patch | |
|
25 | 25 | |
|
26 |
from vcsserver import exceptions, |
|
|
26 | from vcsserver import exceptions, hgcompat | |
|
27 | from vcsserver.remote import hg | |
|
27 | 28 | |
|
28 | 29 | |
|
29 | 30 | class TestDiff(object): |
@@ -32,8 +33,8 b' class TestDiff(object):' | |||
|
32 | 33 | factory = Mock() |
|
33 | 34 | hg_remote = hg.HgRemote(factory) |
|
34 | 35 | with patch('mercurial.patch.diff') as diff_mock: |
|
35 | diff_mock.side_effect = LookupError( | |
|
36 | 'deadbeef', 'index', 'message') | |
|
36 | diff_mock.side_effect = LookupError(b'deadbeef', b'index', b'message') | |
|
37 | ||
|
37 | 38 | with pytest.raises(Exception) as exc_info: |
|
38 | 39 | hg_remote.diff( |
|
39 | 40 | wire={}, commit_id_1='deadbeef', commit_id_2='deadbee1', |
@@ -51,13 +52,13 b' class TestReraiseSafeExceptions(object):' | |||
|
51 | 52 | decorator = hg.reraise_safe_exceptions(None) |
|
52 | 53 | for method_name, method in methods: |
|
53 | 54 | if not method_name.startswith('_') and method_name not in ['vcsserver_invalidate_cache']: |
|
54 |
assert method. |
|
|
55 | assert method.__func__.__code__ == decorator.__code__ | |
|
55 | 56 | |
|
56 | 57 | @pytest.mark.parametrize('side_effect, expected_type', [ |
|
57 | (hgcompat.Abort(), 'abort'), | |
|
58 | (hgcompat.InterventionRequired(), 'abort'), | |
|
58 | (hgcompat.Abort('failed-abort'), 'abort'), | |
|
59 | (hgcompat.InterventionRequired('intervention-required'), 'abort'), | |
|
59 | 60 | (hgcompat.RepoLookupError(), 'lookup'), |
|
60 | (hgcompat.LookupError('deadbeef', 'index', 'message'), 'lookup'), | |
|
61 | (hgcompat.LookupError(b'deadbeef', b'index', b'message'), 'lookup'), | |
|
61 | 62 | (hgcompat.RepoError(), 'error'), |
|
62 | 63 | (hgcompat.RequirementError(), 'requirement'), |
|
63 | 64 | ]) |
@@ -75,10 +76,9 b' class TestReraiseSafeExceptions(object):' | |||
|
75 | 76 | @hg.reraise_safe_exceptions |
|
76 | 77 | def fake_method(): |
|
77 | 78 | try: |
|
78 | raise hgcompat.Abort() | |
|
79 | raise hgcompat.Abort('test-abort') | |
|
79 | 80 | except: |
|
80 | self.original_traceback = traceback.format_tb( | |
|
81 | sys.exc_info()[2]) | |
|
81 | self.original_traceback = traceback.format_tb(sys.exc_info()[2]) | |
|
82 | 82 | raise |
|
83 | 83 | |
|
84 | 84 | try: |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -21,14 +21,14 b' import pytest' | |||
|
21 | 21 | from vcsserver import hgcompat, hgpatches |
|
22 | 22 | |
|
23 | 23 | |
|
24 | LARGEFILES_CAPABILITY = 'largefiles=serve' | |
|
24 | LARGEFILES_CAPABILITY = b'largefiles=serve' | |
|
25 | 25 | |
|
26 | 26 | |
|
27 | 27 | def test_patch_largefiles_capabilities_applies_patch( |
|
28 | 28 | patched_capabilities): |
|
29 | 29 | lfproto = hgcompat.largefiles.proto |
|
30 | 30 | hgpatches.patch_largefiles_capabilities() |
|
31 |
assert lfproto._capabilities. |
|
|
31 | assert lfproto._capabilities.__name__ == '_dynamic_capabilities' | |
|
32 | 32 | |
|
33 | 33 | |
|
34 | 34 | def test_dynamic_capabilities_uses_original_function_if_not_enabled( |
@@ -72,11 +72,6 b' def test_dynamic_capabilities_uses_large' | |||
|
72 | 72 | assert LARGEFILES_CAPABILITY in caps |
|
73 | 73 | |
|
74 | 74 | |
|
75 | def test_hgsubversion_import(): | |
|
76 | from hgsubversion import svnrepo | |
|
77 | assert svnrepo | |
|
78 | ||
|
79 | ||
|
80 | 75 | @pytest.fixture |
|
81 | 76 | def patched_capabilities(request): |
|
82 | 77 | """ |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -15,17 +15,18 b'' | |||
|
15 | 15 | # along with this program; if not, write to the Free Software Foundation, |
|
16 | 16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
17 | 17 | |
|
18 | import contextlib | |
|
19 | import io | |
|
20 | 18 | import threading |
|
21 | from BaseHTTPServer import BaseHTTPRequestHandler | |
|
22 | from SocketServer import TCPServer | |
|
19 | import msgpack | |
|
20 | ||
|
21 | from http.server import BaseHTTPRequestHandler | |
|
22 | from socketserver import TCPServer | |
|
23 | 23 | |
|
24 | 24 | import mercurial.ui |
|
25 | 25 | import mock |
|
26 | 26 | import pytest |
|
27 | import simplejson as json | |
|
28 | 27 | |
|
28 | from vcsserver.hooks import HooksHttpClient | |
|
29 | from vcsserver.lib.rc_json import json | |
|
29 | 30 | from vcsserver import hooks |
|
30 | 31 | |
|
31 | 32 | |
@@ -44,7 +45,7 b' def get_hg_ui(extras=None):' | |||
|
44 | 45 | } |
|
45 | 46 | required_extras.update(extras) |
|
46 | 47 | hg_ui = mercurial.ui.ui() |
|
47 | hg_ui.setconfig('rhodecode', 'RC_SCM_DATA', json.dumps(required_extras)) | |
|
48 | hg_ui.setconfig(b'rhodecode', b'RC_SCM_DATA', json.dumps(required_extras)) | |
|
48 | 49 | |
|
49 | 50 | return hg_ui |
|
50 | 51 | |
@@ -67,6 +68,7 b' def test_git_post_receive_is_disabled():' | |||
|
67 | 68 | |
|
68 | 69 | def test_git_post_receive_calls_repo_size(): |
|
69 | 70 | extras = {'hooks': ['push', 'repo_size']} |
|
71 | ||
|
70 | 72 | with mock.patch.object(hooks, '_call_hook') as call_hook_mock: |
|
71 | 73 | hooks.git_post_receive( |
|
72 | 74 | None, '', {'RC_SCM_DATA': json.dumps(extras)}) |
@@ -81,6 +83,7 b' def test_git_post_receive_calls_repo_siz' | |||
|
81 | 83 | |
|
82 | 84 | def test_git_post_receive_does_not_call_disabled_repo_size(): |
|
83 | 85 | extras = {'hooks': ['push']} |
|
86 | ||
|
84 | 87 | with mock.patch.object(hooks, '_call_hook') as call_hook_mock: |
|
85 | 88 | hooks.git_post_receive( |
|
86 | 89 | None, '', {'RC_SCM_DATA': json.dumps(extras)}) |
@@ -149,18 +152,19 b' class TestHooksHttpClient(object):' | |||
|
149 | 152 | client = hooks.HooksHttpClient(uri) |
|
150 | 153 | assert client.hooks_uri == uri |
|
151 | 154 | |
|
152 |
def test_serialize_returns_ |
|
|
155 | def test_serialize_returns_serialized_string(self): | |
|
153 | 156 | client = hooks.HooksHttpClient('localhost:3000') |
|
154 | 157 | hook_name = 'test' |
|
155 | 158 | extras = { |
|
156 | 159 | 'first': 1, |
|
157 | 160 | 'second': 'two' |
|
158 | 161 | } |
|
159 | result = client._serialize(hook_name, extras) | |
|
160 |
expected_result = |
|
|
162 | hooks_proto, result = client._serialize(hook_name, extras) | |
|
163 | expected_result = msgpack.packb({ | |
|
161 | 164 | 'method': hook_name, |
|
162 | 'extras': extras | |
|
165 | 'extras': extras, | |
|
163 | 166 | }) |
|
167 | assert hooks_proto == {'rc-hooks-protocol': 'msgpack.v1', 'Connection': 'keep-alive'} | |
|
164 | 168 | assert result == expected_result |
|
165 | 169 | |
|
166 | 170 | def test_call_queries_http_server(self, http_mirror): |
@@ -171,10 +175,10 b' class TestHooksHttpClient(object):' | |||
|
171 | 175 | 'second': 'two' |
|
172 | 176 | } |
|
173 | 177 | result = client(hook_name, extras) |
|
174 | expected_result = { | |
|
178 | expected_result = msgpack.unpackb(msgpack.packb({ | |
|
175 | 179 | 'method': hook_name, |
|
176 | 180 | 'extras': extras |
|
177 | } | |
|
181 | }), raw=False) | |
|
178 | 182 | assert result == expected_result |
|
179 | 183 | |
|
180 | 184 | |
@@ -211,9 +215,10 b' def http_mirror(request):' | |||
|
211 | 215 | |
|
212 | 216 | |
|
213 | 217 | class MirrorHttpHandler(BaseHTTPRequestHandler): |
|
218 | ||
|
214 | 219 | def do_POST(self): |
|
215 | 220 | length = int(self.headers['Content-Length']) |
|
216 |
body = self.rfile.read(length) |
|
|
221 | body = self.rfile.read(length) | |
|
217 | 222 | self.send_response(200) |
|
218 | 223 | self.end_headers() |
|
219 | 224 | self.wfile.write(body) |
@@ -239,3 +244,43 b' class MirrorHttpServer(object):' | |||
|
239 | 244 | @property |
|
240 | 245 | def uri(self): |
|
241 | 246 | return '{}:{}'.format(self.ip_address, self.port) |
|
247 | ||
|
248 | ||
|
249 | def test_hooks_http_client_init(): | |
|
250 | hooks_uri = 'http://localhost:8000' | |
|
251 | client = HooksHttpClient(hooks_uri) | |
|
252 | assert client.hooks_uri == hooks_uri | |
|
253 | ||
|
254 | ||
|
255 | def test_hooks_http_client_call(): | |
|
256 | hooks_uri = 'http://localhost:8000' | |
|
257 | ||
|
258 | method = 'test_method' | |
|
259 | extras = {'key': 'value'} | |
|
260 | ||
|
261 | with \ | |
|
262 | mock.patch('http.client.HTTPConnection') as mock_connection,\ | |
|
263 | mock.patch('msgpack.load') as mock_load: | |
|
264 | ||
|
265 | client = HooksHttpClient(hooks_uri) | |
|
266 | ||
|
267 | mock_load.return_value = {'result': 'success'} | |
|
268 | response = mock.MagicMock() | |
|
269 | response.status = 200 | |
|
270 | mock_connection.request.side_effect = None | |
|
271 | mock_connection.getresponse.return_value = response | |
|
272 | ||
|
273 | result = client(method, extras) | |
|
274 | ||
|
275 | mock_connection.assert_called_with(hooks_uri) | |
|
276 | mock_connection.return_value.request.assert_called_once() | |
|
277 | assert result == {'result': 'success'} | |
|
278 | ||
|
279 | ||
|
280 | def test_hooks_http_client_serialize(): | |
|
281 | method = 'test_method' | |
|
282 | extras = {'key': 'value'} | |
|
283 | headers, body = HooksHttpClient._serialize(method, extras) | |
|
284 | ||
|
285 | assert headers == {'rc-hooks-protocol': HooksHttpClient.proto, 'Connection': 'keep-alive'} | |
|
286 | assert msgpack.unpackb(body) == {'method': method, 'extras': extras} |
@@ -30,13 +30,13 b' def data():' | |||
|
30 | 30 | |
|
31 | 31 | def test_http_app_streaming_with_data(data, repeat, vcs_app): |
|
32 | 32 | app = vcs_app |
|
33 |
for x in |
|
|
33 | for x in range(repeat // 10): | |
|
34 | 34 | response = app.post('/stream/git/', params=data) |
|
35 | 35 | assert response.status_code == 200 |
|
36 | 36 | |
|
37 | 37 | |
|
38 | 38 | def test_http_app_streaming_no_data(repeat, vcs_app): |
|
39 | 39 | app = vcs_app |
|
40 |
for x in |
|
|
40 | for x in range(repeat // 10): | |
|
41 | 41 | response = app.post('/stream/git/') |
|
42 | 42 | assert response.status_code == 200 |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -23,6 +23,7 b' import vcsserver' | |||
|
23 | 23 | import tempfile |
|
24 | 24 | from vcsserver import hook_utils |
|
25 | 25 | from vcsserver.tests.fixture import no_newline_id_generator |
|
26 | from vcsserver.str_utils import safe_bytes, safe_str | |
|
26 | 27 | from vcsserver.utils import AttributeDict |
|
27 | 28 | |
|
28 | 29 | |
@@ -31,7 +32,7 b' class TestCheckRhodecodeHook(object):' | |||
|
31 | 32 | def test_returns_false_when_hook_file_is_wrong_found(self, tmpdir): |
|
32 | 33 | hook = os.path.join(str(tmpdir), 'fake_hook_file.py') |
|
33 | 34 | with open(hook, 'wb') as f: |
|
34 | f.write('dummy test') | |
|
35 | f.write(b'dummy test') | |
|
35 | 36 | result = hook_utils.check_rhodecode_hook(hook) |
|
36 | 37 | assert result is False |
|
37 | 38 | |
@@ -47,7 +48,7 b' class TestCheckRhodecodeHook(object):' | |||
|
47 | 48 | def test_signatures(self, file_content, expected_result, tmpdir): |
|
48 | 49 | hook = os.path.join(str(tmpdir), 'fake_hook_file_1.py') |
|
49 | 50 | with open(hook, 'wb') as f: |
|
50 | f.write(file_content) | |
|
51 | f.write(safe_bytes(file_content)) | |
|
51 | 52 | |
|
52 | 53 | result = hook_utils.check_rhodecode_hook(hook) |
|
53 | 54 | |
@@ -71,8 +72,7 b' class BaseInstallHooks(object):' | |||
|
71 | 72 | content = hook_file.read() |
|
72 | 73 | |
|
73 | 74 | expected_env = '#!{}'.format(executable) |
|
74 | expected_rc_version = "\nRC_HOOK_VER = '{}'\n".format( | |
|
75 | vcsserver.__version__) | |
|
75 | expected_rc_version = "\nRC_HOOK_VER = '{}'\n".format(vcsserver.__version__) | |
|
76 | 76 | assert content.strip().startswith(expected_env) |
|
77 | 77 | assert expected_rc_version in content |
|
78 | 78 |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -42,8 +42,7 b' def test_applies_largefiles_patch_only_i' | |||
|
42 | 42 | ('bad', 'bad'), |
|
43 | 43 | ('query&foo=bar', 'query&foo=bar'), |
|
44 | 44 | ('equery&auth_token=bar', 'equery&auth_token=*****'), |
|
45 | ('a;b;c;query&foo=bar&auth_token=secret', | |
|
46 | 'a&b&c&query&foo=bar&auth_token=*****'), | |
|
45 | ('a;b;c;query&foo=bar&auth_token=secret', 'a;b;c;query&foo=bar&auth_token=*****'), | |
|
47 | 46 | ('', ''), |
|
48 | 47 | (None, None), |
|
49 | 48 | ('foo=bar', 'foo=bar'), |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -16,6 +16,7 b'' | |||
|
16 | 16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
17 | 17 | |
|
18 | 18 | import io |
|
19 | import more_itertools | |
|
19 | 20 | |
|
20 | 21 | import dulwich.protocol |
|
21 | 22 | import mock |
@@ -25,7 +26,7 b' import webtest' | |||
|
25 | 26 | |
|
26 | 27 | from vcsserver import hooks, pygrack |
|
27 | 28 | |
|
28 | # pylint: disable=redefined-outer-name,protected-access | |
|
29 | from vcsserver.str_utils import ascii_bytes | |
|
29 | 30 | |
|
30 | 31 | |
|
31 | 32 | @pytest.fixture() |
@@ -75,8 +76,7 b' def test_pre_pull_hook_fails_with_sideba' | |||
|
75 | 76 | '0000', |
|
76 | 77 | '0009done\n', |
|
77 | 78 | ]) |
|
78 | with mock.patch('vcsserver.hooks.git_pre_pull', | |
|
79 | return_value=hooks.HookResponse(1, 'foo')): | |
|
79 | with mock.patch('vcsserver.hooks.git_pre_pull', return_value=hooks.HookResponse(1, 'foo')): | |
|
80 | 80 | response = pygrack_app.post( |
|
81 | 81 | '/git-upload-pack', params=request, |
|
82 | 82 | content_type='application/x-git-upload-pack') |
@@ -86,8 +86,8 b' def test_pre_pull_hook_fails_with_sideba' | |||
|
86 | 86 | packets = list(proto.read_pkt_seq()) |
|
87 | 87 | |
|
88 | 88 | expected_packets = [ |
|
89 | 'NAK\n', '\x02foo', '\x02Pre pull hook failed: aborting\n', | |
|
90 | '\x01' + pygrack.GitRepository.EMPTY_PACK, | |
|
89 | b'NAK\n', b'\x02foo', b'\x02Pre pull hook failed: aborting\n', | |
|
90 | b'\x01' + pygrack.GitRepository.EMPTY_PACK, | |
|
91 | 91 | ] |
|
92 | 92 | assert packets == expected_packets |
|
93 | 93 | |
@@ -115,12 +115,16 b' def test_pull_has_hook_messages(pygrack_' | |||
|
115 | 115 | '0000', |
|
116 | 116 | '0009done\n', |
|
117 | 117 | ]) |
|
118 | ||
|
119 | pre_pull = 'pre_pull_output' | |
|
120 | post_pull = 'post_pull_output' | |
|
121 | ||
|
118 | 122 | with mock.patch('vcsserver.hooks.git_pre_pull', |
|
119 |
return_value=hooks.HookResponse(0, |
|
|
123 | return_value=hooks.HookResponse(0, pre_pull)): | |
|
120 | 124 | with mock.patch('vcsserver.hooks.git_post_pull', |
|
121 |
return_value=hooks.HookResponse(1, |
|
|
125 | return_value=hooks.HookResponse(1, post_pull)): | |
|
122 | 126 | with mock.patch('vcsserver.subprocessio.SubprocessIOChunker', |
|
123 | return_value=['0008NAK\n0009subp\n0000']): | |
|
127 | return_value=more_itertools.always_iterable([b'0008NAK\n0009subp\n0000'])): | |
|
124 | 128 | response = pygrack_app.post( |
|
125 | 129 | '/git-upload-pack', params=request, |
|
126 | 130 | content_type='application/x-git-upload-pack') |
@@ -129,13 +133,17 b' def test_pull_has_hook_messages(pygrack_' | |||
|
129 | 133 | proto = dulwich.protocol.Protocol(data.read, None) |
|
130 | 134 | packets = list(proto.read_pkt_seq()) |
|
131 | 135 | |
|
132 |
assert packets == ['NAK\n', |
|
|
136 | assert packets == [b'NAK\n', | |
|
137 | # pre-pull only outputs if IT FAILS as in != 0 ret code | |
|
138 | #b'\x02pre_pull_output', | |
|
139 | b'subp\n', | |
|
140 | b'\x02post_pull_output'] | |
|
133 | 141 | |
|
134 | 142 | |
|
135 | 143 | def test_get_want_capabilities(pygrack_instance): |
|
136 | 144 | data = io.BytesIO( |
|
137 | '0054want 74730d410fcb6603ace96f1dc55ea6196122532d ' + | |
|
138 | 'multi_ack side-band-64k ofs-delta\n00000009done\n') | |
|
145 | b'0054want 74730d410fcb6603ace96f1dc55ea6196122532d ' + | |
|
146 | b'multi_ack side-band-64k ofs-delta\n00000009done\n') | |
|
139 | 147 | |
|
140 | 148 | request = webob.Request({ |
|
141 | 149 | 'wsgi.input': data, |
@@ -146,20 +154,20 b' def test_get_want_capabilities(pygrack_i' | |||
|
146 | 154 | capabilities = pygrack_instance._get_want_capabilities(request) |
|
147 | 155 | |
|
148 | 156 | assert capabilities == frozenset( |
|
149 | ('ofs-delta', 'multi_ack', 'side-band-64k')) | |
|
157 | (b'ofs-delta', b'multi_ack', b'side-band-64k')) | |
|
150 | 158 | assert data.tell() == 0 |
|
151 | 159 | |
|
152 | 160 | |
|
153 | 161 | @pytest.mark.parametrize('data,capabilities,expected', [ |
|
154 | 162 | ('foo', [], []), |
|
155 | ('', ['side-band-64k'], []), | |
|
156 | ('', ['side-band'], []), | |
|
157 |
('foo', [ |
|
|
158 |
('foo', [ |
|
|
159 |
('f'*1000, [ |
|
|
160 |
('f'*1000, [ |
|
|
161 |
('f'*65520, [ |
|
|
162 |
('f'*65520, [ |
|
|
163 | ('', [pygrack.CAPABILITY_SIDE_BAND_64K], []), | |
|
164 | ('', [pygrack.CAPABILITY_SIDE_BAND], []), | |
|
165 | ('foo', [pygrack.CAPABILITY_SIDE_BAND_64K], [b'0008\x02foo']), | |
|
166 | ('foo', [pygrack.CAPABILITY_SIDE_BAND], [b'0008\x02foo']), | |
|
167 | ('f'*1000, [pygrack.CAPABILITY_SIDE_BAND_64K], [b'03ed\x02' + b'f' * 1000]), | |
|
168 | ('f'*1000, [pygrack.CAPABILITY_SIDE_BAND], [b'03e8\x02' + b'f' * 995, b'000a\x02fffff']), | |
|
169 | ('f'*65520, [pygrack.CAPABILITY_SIDE_BAND_64K], [b'fff0\x02' + b'f' * 65515, b'000a\x02fffff']), | |
|
170 | ('f'*65520, [pygrack.CAPABILITY_SIDE_BAND], [b'03e8\x02' + b'f' * 995] * 65 + [b'0352\x02' + b'f' * 845]), | |
|
163 | 171 | ], ids=[ |
|
164 | 172 | 'foo-empty', |
|
165 | 173 | 'empty-64k', 'empty', |
@@ -174,54 +182,59 b' def test_get_messages(pygrack_instance, ' | |||
|
174 | 182 | |
|
175 | 183 | @pytest.mark.parametrize('response,capabilities,pre_pull_messages,post_pull_messages', [ |
|
176 | 184 | # Unexpected response |
|
177 |
('unexpected_response', [ |
|
|
185 | ([b'unexpected_response[no_initial_header]'], [pygrack.CAPABILITY_SIDE_BAND_64K], 'foo', 'bar'), | |
|
178 | 186 | # No sideband |
|
179 | ('no-sideband', [], 'foo', 'bar'), | |
|
187 | ([b'no-sideband'], [], 'foo', 'bar'), | |
|
180 | 188 | # No messages |
|
181 |
('no-messages', [ |
|
|
189 | ([b'no-messages'], [pygrack.CAPABILITY_SIDE_BAND_64K], '', ''), | |
|
182 | 190 | ]) |
|
183 | 191 | def test_inject_messages_to_response_nothing_to_do( |
|
184 | pygrack_instance, response, capabilities, pre_pull_messages, | |
|
185 | post_pull_messages): | |
|
186 | new_response = pygrack_instance._inject_messages_to_response( | |
|
187 | response, capabilities, pre_pull_messages, post_pull_messages) | |
|
192 | pygrack_instance, response, capabilities, pre_pull_messages, post_pull_messages): | |
|
188 | 193 | |
|
189 | assert new_response == response | |
|
194 | new_response = pygrack_instance._build_post_pull_response( | |
|
195 | more_itertools.always_iterable(response), capabilities, pre_pull_messages, post_pull_messages) | |
|
196 | ||
|
197 | assert list(new_response) == response | |
|
190 | 198 | |
|
191 | 199 | |
|
192 | 200 | @pytest.mark.parametrize('capabilities', [ |
|
193 | ['side-band'], | |
|
194 | ['side-band-64k'], | |
|
201 | [pygrack.CAPABILITY_SIDE_BAND], | |
|
202 | [pygrack.CAPABILITY_SIDE_BAND_64K], | |
|
195 | 203 | ]) |
|
196 | def test_inject_messages_to_response_single_element(pygrack_instance, | |
|
197 | capabilities): | |
|
198 | response = ['0008NAK\n0009subp\n0000'] | |
|
199 | new_response = pygrack_instance._inject_messages_to_response( | |
|
200 | response, capabilities, 'foo', 'bar') | |
|
204 | def test_inject_messages_to_response_single_element(pygrack_instance, capabilities): | |
|
205 | response = [b'0008NAK\n0009subp\n0000'] | |
|
206 | new_response = pygrack_instance._build_post_pull_response( | |
|
207 | more_itertools.always_iterable(response), capabilities, 'foo', 'bar') | |
|
201 | 208 | |
|
202 | expected_response = [ | |
|
203 | '0008NAK\n', '0008\x02foo', '0009subp\n', '0008\x02bar', '0000'] | |
|
209 | expected_response = b''.join([ | |
|
210 | b'0008NAK\n', | |
|
211 | b'0008\x02foo', | |
|
212 | b'0009subp\n', | |
|
213 | b'0008\x02bar', | |
|
214 | b'0000']) | |
|
204 | 215 | |
|
205 | assert new_response == expected_response | |
|
216 | assert b''.join(new_response) == expected_response | |
|
206 | 217 | |
|
207 | 218 | |
|
208 | 219 | @pytest.mark.parametrize('capabilities', [ |
|
209 | ['side-band'], | |
|
210 | ['side-band-64k'], | |
|
220 | [pygrack.CAPABILITY_SIDE_BAND], | |
|
221 | [pygrack.CAPABILITY_SIDE_BAND_64K], | |
|
211 | 222 | ]) |
|
212 | def test_inject_messages_to_response_multi_element(pygrack_instance, | |
|
213 | capabilities): | |
|
214 | response = [ | |
|
215 | '0008NAK\n000asubp1\n', '000asubp2\n', '000asubp3\n', '000asubp4\n0000'] | |
|
216 |
new_response = pygrack_instance._ |
|
|
217 | response, capabilities, 'foo', 'bar') | |
|
223 | def test_inject_messages_to_response_multi_element(pygrack_instance, capabilities): | |
|
224 | response = more_itertools.always_iterable([ | |
|
225 | b'0008NAK\n000asubp1\n', b'000asubp2\n', b'000asubp3\n', b'000asubp4\n0000' | |
|
226 | ]) | |
|
227 | new_response = pygrack_instance._build_post_pull_response(response, capabilities, 'foo', 'bar') | |
|
218 | 228 | |
|
219 | expected_response = [ | |
|
220 | '0008NAK\n', '0008\x02foo', '000asubp1\n', '000asubp2\n', '000asubp3\n', | |
|
221 | '000asubp4\n', '0008\x02bar', '0000' | |
|
222 | ] | |
|
229 | expected_response = b''.join([ | |
|
230 | b'0008NAK\n', | |
|
231 | b'0008\x02foo', | |
|
232 | b'000asubp1\n', b'000asubp2\n', b'000asubp3\n', b'000asubp4\n', | |
|
233 | b'0008\x02bar', | |
|
234 | b'0000' | |
|
235 | ]) | |
|
223 | 236 | |
|
224 | assert new_response == expected_response | |
|
237 | assert b''.join(new_response) == expected_response | |
|
225 | 238 | |
|
226 | 239 | |
|
227 | 240 | def test_build_failed_pre_pull_response_no_sideband(pygrack_instance): |
@@ -231,19 +244,52 b' def test_build_failed_pre_pull_response_' | |||
|
231 | 244 | |
|
232 | 245 | |
|
233 | 246 | @pytest.mark.parametrize('capabilities', [ |
|
234 | ['side-band'], | |
|
235 | ['side-band-64k'], | |
|
236 | ['side-band-64k', 'no-progress'], | |
|
247 | [pygrack.CAPABILITY_SIDE_BAND], | |
|
248 | [pygrack.CAPABILITY_SIDE_BAND_64K], | |
|
249 | [pygrack.CAPABILITY_SIDE_BAND_64K, b'no-progress'], | |
|
237 | 250 | ]) |
|
238 | 251 | def test_build_failed_pre_pull_response(pygrack_instance, capabilities): |
|
239 | response = pygrack_instance._build_failed_pre_pull_response( | |
|
240 | capabilities, 'foo') | |
|
252 | response = pygrack_instance._build_failed_pre_pull_response(capabilities, 'foo') | |
|
241 | 253 | |
|
242 | 254 | expected_response = [ |
|
243 | '0008NAK\n', '0008\x02foo', '0024\x02Pre pull hook failed: aborting\n', | |
|
244 | '%04x\x01%s' % (len(pygrack.GitRepository.EMPTY_PACK) + 5, | |
|
245 |
|
|
|
246 | '0000', | |
|
255 | b'0008NAK\n', b'0008\x02foo', b'0024\x02Pre pull hook failed: aborting\n', | |
|
256 | b'%04x\x01%s' % (len(pygrack.GitRepository.EMPTY_PACK) + 5, pygrack.GitRepository.EMPTY_PACK), | |
|
257 | pygrack.GitRepository.FLUSH_PACKET, | |
|
247 | 258 | ] |
|
248 | 259 | |
|
249 | 260 | assert response == expected_response |
|
261 | ||
|
262 | ||
|
263 | def test_inject_messages_to_response_generator(pygrack_instance): | |
|
264 | ||
|
265 | def response_generator(): | |
|
266 | response = [ | |
|
267 | # protocol start | |
|
268 | b'0008NAK\n', | |
|
269 | ] | |
|
270 | response += [ascii_bytes(f'000asubp{x}\n') for x in range(1000)] | |
|
271 | response += [ | |
|
272 | # protocol end | |
|
273 | pygrack.GitRepository.FLUSH_PACKET | |
|
274 | ] | |
|
275 | for elem in response: | |
|
276 | yield elem | |
|
277 | ||
|
278 | new_response = pygrack_instance._build_post_pull_response( | |
|
279 | response_generator(), [pygrack.CAPABILITY_SIDE_BAND_64K, b'no-progress'], 'PRE_PULL_MSG\n', 'POST_PULL_MSG\n') | |
|
280 | ||
|
281 | assert iter(new_response) | |
|
282 | ||
|
283 | expected_response = b''.join([ | |
|
284 | # start | |
|
285 | b'0008NAK\n0012\x02PRE_PULL_MSG\n', | |
|
286 | ] + [ | |
|
287 | # ... rest | |
|
288 | ascii_bytes(f'000asubp{x}\n') for x in range(1000) | |
|
289 | ] + [ | |
|
290 | # final message, | |
|
291 | b'0013\x02POST_PULL_MSG\n0000', | |
|
292 | ||
|
293 | ]) | |
|
294 | ||
|
295 | assert b''.join(new_response) == expected_response |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -25,10 +25,11 b' import pytest' | |||
|
25 | 25 | import webtest |
|
26 | 26 | |
|
27 | 27 | from vcsserver import scm_app |
|
28 | from vcsserver.str_utils import ascii_bytes | |
|
28 | 29 | |
|
29 | 30 | |
|
30 | 31 | def test_hg_does_not_accept_invalid_cmd(tmpdir): |
|
31 | repo = mercurial.hg.repository(mercurial.ui.ui(), str(tmpdir), create=True) | |
|
32 | repo = mercurial.hg.repository(mercurial.ui.ui(), ascii_bytes(str(tmpdir)), create=True) | |
|
32 | 33 | app = webtest.TestApp(scm_app.HgWeb(repo)) |
|
33 | 34 | |
|
34 | 35 | response = app.get('/repo?cmd=invalidcmd', expect_errors=True) |
@@ -37,7 +38,7 b' def test_hg_does_not_accept_invalid_cmd(' | |||
|
37 | 38 | |
|
38 | 39 | |
|
39 | 40 | def test_create_hg_wsgi_app_requirement_error(tmpdir): |
|
40 | repo = mercurial.hg.repository(mercurial.ui.ui(), str(tmpdir), create=True) | |
|
41 | repo = mercurial.hg.repository(mercurial.ui.ui(), ascii_bytes(str(tmpdir)), create=True) | |
|
41 | 42 | config = ( |
|
42 | 43 | ('paths', 'default', ''), |
|
43 | 44 | ) |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -22,22 +22,23 b' import sys' | |||
|
22 | 22 | import pytest |
|
23 | 23 | |
|
24 | 24 | from vcsserver import subprocessio |
|
25 | from vcsserver.str_utils import ascii_bytes | |
|
25 | 26 | |
|
26 | 27 | |
|
27 |
class |
|
|
28 | class FileLikeObj(object): # pragma: no cover | |
|
28 | 29 | |
|
29 | def __init__(self, data, size): | |
|
30 | chunks = size / len(data) | |
|
30 | def __init__(self, data: bytes, size): | |
|
31 | chunks = size // len(data) | |
|
31 | 32 | |
|
32 | 33 | self.stream = self._get_stream(data, chunks) |
|
33 | 34 | |
|
34 | 35 | def _get_stream(self, data, chunks): |
|
35 |
for x in |
|
|
36 | for x in range(chunks): | |
|
36 | 37 | yield data |
|
37 | 38 | |
|
38 | 39 | def read(self, n): |
|
39 | 40 | |
|
40 | buffer_stream = '' | |
|
41 | buffer_stream = b'' | |
|
41 | 42 | for chunk in self.stream: |
|
42 | 43 | buffer_stream += chunk |
|
43 | 44 | if len(buffer_stream) >= n: |
@@ -51,7 +52,7 b' class KindaFilelike(object): # pragma: ' | |||
|
51 | 52 | def environ(): |
|
52 | 53 | """Delete coverage variables, as they make the tests fail.""" |
|
53 | 54 | env = dict(os.environ) |
|
54 | for key in env.keys(): | |
|
55 | for key in list(env.keys()): | |
|
55 | 56 | if key.startswith('COV_CORE_'): |
|
56 | 57 | del env[key] |
|
57 | 58 | |
@@ -63,93 +64,92 b' def _get_python_args(script):' | |||
|
63 | 64 | |
|
64 | 65 | |
|
65 | 66 | def test_raise_exception_on_non_zero_return_code(environ): |
|
66 |
args = _get_python_args(' |
|
|
67 |
with pytest.raises( |
|
|
68 |
|
|
|
67 | call_args = _get_python_args('raise ValueError("fail")') | |
|
68 | with pytest.raises(OSError): | |
|
69 | b''.join(subprocessio.SubprocessIOChunker(call_args, shell=False, env=environ)) | |
|
69 | 70 | |
|
70 | 71 | |
|
71 | 72 | def test_does_not_fail_on_non_zero_return_code(environ): |
|
72 | args = _get_python_args('sys.exit(1)') | |
|
73 | output = ''.join( | |
|
74 | subprocessio.SubprocessIOChunker( | |
|
75 | args, shell=False, fail_on_return_code=False, env=environ | |
|
76 | ) | |
|
77 | ) | |
|
73 | call_args = _get_python_args('sys.stdout.write("hello"); sys.exit(1)') | |
|
74 | proc = subprocessio.SubprocessIOChunker(call_args, shell=False, fail_on_return_code=False, env=environ) | |
|
75 | output = b''.join(proc) | |
|
78 | 76 | |
|
79 | assert output == '' | |
|
77 | assert output == b'hello' | |
|
80 | 78 | |
|
81 | 79 | |
|
82 | 80 | def test_raise_exception_on_stderr(environ): |
|
83 |
args = _get_python_args('sys.stderr.write(" |
|
|
84 | with pytest.raises(EnvironmentError) as excinfo: | |
|
85 | list(subprocessio.SubprocessIOChunker(args, shell=False, env=environ)) | |
|
81 | call_args = _get_python_args('sys.stderr.write("WRITE_TO_STDERR"); time.sleep(1);') | |
|
86 | 82 | |
|
87 | assert 'exited due to an error:\nX' in str(excinfo.value) | |
|
83 | with pytest.raises(OSError) as excinfo: | |
|
84 | b''.join(subprocessio.SubprocessIOChunker(call_args, shell=False, env=environ)) | |
|
85 | ||
|
86 | assert 'exited due to an error:\nWRITE_TO_STDERR' in str(excinfo.value) | |
|
88 | 87 | |
|
89 | 88 | |
|
90 | 89 | def test_does_not_fail_on_stderr(environ): |
|
91 |
args = _get_python_args('sys.stderr.write(" |
|
|
92 | output = ''.join( | |
|
93 | subprocessio.SubprocessIOChunker( | |
|
94 | args, shell=False, fail_on_stderr=False, env=environ | |
|
95 | ) | |
|
96 | ) | |
|
90 | call_args = _get_python_args('sys.stderr.write("WRITE_TO_STDERR"); sys.stderr.flush; time.sleep(2);') | |
|
91 | proc = subprocessio.SubprocessIOChunker(call_args, shell=False, fail_on_stderr=False, env=environ) | |
|
92 | output = b''.join(proc) | |
|
97 | 93 | |
|
98 | assert output == '' | |
|
94 | assert output == b'' | |
|
99 | 95 | |
|
100 | 96 | |
|
101 |
@pytest.mark.parametrize('size', [ |
|
|
97 | @pytest.mark.parametrize('size', [ | |
|
98 | 1, | |
|
99 | 10 ** 5 | |
|
100 | ]) | |
|
102 | 101 | def test_output_with_no_input(size, environ): |
|
103 | print(type(environ)) | |
|
104 | data = 'X' | |
|
105 | args = _get_python_args('sys.stdout.write("%s" * %d)' % (data, size)) | |
|
106 | output = ''.join(subprocessio.SubprocessIOChunker(args, shell=False, env=environ)) | |
|
102 | call_args = _get_python_args(f'sys.stdout.write("X" * {size});') | |
|
103 | proc = subprocessio.SubprocessIOChunker(call_args, shell=False, env=environ) | |
|
104 | output = b''.join(proc) | |
|
107 | 105 | |
|
108 |
assert output == |
|
|
106 | assert output == ascii_bytes("X" * size) | |
|
109 | 107 | |
|
110 | 108 | |
|
111 |
@pytest.mark.parametrize('size', [ |
|
|
109 | @pytest.mark.parametrize('size', [ | |
|
110 | 1, | |
|
111 | 10 ** 5 | |
|
112 | ]) | |
|
112 | 113 | def test_output_with_no_input_does_not_fail(size, environ): |
|
113 | data = 'X' | |
|
114 | args = _get_python_args('sys.stdout.write("%s" * %d); sys.exit(1)' % (data, size)) | |
|
115 | output = ''.join( | |
|
116 | subprocessio.SubprocessIOChunker( | |
|
117 | args, shell=False, fail_on_return_code=False, env=environ | |
|
118 | ) | |
|
119 | ) | |
|
120 | 114 | |
|
121 | print("{} {}".format(len(data * size), len(output))) | |
|
122 | assert output == data * size | |
|
115 | call_args = _get_python_args(f'sys.stdout.write("X" * {size}); sys.exit(1)') | |
|
116 | proc = subprocessio.SubprocessIOChunker(call_args, shell=False, fail_on_return_code=False, env=environ) | |
|
117 | output = b''.join(proc) | |
|
118 | ||
|
119 | assert output == ascii_bytes("X" * size) | |
|
123 | 120 | |
|
124 | 121 | |
|
125 |
@pytest.mark.parametrize('size', [ |
|
|
122 | @pytest.mark.parametrize('size', [ | |
|
123 | 1, | |
|
124 | 10 ** 5 | |
|
125 | ]) | |
|
126 | 126 | def test_output_with_input(size, environ): |
|
127 | 127 | data_len = size |
|
128 |
inputstream = |
|
|
128 | inputstream = FileLikeObj(b'X', size) | |
|
129 | 129 | |
|
130 | 130 | # This acts like the cat command. |
|
131 | args = _get_python_args('shutil.copyfileobj(sys.stdin, sys.stdout)') | |
|
132 | output = ''.join( | |
|
133 | subprocessio.SubprocessIOChunker( | |
|
134 |
|
|
|
135 | ) | |
|
131 | call_args = _get_python_args('shutil.copyfileobj(sys.stdin, sys.stdout)') | |
|
132 | # note: in this tests we explicitly don't assign chunker to a variable and let it stream directly | |
|
133 | output = b''.join( | |
|
134 | subprocessio.SubprocessIOChunker(call_args, shell=False, input_stream=inputstream, env=environ) | |
|
136 | 135 | ) |
|
137 | 136 | |
|
138 | 137 | assert len(output) == data_len |
|
139 | 138 | |
|
140 | 139 | |
|
141 |
@pytest.mark.parametrize('size', [ |
|
|
140 | @pytest.mark.parametrize('size', [ | |
|
141 | 1, | |
|
142 | 10 ** 5 | |
|
143 | ]) | |
|
142 | 144 | def test_output_with_input_skipping_iterator(size, environ): |
|
143 | 145 | data_len = size |
|
144 |
inputstream = |
|
|
146 | inputstream = FileLikeObj(b'X', size) | |
|
145 | 147 | |
|
146 | 148 | # This acts like the cat command. |
|
147 | args = _get_python_args('shutil.copyfileobj(sys.stdin, sys.stdout)') | |
|
149 | call_args = _get_python_args('shutil.copyfileobj(sys.stdin, sys.stdout)') | |
|
148 | 150 | |
|
149 | 151 | # Note: assigning the chunker makes sure that it is not deleted too early |
|
150 | chunker = subprocessio.SubprocessIOChunker( | |
|
151 | args, shell=False, inputstream=inputstream, env=environ | |
|
152 | ) | |
|
153 | output = ''.join(chunker.output) | |
|
152 | proc = subprocessio.SubprocessIOChunker(call_args, shell=False, input_stream=inputstream, env=environ) | |
|
153 | output = b''.join(proc.stdout) | |
|
154 | 154 | |
|
155 | 155 | assert len(output) == data_len |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -20,10 +20,12 b' import mock' | |||
|
20 | 20 | import pytest |
|
21 | 21 | import sys |
|
22 | 22 | |
|
23 | from vcsserver.str_utils import ascii_bytes | |
|
24 | ||
|
23 | 25 | |
|
24 | 26 | class MockPopen(object): |
|
25 | 27 | def __init__(self, stderr): |
|
26 | self.stdout = io.BytesIO('') | |
|
28 | self.stdout = io.BytesIO(b'') | |
|
27 | 29 | self.stderr = io.BytesIO(stderr) |
|
28 | 30 | self.returncode = 1 |
|
29 | 31 | |
@@ -44,7 +46,7 b" INVALID_CERTIFICATE_STDERR = '\\n'.join([" | |||
|
44 | 46 | @pytest.mark.xfail(sys.platform == "cygwin", |
|
45 | 47 | reason="SVN not packaged for Cygwin") |
|
46 | 48 | def test_import_remote_repository_certificate_error(stderr, expected_reason): |
|
47 | from vcsserver import svn | |
|
49 | from vcsserver.remote import svn | |
|
48 | 50 | factory = mock.Mock() |
|
49 | 51 | factory.repo = mock.Mock(return_value=mock.Mock()) |
|
50 | 52 | |
@@ -52,31 +54,29 b' def test_import_remote_repository_certif' | |||
|
52 | 54 | remote.is_path_valid_repository = lambda wire, path: True |
|
53 | 55 | |
|
54 | 56 | with mock.patch('subprocess.Popen', |
|
55 | return_value=MockPopen(stderr)): | |
|
57 | return_value=MockPopen(ascii_bytes(stderr))): | |
|
56 | 58 | with pytest.raises(Exception) as excinfo: |
|
57 | 59 | remote.import_remote_repository({'path': 'path'}, 'url') |
|
58 | 60 | |
|
59 | expected_error_args = ( | |
|
60 | 'Failed to dump the remote repository from url. Reason:{}'.format(expected_reason),) | |
|
61 | expected_error_args = 'Failed to dump the remote repository from url. Reason:{}'.format(expected_reason) | |
|
61 | 62 | |
|
62 | assert excinfo.value.args == expected_error_args | |
|
63 | assert excinfo.value.args[0] == expected_error_args | |
|
63 | 64 | |
|
64 | 65 | |
|
65 | 66 | def test_svn_libraries_can_be_imported(): |
|
66 | import svn | |
|
67 | 67 | import svn.client |
|
68 | 68 | assert svn.client is not None |
|
69 | 69 | |
|
70 | 70 | |
|
71 | 71 | @pytest.mark.parametrize('example_url, parts', [ |
|
72 |
('http://server.com', ( |
|
|
73 |
('http://user@server.com', ('user', |
|
|
72 | ('http://server.com', ('', '', 'http://server.com')), | |
|
73 | ('http://user@server.com', ('user', '', 'http://user@server.com')), | |
|
74 | 74 | ('http://user:pass@server.com', ('user', 'pass', 'http://user:pass@server.com')), |
|
75 |
('<script>', ( |
|
|
76 |
('http://', ( |
|
|
75 | ('<script>', ('', '', '<script>')), | |
|
76 | ('http://', ('', '', 'http://')), | |
|
77 | 77 | ]) |
|
78 | 78 | def test_username_password_extraction_from_url(example_url, parts): |
|
79 | from vcsserver import svn | |
|
79 | from vcsserver.remote import svn | |
|
80 | 80 | |
|
81 | 81 | factory = mock.Mock() |
|
82 | 82 | factory.repo = mock.Mock(return_value=mock.Mock()) |
@@ -85,3 +85,19 b' def test_username_password_extraction_fr' | |||
|
85 | 85 | remote.is_path_valid_repository = lambda wire, path: True |
|
86 | 86 | |
|
87 | 87 | assert remote.get_url_and_credentials(example_url) == parts |
|
88 | ||
|
89 | ||
|
90 | @pytest.mark.parametrize('call_url', [ | |
|
91 | b'https://svn.code.sf.net/p/svnbook/source/trunk/', | |
|
92 | b'https://marcink@svn.code.sf.net/p/svnbook/source/trunk/', | |
|
93 | b'https://marcink:qweqwe@svn.code.sf.net/p/svnbook/source/trunk/', | |
|
94 | ]) | |
|
95 | def test_check_url(call_url): | |
|
96 | from vcsserver.remote import svn | |
|
97 | factory = mock.Mock() | |
|
98 | factory.repo = mock.Mock(return_value=mock.Mock()) | |
|
99 | ||
|
100 | remote = svn.SvnRemote(factory) | |
|
101 | remote.is_path_valid_repository = lambda wire, path: True | |
|
102 | assert remote.check_url(call_url, {'dummy': 'config'}) | |
|
103 |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -19,25 +19,26 b' import wsgiref.simple_server' | |||
|
19 | 19 | import wsgiref.validate |
|
20 | 20 | |
|
21 | 21 | from vcsserver import wsgi_app_caller |
|
22 | ||
|
23 | ||
|
24 | # pylint: disable=protected-access,too-many-public-methods | |
|
22 | from vcsserver.str_utils import ascii_bytes, safe_str | |
|
25 | 23 | |
|
26 | 24 | |
|
27 | 25 | @wsgiref.validate.validator |
|
28 | 26 | def demo_app(environ, start_response): |
|
29 | 27 | """WSGI app used for testing.""" |
|
28 | ||
|
29 | input_data = safe_str(environ['wsgi.input'].read(1024)) | |
|
30 | ||
|
30 | 31 | data = [ |
|
31 | 32 | 'Hello World!\n', |
|
32 | 'input_data=%s\n' % environ['wsgi.input'].read(), | |
|
33 | f'input_data={input_data}\n', | |
|
33 | 34 | ] |
|
34 | 35 | for key, value in sorted(environ.items()): |
|
35 |
data.append(' |
|
|
36 | data.append(f'{key}={value}\n') | |
|
36 | 37 | |
|
37 | 38 | write = start_response("200 OK", [('Content-Type', 'text/plain')]) |
|
38 | write('Old school write method\n') | |
|
39 | write('***********************\n') | |
|
40 | return data | |
|
39 | write(b'Old school write method\n') | |
|
40 | write(b'***********************\n') | |
|
41 | return list(map(ascii_bytes, data)) | |
|
41 | 42 | |
|
42 | 43 | |
|
43 | 44 | BASE_ENVIRON = { |
@@ -53,11 +54,11 b' BASE_ENVIRON = {' | |||
|
53 | 54 | |
|
54 | 55 | def test_complete_environ(): |
|
55 | 56 | environ = dict(BASE_ENVIRON) |
|
56 | data = "data" | |
|
57 | data = b"data" | |
|
57 | 58 | wsgi_app_caller._complete_environ(environ, data) |
|
58 | 59 | wsgiref.validate.check_environ(environ) |
|
59 | 60 | |
|
60 | assert data == environ['wsgi.input'].read() | |
|
61 | assert data == environ['wsgi.input'].read(1024) | |
|
61 | 62 | |
|
62 | 63 | |
|
63 | 64 | def test_start_response(): |
@@ -81,16 +82,17 b' def test_start_response_with_error():' | |||
|
81 | 82 | |
|
82 | 83 | |
|
83 | 84 | def test_wsgi_app_caller(): |
|
84 | caller = wsgi_app_caller.WSGIAppCaller(demo_app) | |
|
85 | 85 | environ = dict(BASE_ENVIRON) |
|
86 | 86 | input_data = 'some text' |
|
87 | ||
|
88 | caller = wsgi_app_caller.WSGIAppCaller(demo_app) | |
|
87 | 89 | responses, status, headers = caller.handle(environ, input_data) |
|
88 | response = ''.join(responses) | |
|
90 | response = b''.join(responses) | |
|
89 | 91 | |
|
90 | 92 | assert status == '200 OK' |
|
91 | 93 | assert headers == [('Content-Type', 'text/plain')] |
|
92 | assert response.startswith( | |
|
93 | 'Old school write method\n***********************\n') | |
|
94 |
assert ' |
|
|
95 | assert 'foo.var=bla\n' in response | |
|
96 |
assert 'input_data= |
|
|
94 | assert response.startswith(b'Old school write method\n***********************\n') | |
|
95 | assert b'Hello World!\n' in response | |
|
96 | assert b'foo.var=bla\n' in response | |
|
97 | ||
|
98 | assert ascii_bytes(f'input_data={input_data}\n') in response |
@@ -1,6 +1,4 b'' | |||
|
1 | # -*- coding: utf-8 -*- | |
|
2 | ||
|
3 | # Copyright (C) 2016-2020 RhodeCode GmbH | |
|
1 | # Copyright (C) 2016-2023 RhodeCode GmbH | |
|
4 | 2 | # |
|
5 | 3 | # This program is free software: you can redistribute it and/or modify |
|
6 | 4 | # it under the terms of the GNU Affero General Public License, version 3 |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -14,13 +14,14 b'' | |||
|
14 | 14 | # You should have received a copy of the GNU General Public License |
|
15 | 15 | # along with this program; if not, write to the Free Software Foundation, |
|
16 | 16 | # Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
17 | ||
|
17 | import base64 | |
|
18 | 18 | import time |
|
19 | 19 | import logging |
|
20 | 20 | |
|
21 | import msgpack | |
|
22 | ||
|
21 | 23 | import vcsserver |
|
22 | from vcsserver.utils import safe_str | |
|
23 | ||
|
24 | from vcsserver.str_utils import safe_str, ascii_str | |
|
24 | 25 | |
|
25 | 26 | log = logging.getLogger(__name__) |
|
26 | 27 | |
@@ -34,12 +35,24 b' def get_user_agent(environ):' | |||
|
34 | 35 | return environ.get('HTTP_USER_AGENT') |
|
35 | 36 | |
|
36 | 37 | |
|
37 | def get_vcs_method(environ): | |
|
38 | return environ.get('HTTP_X_RC_METHOD') | |
|
38 | def get_call_context(request) -> dict: | |
|
39 | cc = {} | |
|
40 | registry = request.registry | |
|
41 | if hasattr(registry, 'vcs_call_context'): | |
|
42 | cc.update({ | |
|
43 | 'X-RC-Method': registry.vcs_call_context.get('method'), | |
|
44 | 'X-RC-Repo-Name': registry.vcs_call_context.get('repo_name') | |
|
45 | }) | |
|
46 | ||
|
47 | return cc | |
|
39 | 48 | |
|
40 | 49 | |
|
41 | def get_vcs_repo(environ): | |
|
42 | return environ.get('HTTP_X_RC_REPO_NAME') | |
|
50 | def get_headers_call_context(environ, strict=True): | |
|
51 | if 'HTTP_X_RC_VCS_STREAM_CALL_CONTEXT' in environ: | |
|
52 | packed_cc = base64.b64decode(environ['HTTP_X_RC_VCS_STREAM_CALL_CONTEXT']) | |
|
53 | return msgpack.unpackb(packed_cc) | |
|
54 | elif strict: | |
|
55 | raise ValueError('Expected header HTTP_X_RC_VCS_STREAM_CALL_CONTEXT not found') | |
|
43 | 56 | |
|
44 | 57 | |
|
45 | 58 | class RequestWrapperTween(object): |
@@ -54,23 +67,26 b' class RequestWrapperTween(object):' | |||
|
54 | 67 | log.debug('Starting request time measurement') |
|
55 | 68 | response = None |
|
56 | 69 | |
|
57 | ua = get_user_agent(request.environ) | |
|
58 | vcs_method = get_vcs_method(request.environ) | |
|
59 | repo_name = get_vcs_repo(request.environ) | |
|
60 | ||
|
61 | 70 | try: |
|
62 | 71 | response = self.handler(request) |
|
63 | 72 | finally: |
|
73 | ua = get_user_agent(request.environ) | |
|
74 | call_context = get_call_context(request) | |
|
75 | vcs_method = call_context.get('X-RC-Method', '_NO_VCS_METHOD') | |
|
76 | repo_name = call_context.get('X-RC-Repo-Name', '') | |
|
77 | ||
|
64 | 78 | count = request.request_count() |
|
65 | 79 | _ver_ = vcsserver.__version__ |
|
66 | 80 | _path = safe_str(get_access_path(request.environ)) |
|
81 | ||
|
67 | 82 | ip = '127.0.0.1' |
|
68 | 83 | match_route = request.matched_route.name if request.matched_route else "NOT_FOUND" |
|
69 | 84 | resp_code = getattr(response, 'status_code', 'UNDEFINED') |
|
70 | 85 | |
|
86 | _view_path = f"{repo_name}@{_path}/{vcs_method}" | |
|
87 | ||
|
71 | 88 | total = time.time() - start |
|
72 | 89 | |
|
73 | _view_path = "{}/{}@{}".format(_path, vcs_method, repo_name) | |
|
74 | 90 | log.info( |
|
75 | 91 | 'Req[%4s] IP: %s %s Request to %s time: %.4fs [%s], VCSServer %s', |
|
76 | 92 | count, ip, request.environ.get('REQUEST_METHOD'), |
@@ -87,15 +103,15 b' class RequestWrapperTween(object):' | |||
|
87 | 103 | statsd.timing( |
|
88 | 104 | "vcsserver_req_timing.histogram", elapsed_time_ms, |
|
89 | 105 | tags=[ |
|
90 |
"view_name:{}" |
|
|
91 |
"code:{}" |
|
|
106 | f"view_name:{match_route}", | |
|
107 | f"code:{resp_code}" | |
|
92 | 108 | ], |
|
93 | 109 | use_decimals=False |
|
94 | 110 | ) |
|
95 | 111 | statsd.incr( |
|
96 | 112 | "vcsserver_req_total", tags=[ |
|
97 |
"view_name:{}" |
|
|
98 |
"code:{}" |
|
|
113 | f"view_name:{match_route}", | |
|
114 | f"code:{resp_code}" | |
|
99 | 115 | ]) |
|
100 | 116 | |
|
101 | 117 | return response |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -20,91 +20,35 b' import hashlib' | |||
|
20 | 20 | log = logging.getLogger(__name__) |
|
21 | 21 | |
|
22 | 22 | |
|
23 | def safe_int(val, default=None): | |
|
24 | """ | |
|
25 | Returns int() of val if val is not convertable to int use default | |
|
26 | instead | |
|
27 | ||
|
28 | :param val: | |
|
29 | :param default: | |
|
30 | """ | |
|
31 | ||
|
32 | try: | |
|
33 | val = int(val) | |
|
34 | except (ValueError, TypeError): | |
|
35 | val = default | |
|
36 | ||
|
37 | return val | |
|
38 | ||
|
39 | ||
|
40 | def safe_str(unicode_, to_encoding=None): | |
|
41 | """ | |
|
42 | safe str function. Does few trick to turn unicode_ into string | |
|
43 | ||
|
44 | :param unicode_: unicode to encode | |
|
45 | :param to_encoding: encode to this type UTF8 default | |
|
46 | :rtype: str | |
|
47 | :returns: str object | |
|
48 | """ | |
|
49 | to_encoding = to_encoding or ['utf8'] | |
|
50 | # if it's not basestr cast to str | |
|
51 | if not isinstance(unicode_, basestring): | |
|
52 | return str(unicode_) | |
|
53 | ||
|
54 | if isinstance(unicode_, str): | |
|
55 | return unicode_ | |
|
56 | ||
|
57 | if not isinstance(to_encoding, (list, tuple)): | |
|
58 | to_encoding = [to_encoding] | |
|
23 | class AttributeDictBase(dict): | |
|
24 | def __getstate__(self): | |
|
25 | odict = self.__dict__ # get attribute dictionary | |
|
26 | return odict | |
|
59 | 27 | |
|
60 | for enc in to_encoding: | |
|
61 | try: | |
|
62 | return unicode_.encode(enc) | |
|
63 | except UnicodeEncodeError: | |
|
64 | pass | |
|
65 | ||
|
66 | return unicode_.encode(to_encoding[0], 'replace') | |
|
67 | ||
|
68 | ||
|
69 | def safe_unicode(str_, from_encoding=None): | |
|
70 | """ | |
|
71 | safe unicode function. Does few trick to turn str_ into unicode | |
|
72 | ||
|
73 | :param str_: string to decode | |
|
74 | :param from_encoding: encode from this type UTF8 default | |
|
75 | :rtype: unicode | |
|
76 | :returns: unicode object | |
|
77 | """ | |
|
78 | from_encoding = from_encoding or ['utf8'] | |
|
28 | def __setstate__(self, dict): | |
|
29 | self.__dict__ = dict | |
|
79 | 30 | |
|
80 | if isinstance(str_, unicode): | |
|
81 | return str_ | |
|
82 | ||
|
83 | if not isinstance(from_encoding, (list, tuple)): | |
|
84 | from_encoding = [from_encoding] | |
|
85 | ||
|
86 | try: | |
|
87 | return unicode(str_) | |
|
88 | except UnicodeDecodeError: | |
|
89 | pass | |
|
90 | ||
|
91 | for enc in from_encoding: | |
|
92 | try: | |
|
93 | return unicode(str_, enc) | |
|
94 | except UnicodeDecodeError: | |
|
95 | pass | |
|
96 | ||
|
97 | return unicode(str_, from_encoding[0], 'replace') | |
|
98 | ||
|
99 | ||
|
100 | class AttributeDict(dict): | |
|
101 | def __getattr__(self, attr): | |
|
102 | return self.get(attr, None) | |
|
103 | 31 | __setattr__ = dict.__setitem__ |
|
104 | 32 | __delattr__ = dict.__delitem__ |
|
105 | 33 | |
|
106 | 34 | |
|
35 | class StrictAttributeDict(AttributeDictBase): | |
|
36 | """ | |
|
37 | Strict Version of Attribute dict which raises an Attribute error when | |
|
38 | requested attribute is not set | |
|
39 | """ | |
|
40 | def __getattr__(self, attr): | |
|
41 | try: | |
|
42 | return self[attr] | |
|
43 | except KeyError: | |
|
44 | raise AttributeError('{} object has no attribute {}'.format( | |
|
45 | self.__class__, attr)) | |
|
46 | ||
|
47 | ||
|
48 | class AttributeDict(AttributeDictBase): | |
|
49 | def __getattr__(self, attr): | |
|
50 | return self.get(attr, None) | |
|
51 | ||
|
52 | ||
|
107 | 53 | def sha1(val): |
|
108 | 54 | return hashlib.sha1(val).hexdigest() |
|
109 | ||
|
110 |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -17,17 +17,18 b'' | |||
|
17 | 17 | |
|
18 | 18 | from vcsserver.lib import rc_cache |
|
19 | 19 | |
|
20 | ||
|
20 | 21 | class RemoteBase(object): |
|
21 | 22 | EMPTY_COMMIT = '0' * 40 |
|
22 | 23 | |
|
23 | 24 | def _region(self, wire): |
|
24 | 25 | cache_repo_id = wire.get('cache_repo_id', '') |
|
25 |
cache_namespace_uid = 'cache_repo.{}' |
|
|
26 | cache_namespace_uid = f'cache_repo.{cache_repo_id}' | |
|
26 | 27 | return rc_cache.get_or_create_region('repo_object', cache_namespace_uid) |
|
27 | 28 | |
|
28 | 29 | def _cache_on(self, wire): |
|
29 | 30 | context = wire.get('context', '') |
|
30 |
context_uid = '{}' |
|
|
31 | context_uid = f'{context}' | |
|
31 | 32 | repo_id = wire.get('repo_id', '') |
|
32 | 33 | cache = wire.get('cache', True) |
|
33 | 34 | cache_on = context and cache |
@@ -37,10 +38,10 b' class RemoteBase(object):' | |||
|
37 | 38 | from vcsserver.lib import rc_cache |
|
38 | 39 | repo_id = wire.get('repo_id', '') |
|
39 | 40 | cache_repo_id = wire.get('cache_repo_id', '') |
|
40 |
cache_namespace_uid = 'cache_repo.{}' |
|
|
41 | cache_namespace_uid = f'cache_repo.{cache_repo_id}' | |
|
41 | 42 | |
|
42 | 43 | if delete: |
|
43 | 44 | rc_cache.clear_cache_namespace( |
|
44 |
'repo_object', cache_namespace_uid, |
|
|
45 | 'repo_object', cache_namespace_uid, method=rc_cache.CLEAR_DELETE) | |
|
45 | 46 | |
|
46 | 47 | return {'invalidated': {'repo_id': repo_id, 'delete': delete}} |
@@ -1,5 +1,5 b'' | |||
|
1 | 1 | # RhodeCode VCSServer provides access to different vcs backends via network. |
|
2 |
# Copyright (C) 2014-202 |
|
|
2 | # Copyright (C) 2014-2023 RhodeCode GmbH | |
|
3 | 3 | # |
|
4 | 4 | # This program is free software; you can redistribute it and/or modify |
|
5 | 5 | # it under the terms of the GNU General Public License as published by |
@@ -23,19 +23,20 b' import io' | |||
|
23 | 23 | import logging |
|
24 | 24 | import os |
|
25 | 25 | |
|
26 | from vcsserver.str_utils import ascii_bytes | |
|
26 | 27 | |
|
27 | 28 | log = logging.getLogger(__name__) |
|
28 | 29 | |
|
29 | 30 | DEV_NULL = open(os.devnull) |
|
30 | 31 | |
|
31 | 32 | |
|
32 | def _complete_environ(environ, input_data): | |
|
33 | def _complete_environ(environ, input_data: bytes): | |
|
33 | 34 | """Update the missing wsgi.* variables of a WSGI environment. |
|
34 | 35 | |
|
35 | 36 | :param environ: WSGI environment to update |
|
36 | 37 | :type environ: dict |
|
37 | 38 | :param input_data: data to be read by the app |
|
38 |
:type input_data: |
|
|
39 | :type input_data: bytes | |
|
39 | 40 | """ |
|
40 | 41 | environ.update({ |
|
41 | 42 | 'wsgi.version': (1, 0), |
@@ -92,20 +93,19 b' class WSGIAppCaller(object):' | |||
|
92 | 93 | :param environ: WSGI environment to update |
|
93 | 94 | :type environ: dict |
|
94 | 95 | :param input_data: data to be read by the app |
|
95 | :type input_data: str | |
|
96 | :type input_data: str/bytes | |
|
96 | 97 | |
|
97 | 98 | :returns: a tuple with the contents, status and headers |
|
98 | 99 | :rtype: (list<str>, str, list<(str, str)>) |
|
99 | 100 | """ |
|
100 | _complete_environ(environ, input_data) | |
|
101 | _complete_environ(environ, ascii_bytes(input_data, allow_bytes=True)) | |
|
101 | 102 | start_response = _StartResponse() |
|
102 | 103 | log.debug("Calling wrapped WSGI application") |
|
103 | 104 | responses = self.app(environ, start_response) |
|
104 | 105 | responses_list = list(responses) |
|
105 | 106 | existing_responses = start_response.content |
|
106 | 107 | if existing_responses: |
|
107 | log.debug( | |
|
108 | "Adding returned response to response written via write()") | |
|
108 | log.debug("Adding returned response to response written via write()") | |
|
109 | 109 | existing_responses.extend(responses_list) |
|
110 | 110 | responses_list = existing_responses |
|
111 | 111 | if hasattr(responses, 'close'): |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed | |
This diff has been collapsed as it changes many lines, (1102 lines changed) Show them Hide them |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
|
1 | NO CONTENT: file was removed |
General Comments 0
You need to be logged in to leave comments.
Login now