test_client.py 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716
  1. # test_client.py -- Compatibility tests for git client.
  2. # Copyright (C) 2010 Google, Inc.
  3. #
  4. # Dulwich is dual-licensed under the Apache License, Version 2.0 and the GNU
  5. # General Public License as public by the Free Software Foundation; version 2.0
  6. # or (at your option) any later version. You can redistribute it and/or
  7. # modify it under the terms of either of these two licenses.
  8. #
  9. # Unless required by applicable law or agreed to in writing, software
  10. # distributed under the License is distributed on an "AS IS" BASIS,
  11. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12. # See the License for the specific language governing permissions and
  13. # limitations under the License.
  14. #
  15. # You should have received a copy of the licenses; if not, see
  16. # <http://www.gnu.org/licenses/> for a copy of the GNU General Public License
  17. # and <http://www.apache.org/licenses/LICENSE-2.0> for a copy of the Apache
  18. # License, Version 2.0.
  19. #
  20. """Compatibility tests between the Dulwich client and the cgit server."""
  21. import copy
  22. import http.server
  23. import os
  24. import select
  25. import signal
  26. import stat
  27. import subprocess
  28. import sys
  29. import tarfile
  30. import tempfile
  31. import threading
  32. from contextlib import suppress
  33. from io import BytesIO
  34. from unittest.mock import patch
  35. from urllib.parse import unquote
  36. from dulwich import client, file, index, objects, protocol, repo
  37. from .. import SkipTest, expectedFailure
  38. from .utils import (
  39. _DEFAULT_GIT,
  40. CompatTestCase,
  41. check_for_daemon,
  42. import_repo_to_dir,
  43. rmtree_ro,
  44. run_git_or_fail,
  45. )
  46. if sys.platform == "win32":
  47. import ctypes
  48. class DulwichClientTestBase:
  49. """Tests for client/server compatibility."""
  50. def setUp(self):
  51. self.gitroot = os.path.dirname(
  52. import_repo_to_dir("server_new.export").rstrip(os.sep)
  53. )
  54. self.dest = os.path.join(self.gitroot, "dest")
  55. file.ensure_dir_exists(self.dest)
  56. run_git_or_fail(["init", "--quiet", "--bare"], cwd=self.dest)
  57. def tearDown(self):
  58. rmtree_ro(self.gitroot)
  59. def assertDestEqualsSrc(self):
  60. repo_dir = os.path.join(self.gitroot, "server_new.export")
  61. dest_repo_dir = os.path.join(self.gitroot, "dest")
  62. with repo.Repo(repo_dir) as src:
  63. with repo.Repo(dest_repo_dir) as dest:
  64. self.assertReposEqual(src, dest)
  65. def _client(self):
  66. raise NotImplementedError
  67. def _build_path(self):
  68. raise NotImplementedError
  69. def _do_send_pack(self):
  70. c = self._client()
  71. srcpath = os.path.join(self.gitroot, "server_new.export")
  72. with repo.Repo(srcpath) as src:
  73. sendrefs = dict(src.get_refs())
  74. del sendrefs[b"HEAD"]
  75. c.send_pack(
  76. self._build_path("/dest"),
  77. lambda _: sendrefs,
  78. src.generate_pack_data,
  79. )
  80. def test_send_pack(self):
  81. self._do_send_pack()
  82. self.assertDestEqualsSrc()
  83. def test_send_pack_nothing_to_send(self):
  84. self._do_send_pack()
  85. self.assertDestEqualsSrc()
  86. # nothing to send, but shouldn't raise either.
  87. self._do_send_pack()
  88. @staticmethod
  89. def _add_file(repo, tree_id, filename, contents):
  90. tree = repo[tree_id]
  91. blob = objects.Blob()
  92. blob.data = contents.encode("utf-8")
  93. repo.object_store.add_object(blob)
  94. tree.add(filename.encode("utf-8"), stat.S_IFREG | 0o644, blob.id)
  95. repo.object_store.add_object(tree)
  96. return tree.id
  97. def test_send_pack_from_shallow_clone(self):
  98. c = self._client()
  99. server_new_path = os.path.join(self.gitroot, "server_new.export")
  100. run_git_or_fail(["config", "http.uploadpack", "true"], cwd=server_new_path)
  101. run_git_or_fail(["config", "http.receivepack", "true"], cwd=server_new_path)
  102. remote_path = self._build_path("/server_new.export")
  103. with repo.Repo(self.dest) as local:
  104. result = c.fetch(remote_path, local, depth=1)
  105. for r in result.refs.items():
  106. local.refs.set_if_equals(r[0], None, r[1])
  107. tree_id = local[local.head()].tree
  108. for filename, contents in [
  109. ("bar", "bar contents"),
  110. ("zop", "zop contents"),
  111. ]:
  112. tree_id = self._add_file(local, tree_id, filename, contents)
  113. commit_id = local.do_commit(
  114. message=b"add " + filename.encode("utf-8"),
  115. committer=b"Joe Example <joe@example.com>",
  116. tree=tree_id,
  117. )
  118. sendrefs = dict(local.get_refs())
  119. del sendrefs[b"HEAD"]
  120. c.send_pack(remote_path, lambda _: sendrefs, local.generate_pack_data)
  121. with repo.Repo(server_new_path) as remote:
  122. self.assertEqual(remote.head(), commit_id)
  123. def test_send_without_report_status(self):
  124. c = self._client()
  125. c._send_capabilities.remove(b"report-status")
  126. srcpath = os.path.join(self.gitroot, "server_new.export")
  127. with repo.Repo(srcpath) as src:
  128. sendrefs = dict(src.get_refs())
  129. del sendrefs[b"HEAD"]
  130. c.send_pack(
  131. self._build_path("/dest"),
  132. lambda _: sendrefs,
  133. src.generate_pack_data,
  134. )
  135. self.assertDestEqualsSrc()
  136. def make_dummy_commit(self, dest):
  137. b = objects.Blob.from_string(b"hi")
  138. dest.object_store.add_object(b)
  139. t = index.commit_tree(dest.object_store, [(b"hi", b.id, 0o100644)])
  140. c = objects.Commit()
  141. c.author = c.committer = b"Foo Bar <foo@example.com>"
  142. c.author_time = c.commit_time = 0
  143. c.author_timezone = c.commit_timezone = 0
  144. c.message = b"hi"
  145. c.tree = t
  146. dest.object_store.add_object(c)
  147. return c.id
  148. def disable_ff_and_make_dummy_commit(self):
  149. # disable non-fast-forward pushes to the server
  150. dest = repo.Repo(os.path.join(self.gitroot, "dest"))
  151. run_git_or_fail(
  152. ["config", "receive.denyNonFastForwards", "true"], cwd=dest.path
  153. )
  154. commit_id = self.make_dummy_commit(dest)
  155. return dest, commit_id
  156. def compute_send(self, src):
  157. sendrefs = dict(src.get_refs())
  158. del sendrefs[b"HEAD"]
  159. return sendrefs, src.generate_pack_data
  160. def test_send_pack_one_error(self):
  161. dest, dummy_commit = self.disable_ff_and_make_dummy_commit()
  162. dest.refs[b"refs/heads/master"] = dummy_commit
  163. repo_dir = os.path.join(self.gitroot, "server_new.export")
  164. with repo.Repo(repo_dir) as src:
  165. sendrefs, gen_pack = self.compute_send(src)
  166. c = self._client()
  167. result = c.send_pack(
  168. self._build_path("/dest"), lambda _: sendrefs, gen_pack
  169. )
  170. self.assertEqual(
  171. {
  172. b"refs/heads/branch": None,
  173. b"refs/heads/master": "non-fast-forward",
  174. },
  175. result.ref_status,
  176. )
  177. def test_send_pack_multiple_errors(self):
  178. dest, dummy = self.disable_ff_and_make_dummy_commit()
  179. # set up for two non-ff errors
  180. branch, master = b"refs/heads/branch", b"refs/heads/master"
  181. dest.refs[branch] = dest.refs[master] = dummy
  182. repo_dir = os.path.join(self.gitroot, "server_new.export")
  183. with repo.Repo(repo_dir) as src:
  184. sendrefs, gen_pack = self.compute_send(src)
  185. c = self._client()
  186. result = c.send_pack(
  187. self._build_path("/dest"), lambda _: sendrefs, gen_pack
  188. )
  189. self.assertEqual(
  190. {branch: "non-fast-forward", master: "non-fast-forward"},
  191. result.ref_status,
  192. )
  193. def test_archive(self):
  194. c = self._client()
  195. f = BytesIO()
  196. c.archive(self._build_path("/server_new.export"), b"HEAD", f.write)
  197. f.seek(0)
  198. tf = tarfile.open(fileobj=f)
  199. self.assertEqual(["baz", "foo"], tf.getnames())
  200. def test_fetch_pack(self):
  201. c = self._client()
  202. with repo.Repo(os.path.join(self.gitroot, "dest")) as dest:
  203. result = c.fetch(self._build_path("/server_new.export"), dest)
  204. self.assertEqual(
  205. {b"HEAD": b"refs/heads/master"},
  206. result.symrefs,
  207. )
  208. for r in result.refs.items():
  209. dest.refs.set_if_equals(r[0], None, r[1])
  210. self.assertDestEqualsSrc()
  211. def test_fetch_pack_with_nondefault_symref(self):
  212. c = self._client()
  213. src = repo.Repo(os.path.join(self.gitroot, "server_new.export"))
  214. src.refs.add_if_new(b"refs/heads/main", src.refs[b"refs/heads/master"])
  215. src.refs.set_symbolic_ref(b"HEAD", b"refs/heads/main")
  216. with repo.Repo(os.path.join(self.gitroot, "dest")) as dest:
  217. result = c.fetch(self._build_path("/server_new.export"), dest)
  218. self.assertEqual(
  219. {b"HEAD": b"refs/heads/main"},
  220. result.symrefs,
  221. )
  222. for r in result.refs.items():
  223. dest.refs.set_if_equals(r[0], None, r[1])
  224. self.assertDestEqualsSrc()
  225. def test_fetch_pack_depth(self):
  226. c = self._client()
  227. with repo.Repo(os.path.join(self.gitroot, "dest")) as dest:
  228. result = c.fetch(self._build_path("/server_new.export"), dest, depth=1)
  229. for r in result.refs.items():
  230. dest.refs.set_if_equals(r[0], None, r[1])
  231. self.assertEqual(
  232. dest.get_shallow(),
  233. {
  234. b"35e0b59e187dd72a0af294aedffc213eaa4d03ff",
  235. b"514dc6d3fbfe77361bcaef320c4d21b72bc10be9",
  236. },
  237. )
  238. def test_repeat(self):
  239. c = self._client()
  240. with repo.Repo(os.path.join(self.gitroot, "dest")) as dest:
  241. result = c.fetch(self._build_path("/server_new.export"), dest)
  242. for r in result.refs.items():
  243. dest.refs.set_if_equals(r[0], None, r[1])
  244. self.assertDestEqualsSrc()
  245. result = c.fetch(self._build_path("/server_new.export"), dest)
  246. for r in result.refs.items():
  247. dest.refs.set_if_equals(r[0], None, r[1])
  248. self.assertDestEqualsSrc()
  249. def test_fetch_empty_pack(self):
  250. c = self._client()
  251. with repo.Repo(os.path.join(self.gitroot, "dest")) as dest:
  252. result = c.fetch(self._build_path("/server_new.export"), dest)
  253. for r in result.refs.items():
  254. dest.refs.set_if_equals(r[0], None, r[1])
  255. self.assertDestEqualsSrc()
  256. def dw(refs, **kwargs):
  257. return list(refs.values())
  258. result = c.fetch(
  259. self._build_path("/server_new.export"),
  260. dest,
  261. determine_wants=dw,
  262. )
  263. for r in result.refs.items():
  264. dest.refs.set_if_equals(r[0], None, r[1])
  265. self.assertDestEqualsSrc()
  266. def test_incremental_fetch_pack(self):
  267. self.test_fetch_pack()
  268. dest, dummy = self.disable_ff_and_make_dummy_commit()
  269. dest.refs[b"refs/heads/master"] = dummy
  270. c = self._client()
  271. repo_dir = os.path.join(self.gitroot, "server_new.export")
  272. with repo.Repo(repo_dir) as dest:
  273. result = c.fetch(self._build_path("/dest"), dest)
  274. for r in result.refs.items():
  275. dest.refs.set_if_equals(r[0], None, r[1])
  276. self.assertDestEqualsSrc()
  277. def test_fetch_pack_no_side_band_64k(self):
  278. if protocol.DEFAULT_GIT_PROTOCOL_VERSION_FETCH >= 2:
  279. raise SkipTest("side-band-64k cannot be disabled with git protocol v2")
  280. c = self._client()
  281. c._fetch_capabilities.remove(b"side-band-64k")
  282. with repo.Repo(os.path.join(self.gitroot, "dest")) as dest:
  283. result = c.fetch(self._build_path("/server_new.export"), dest)
  284. for r in result.refs.items():
  285. dest.refs.set_if_equals(r[0], None, r[1])
  286. self.assertDestEqualsSrc()
  287. def test_fetch_pack_zero_sha(self):
  288. # zero sha1s are already present on the client, and should
  289. # be ignored
  290. c = self._client()
  291. with repo.Repo(os.path.join(self.gitroot, "dest")) as dest:
  292. result = c.fetch(
  293. self._build_path("/server_new.export"),
  294. dest,
  295. lambda refs, **kwargs: [protocol.ZERO_SHA],
  296. )
  297. for r in result.refs.items():
  298. dest.refs.set_if_equals(r[0], None, r[1])
  299. def test_send_remove_branch(self):
  300. with repo.Repo(os.path.join(self.gitroot, "dest")) as dest:
  301. dummy_commit = self.make_dummy_commit(dest)
  302. dest.refs[b"refs/heads/master"] = dummy_commit
  303. dest.refs[b"refs/heads/abranch"] = dummy_commit
  304. sendrefs = dict(dest.refs)
  305. sendrefs[b"refs/heads/abranch"] = b"00" * 20
  306. del sendrefs[b"HEAD"]
  307. def gen_pack(have, want, ofs_delta=False, progress=None):
  308. return 0, []
  309. c = self._client()
  310. self.assertEqual(dest.refs[b"refs/heads/abranch"], dummy_commit)
  311. c.send_pack(self._build_path("/dest"), lambda _: sendrefs, gen_pack)
  312. self.assertNotIn(b"refs/heads/abranch", dest.refs)
  313. def test_send_new_branch_empty_pack(self):
  314. with repo.Repo(os.path.join(self.gitroot, "dest")) as dest:
  315. dummy_commit = self.make_dummy_commit(dest)
  316. dest.refs[b"refs/heads/master"] = dummy_commit
  317. dest.refs[b"refs/heads/abranch"] = dummy_commit
  318. sendrefs = {b"refs/heads/bbranch": dummy_commit}
  319. def gen_pack(have, want, ofs_delta=False, progress=None):
  320. return 0, []
  321. c = self._client()
  322. self.assertEqual(dest.refs[b"refs/heads/abranch"], dummy_commit)
  323. c.send_pack(self._build_path("/dest"), lambda _: sendrefs, gen_pack)
  324. self.assertEqual(dummy_commit, dest.refs[b"refs/heads/abranch"])
  325. def test_get_refs(self):
  326. c = self._client()
  327. refs = c.get_refs(self._build_path("/server_new.export"))
  328. repo_dir = os.path.join(self.gitroot, "server_new.export")
  329. with repo.Repo(repo_dir) as dest:
  330. self.assertDictEqual(dest.refs.as_dict(), refs)
  331. class DulwichTCPClientTest(CompatTestCase, DulwichClientTestBase):
  332. def setUp(self):
  333. CompatTestCase.setUp(self)
  334. DulwichClientTestBase.setUp(self)
  335. if check_for_daemon(limit=1):
  336. raise SkipTest(
  337. f"git-daemon was already running on port {protocol.TCP_GIT_PORT}"
  338. )
  339. fd, self.pidfile = tempfile.mkstemp(
  340. prefix="dulwich-test-git-client", suffix=".pid"
  341. )
  342. os.fdopen(fd).close()
  343. args = [
  344. _DEFAULT_GIT,
  345. "daemon",
  346. "--verbose",
  347. "--export-all",
  348. f"--pid-file={self.pidfile}",
  349. f"--base-path={self.gitroot}",
  350. "--enable=receive-pack",
  351. "--enable=upload-archive",
  352. "--listen=localhost",
  353. "--reuseaddr",
  354. self.gitroot,
  355. ]
  356. self.process = subprocess.Popen(
  357. args,
  358. cwd=self.gitroot,
  359. stdout=subprocess.PIPE,
  360. stderr=subprocess.PIPE,
  361. )
  362. if not check_for_daemon():
  363. raise SkipTest("git-daemon failed to start")
  364. def tearDown(self):
  365. with open(self.pidfile) as f:
  366. pid = int(f.read().strip())
  367. if sys.platform == "win32":
  368. PROCESS_TERMINATE = 1
  369. handle = ctypes.windll.kernel32.OpenProcess(PROCESS_TERMINATE, False, pid)
  370. ctypes.windll.kernel32.TerminateProcess(handle, -1)
  371. ctypes.windll.kernel32.CloseHandle(handle)
  372. else:
  373. with suppress(OSError):
  374. os.kill(pid, signal.SIGKILL)
  375. os.unlink(self.pidfile)
  376. self.process.wait()
  377. self.process.stdout.close()
  378. self.process.stderr.close()
  379. DulwichClientTestBase.tearDown(self)
  380. CompatTestCase.tearDown(self)
  381. def _client(self):
  382. return client.TCPGitClient("localhost")
  383. def _build_path(self, path):
  384. return path
  385. if sys.platform == "win32" and protocol.DEFAULT_GIT_PROTOCOL_VERSION_FETCH < 2:
  386. @expectedFailure
  387. def test_fetch_pack_no_side_band_64k(self):
  388. DulwichClientTestBase.test_fetch_pack_no_side_band_64k(self)
  389. def test_send_remove_branch(self):
  390. # This test fails intermittently on my machine, probably due to some sort
  391. # of race condition. Probably also related to #1015
  392. self.skipTest("skip flaky test; see #1015")
  393. @patch("dulwich.protocol.DEFAULT_GIT_PROTOCOL_VERSION_FETCH", new=0)
  394. class DulwichTCPClientTestGitProtov0(DulwichTCPClientTest):
  395. pass
  396. class TestSSHVendor:
  397. @staticmethod
  398. def run_command(
  399. host,
  400. command,
  401. username=None,
  402. port=None,
  403. password=None,
  404. key_filename=None,
  405. protocol_version=None,
  406. ):
  407. cmd, path = command.split(" ")
  408. cmd = cmd.split("-", 1)
  409. path = path.replace("'", "")
  410. env = dict(os.environ)
  411. if protocol_version is None:
  412. protocol_version = protocol.DEFAULT_GIT_PROTOCOL_VERSION_FETCH
  413. if protocol_version > 0:
  414. env["GIT_PROTOCOL"] = f"version={protocol_version}"
  415. p = subprocess.Popen(
  416. [*cmd, path],
  417. bufsize=0,
  418. stdin=subprocess.PIPE,
  419. stdout=subprocess.PIPE,
  420. stderr=subprocess.PIPE,
  421. env=env,
  422. )
  423. return client.SubprocessWrapper(p)
  424. class DulwichMockSSHClientTest(CompatTestCase, DulwichClientTestBase):
  425. def setUp(self):
  426. CompatTestCase.setUp(self)
  427. DulwichClientTestBase.setUp(self)
  428. self.real_vendor = client.get_ssh_vendor
  429. client.get_ssh_vendor = TestSSHVendor
  430. def tearDown(self):
  431. DulwichClientTestBase.tearDown(self)
  432. CompatTestCase.tearDown(self)
  433. client.get_ssh_vendor = self.real_vendor
  434. def _client(self):
  435. return client.SSHGitClient("localhost")
  436. def _build_path(self, path):
  437. return self.gitroot + path
  438. @patch("dulwich.protocol.DEFAULT_GIT_PROTOCOL_VERSION_FETCH", new=0)
  439. class DulwichMockSSHClientTestGitProtov0(DulwichMockSSHClientTest):
  440. pass
  441. class DulwichSubprocessClientTest(CompatTestCase, DulwichClientTestBase):
  442. def setUp(self):
  443. CompatTestCase.setUp(self)
  444. DulwichClientTestBase.setUp(self)
  445. def tearDown(self):
  446. DulwichClientTestBase.tearDown(self)
  447. CompatTestCase.tearDown(self)
  448. def _client(self):
  449. return client.SubprocessGitClient()
  450. def _build_path(self, path):
  451. return self.gitroot + path
  452. @patch("dulwich.protocol.DEFAULT_GIT_PROTOCOL_VERSION_FETCH", new=0)
  453. class DulwichSubprocessClientTestGitProtov0(DulwichSubprocessClientTest):
  454. pass
  455. class GitHTTPRequestHandler(http.server.SimpleHTTPRequestHandler):
  456. """HTTP Request handler that calls out to 'git http-backend'."""
  457. # Make rfile unbuffered -- we need to read one line and then pass
  458. # the rest to a subprocess, so we can't use buffered input.
  459. rbufsize = 0
  460. def do_POST(self):
  461. self.run_backend()
  462. def do_GET(self):
  463. self.run_backend()
  464. def send_head(self):
  465. return self.run_backend()
  466. def log_request(self, code="-", size="-"):
  467. # Let's be quiet, the test suite is noisy enough already
  468. pass
  469. def run_backend(self):
  470. """Call out to git http-backend."""
  471. # Based on CGIHTTPServer.CGIHTTPRequestHandler.run_cgi:
  472. # Copyright (c) 2001-2010 Python Software Foundation;
  473. # All Rights Reserved
  474. # Licensed under the Python Software Foundation License.
  475. rest = self.path
  476. # find an explicit query string, if present.
  477. i = rest.rfind("?")
  478. if i >= 0:
  479. rest, query = rest[:i], rest[i + 1 :]
  480. else:
  481. query = ""
  482. env = copy.deepcopy(os.environ)
  483. env["SERVER_SOFTWARE"] = self.version_string()
  484. env["SERVER_NAME"] = self.server.server_name
  485. env["GATEWAY_INTERFACE"] = "CGI/1.1"
  486. env["SERVER_PROTOCOL"] = self.protocol_version
  487. env["SERVER_PORT"] = str(self.server.server_port)
  488. env["GIT_PROJECT_ROOT"] = self.server.root_path
  489. env["GIT_HTTP_EXPORT_ALL"] = "1"
  490. env["REQUEST_METHOD"] = self.command
  491. uqrest = unquote(rest)
  492. env["PATH_INFO"] = uqrest
  493. env["SCRIPT_NAME"] = "/"
  494. if query:
  495. env["QUERY_STRING"] = query
  496. host = self.address_string()
  497. if host != self.client_address[0]:
  498. env["REMOTE_HOST"] = host
  499. env["REMOTE_ADDR"] = self.client_address[0]
  500. authorization = self.headers.get("authorization")
  501. if authorization:
  502. authorization = authorization.split()
  503. if len(authorization) == 2:
  504. import base64
  505. import binascii
  506. env["AUTH_TYPE"] = authorization[0]
  507. if authorization[0].lower() == "basic":
  508. try:
  509. authorization = base64.decodestring(authorization[1])
  510. except binascii.Error:
  511. pass
  512. else:
  513. authorization = authorization.split(":")
  514. if len(authorization) == 2:
  515. env["REMOTE_USER"] = authorization[0]
  516. # XXX REMOTE_IDENT
  517. content_type = self.headers.get("content-type")
  518. if content_type:
  519. env["CONTENT_TYPE"] = content_type
  520. length = self.headers.get("content-length")
  521. if length:
  522. env["CONTENT_LENGTH"] = length
  523. referer = self.headers.get("referer")
  524. if referer:
  525. env["HTTP_REFERER"] = referer
  526. accept = []
  527. for line in self.headers.getallmatchingheaders("accept"):
  528. if line[:1] in "\t\n\r ":
  529. accept.append(line.strip())
  530. else:
  531. accept = accept + line[7:].split(",")
  532. env["HTTP_ACCEPT"] = ",".join(accept)
  533. ua = self.headers.get("user-agent")
  534. if ua:
  535. env["HTTP_USER_AGENT"] = ua
  536. co = self.headers.get("cookie")
  537. if co:
  538. env["HTTP_COOKIE"] = co
  539. proto = self.headers.get("Git-Protocol")
  540. if proto:
  541. env["GIT_PROTOCOL"] = proto
  542. # XXX Other HTTP_* headers
  543. # Since we're setting the env in the parent, provide empty
  544. # values to override previously set values
  545. for k in (
  546. "QUERY_STRING",
  547. "REMOTE_HOST",
  548. "CONTENT_LENGTH",
  549. "HTTP_USER_AGENT",
  550. "HTTP_COOKIE",
  551. "HTTP_REFERER",
  552. ):
  553. env.setdefault(k, "")
  554. self.wfile.write(b"HTTP/1.1 200 Script output follows\r\n")
  555. self.wfile.write((f"Server: {self.server.server_name}\r\n").encode("ascii"))
  556. self.wfile.write((f"Date: {self.date_time_string()}\r\n").encode("ascii"))
  557. decoded_query = query.replace("+", " ")
  558. try:
  559. nbytes = int(length)
  560. except (TypeError, ValueError):
  561. nbytes = -1
  562. if self.command.lower() == "post":
  563. if nbytes > 0:
  564. data = self.rfile.read(nbytes)
  565. elif self.headers.get("transfer-encoding") == "chunked":
  566. chunks = []
  567. while True:
  568. line = self.rfile.readline()
  569. length = int(line.rstrip(), 16)
  570. chunk = self.rfile.read(length + 2)
  571. chunks.append(chunk[:-2])
  572. if length == 0:
  573. break
  574. data = b"".join(chunks)
  575. env["CONTENT_LENGTH"] = str(len(data))
  576. else:
  577. raise AssertionError
  578. else:
  579. data = None
  580. env["CONTENT_LENGTH"] = "0"
  581. # throw away additional data [see bug #427345]
  582. while select.select([self.rfile._sock], [], [], 0)[0]:
  583. if not self.rfile._sock.recv(1):
  584. break
  585. args = ["http-backend"]
  586. if "=" not in decoded_query:
  587. args.append(decoded_query)
  588. stdout = run_git_or_fail(args, input=data, env=env, stderr=subprocess.PIPE)
  589. self.wfile.write(stdout)
  590. class HTTPGitServer(http.server.HTTPServer):
  591. allow_reuse_address = True
  592. def __init__(self, server_address, root_path) -> None:
  593. http.server.HTTPServer.__init__(self, server_address, GitHTTPRequestHandler)
  594. self.root_path = root_path
  595. self.server_name = "localhost"
  596. def get_url(self):
  597. return f"http://{self.server_name}:{self.server_port}/"
  598. class DulwichHttpClientTest(CompatTestCase, DulwichClientTestBase):
  599. min_git_version = (1, 7, 0, 2)
  600. def setUp(self):
  601. CompatTestCase.setUp(self)
  602. DulwichClientTestBase.setUp(self)
  603. self._httpd = HTTPGitServer(("localhost", 0), self.gitroot)
  604. self.addCleanup(self._httpd.shutdown)
  605. threading.Thread(target=self._httpd.serve_forever).start()
  606. run_git_or_fail(["config", "http.uploadpack", "true"], cwd=self.dest)
  607. run_git_or_fail(["config", "http.receivepack", "true"], cwd=self.dest)
  608. def tearDown(self):
  609. DulwichClientTestBase.tearDown(self)
  610. CompatTestCase.tearDown(self)
  611. self._httpd.shutdown()
  612. self._httpd.socket.close()
  613. def _client(self):
  614. return client.HttpGitClient(self._httpd.get_url())
  615. def _build_path(self, path):
  616. return path
  617. def test_archive(self):
  618. raise SkipTest("exporting archives not supported over http")
  619. @patch("dulwich.protocol.DEFAULT_GIT_PROTOCOL_VERSION_FETCH", new=0)
  620. class DulwichHttpClientTestGitProtov0(DulwichHttpClientTest):
  621. pass