123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272 |
- /*
- * Copyright (C) 2015, Google Inc. and others
- *
- * This program and the accompanying materials are made available under the
- * terms of the Eclipse Distribution License v. 1.0 which is available at
- * https://www.eclipse.org/org/documents/edl-v10.php.
- *
- * SPDX-License-Identifier: BSD-3-Clause
- */
-
- package org.eclipse.jgit.transport;
-
- import static org.junit.Assert.assertEquals;
- import static org.junit.Assert.assertNull;
- import static org.junit.Assert.assertTrue;
- import static org.junit.Assert.fail;
-
- import java.util.ArrayList;
- import java.util.List;
- import java.util.concurrent.atomic.AtomicInteger;
-
- import org.eclipse.jgit.api.Git;
- import org.eclipse.jgit.api.errors.InvalidRemoteException;
- import org.eclipse.jgit.api.errors.TransportException;
- import org.eclipse.jgit.internal.JGitText;
- import org.eclipse.jgit.internal.storage.dfs.DfsRepositoryDescription;
- import org.eclipse.jgit.internal.storage.dfs.InMemoryRepository;
- import org.eclipse.jgit.junit.TestRepository;
- import org.eclipse.jgit.lib.ObjectId;
- import org.eclipse.jgit.lib.Repository;
- import org.eclipse.jgit.revwalk.RevCommit;
- import org.eclipse.jgit.storage.pack.PackStatistics;
- import org.eclipse.jgit.transport.BasePackFetchConnection.FetchConfig;
- import org.eclipse.jgit.transport.resolver.ReceivePackFactory;
- import org.eclipse.jgit.transport.resolver.ServiceNotAuthorizedException;
- import org.eclipse.jgit.transport.resolver.UploadPackFactory;
- import org.junit.After;
- import org.junit.Before;
- import org.junit.Test;
-
- public class TestProtocolTest {
- private static final RefSpec HEADS = new RefSpec("+refs/heads/*:refs/heads/*");
-
- private static final RefSpec MASTER = new RefSpec(
- "+refs/heads/master:refs/heads/master");
-
- private static final int HAVES_PER_ROUND = 32;
- private static final int MAX_HAVES = 256;
-
- private static class User {
- private final String name;
-
- private User(String name) {
- this.name = name;
- }
- }
-
- private static class DefaultUpload implements UploadPackFactory<User> {
- @Override
- public UploadPack create(User req, Repository db) {
- UploadPack up = new UploadPack(db);
- up.setPostUploadHook((PackStatistics stats) -> {
- havesCount = stats.getHaves();
- });
- return up;
- }
- }
-
- private static class DefaultReceive implements ReceivePackFactory<User> {
- @Override
- public ReceivePack create(User req, Repository db) {
- return new ReceivePack(db);
- }
- }
-
- private static long havesCount;
-
- private List<TransportProtocol> protos;
- private TestRepository<InMemoryRepository> local;
- private TestRepository<InMemoryRepository> remote;
-
- @Before
- public void setUp() throws Exception {
- protos = new ArrayList<>();
- local = new TestRepository<>(
- new InMemoryRepository(new DfsRepositoryDescription("local")));
- remote = new TestRepository<>(
- new InMemoryRepository(new DfsRepositoryDescription("remote")));
- }
-
- @After
- public void tearDown() {
- for (TransportProtocol proto : protos) {
- Transport.unregister(proto);
- }
- }
-
- @Test
- public void testFetch() throws Exception {
- ObjectId master = remote.branch("master").commit().create();
-
- TestProtocol<User> proto = registerDefault();
- URIish uri = proto.register(new User("user"), remote.getRepository());
-
- try (Git git = new Git(local.getRepository())) {
- git.fetch()
- .setRemote(uri.toString())
- .setRefSpecs(HEADS)
- .call();
- assertEquals(master,
- local.getRepository().exactRef("refs/heads/master").getObjectId());
- }
- }
-
- @Test
- public void testPush() throws Exception {
- ObjectId master = local.branch("master").commit().create();
-
- TestProtocol<User> proto = registerDefault();
- URIish uri = proto.register(new User("user"), remote.getRepository());
-
- try (Git git = new Git(local.getRepository())) {
- git.push()
- .setRemote(uri.toString())
- .setRefSpecs(HEADS)
- .call();
- assertEquals(master,
- remote.getRepository().exactRef("refs/heads/master").getObjectId());
- }
- }
-
- @Test
- public void testFullNegotiation() throws Exception {
- TestProtocol<User> proto = registerDefault();
- URIish uri = proto.register(new User("user"), remote.getRepository());
-
- // Enough local branches to cause 10 rounds of negotiation,
- // and a unique remote master branch commit with a later timestamp.
- for (int i = 0; i < 10 * HAVES_PER_ROUND; i++) {
- local.branch("local-branch-" + i).commit().create();
- }
- remote.tick(11 * HAVES_PER_ROUND);
- RevCommit master = remote.branch("master").commit()
- .add("readme.txt", "unique commit").create();
-
- try (Git git = new Git(local.getRepository())) {
- assertNull(local.getRepository().exactRef("refs/heads/master"));
- git.fetch().setRemote(uri.toString()).setRefSpecs(MASTER).call();
- assertEquals(master, local.getRepository()
- .exactRef("refs/heads/master").getObjectId());
- assertEquals(10 * HAVES_PER_ROUND, havesCount);
- }
- }
-
- @Test
- public void testMaxHaves() throws Exception {
- TestProtocol<User> proto = registerDefault();
- URIish uri = proto.register(new User("user"), remote.getRepository());
-
- // Enough local branches to cause 10 rounds of negotiation,
- // and a unique remote master branch commit with a later timestamp.
- for (int i = 0; i < 10 * HAVES_PER_ROUND; i++) {
- local.branch("local-branch-" + i).commit().create();
- }
- remote.tick(11 * HAVES_PER_ROUND);
- RevCommit master = remote.branch("master").commit()
- .add("readme.txt", "unique commit").create();
-
- TestProtocol.setFetchConfig(new FetchConfig(true, MAX_HAVES));
- try (Git git = new Git(local.getRepository())) {
- assertNull(local.getRepository().exactRef("refs/heads/master"));
- git.fetch().setRemote(uri.toString()).setRefSpecs(MASTER).call();
- assertEquals(master, local.getRepository()
- .exactRef("refs/heads/master").getObjectId());
- assertTrue(havesCount <= MAX_HAVES);
- }
- }
-
- @Test
- public void testUploadPackFactory() throws Exception {
- ObjectId master = remote.branch("master").commit().create();
-
- final AtomicInteger rejected = new AtomicInteger();
- TestProtocol<User> proto = registerProto((User req, Repository db) -> {
- if (!"user2".equals(req.name)) {
- rejected.incrementAndGet();
- throw new ServiceNotAuthorizedException();
- }
- return new UploadPack(db);
- }, new DefaultReceive());
-
- // Same repository, different users.
- URIish user1Uri = proto.register(new User("user1"), remote.getRepository());
- URIish user2Uri = proto.register(new User("user2"), remote.getRepository());
-
- try (Git git = new Git(local.getRepository())) {
- try {
- git.fetch()
- .setRemote(user1Uri.toString())
- .setRefSpecs(MASTER)
- .call();
- fail("accepted not permitted fetch");
- } catch (InvalidRemoteException expected) {
- // Expected.
- }
- assertEquals(1, rejected.get());
- assertNull(local.getRepository().exactRef("refs/heads/master"));
-
- git.fetch()
- .setRemote(user2Uri.toString())
- .setRefSpecs(MASTER)
- .call();
- assertEquals(1, rejected.get());
- assertEquals(master,
- local.getRepository().exactRef("refs/heads/master").getObjectId());
- }
- }
-
- @Test
- public void testReceivePackFactory() throws Exception {
- ObjectId master = local.branch("master").commit().create();
-
- final AtomicInteger rejected = new AtomicInteger();
- TestProtocol<User> proto = registerProto(new DefaultUpload(),
- (User req, Repository db) -> {
- if (!"user2".equals(req.name)) {
- rejected.incrementAndGet();
- throw new ServiceNotAuthorizedException();
- }
- return new ReceivePack(db);
- });
-
- // Same repository, different users.
- URIish user1Uri = proto.register(new User("user1"), remote.getRepository());
- URIish user2Uri = proto.register(new User("user2"), remote.getRepository());
-
- try (Git git = new Git(local.getRepository())) {
- try {
- git.push()
- .setRemote(user1Uri.toString())
- .setRefSpecs(HEADS)
- .call();
- fail("accepted not permitted push");
- } catch (TransportException expected) {
- assertTrue(expected.getMessage().contains(
- JGitText.get().pushNotPermitted));
- }
- assertEquals(1, rejected.get());
- assertNull(remote.getRepository().exactRef("refs/heads/master"));
-
- git.push()
- .setRemote(user2Uri.toString())
- .setRefSpecs(HEADS)
- .call();
- assertEquals(1, rejected.get());
- assertEquals(master,
- remote.getRepository().exactRef("refs/heads/master").getObjectId());
- }
- }
-
- private TestProtocol<User> registerDefault() {
- return registerProto(new DefaultUpload(), new DefaultReceive());
- }
-
- private TestProtocol<User> registerProto(UploadPackFactory<User> upf,
- ReceivePackFactory<User> rpf) {
- TestProtocol<User> proto = new TestProtocol<>(upf, rpf);
- protos.add(proto);
- Transport.register(proto);
- return proto;
- }
- }
|