Browse Source

Merge changes from topic 'moving-to-request-2'

* changes:
  UploadPack v0: Extract "agent" client capability at parse time
  UploadPack: Return correct peer user agent on v2 requests
stable-5.2
Jonathan Nieder 6 years ago committed by Gerrit Code Review @ Eclipse.org
parent
commit
a579a56e3a
  1. 15
      org.eclipse.jgit.test/tst/org/eclipse/jgit/internal/transport/parser/FirstWantTest.java
  2. 70
      org.eclipse.jgit.test/tst/org/eclipse/jgit/transport/UploadPackTest.java
  3. 38
      org.eclipse.jgit/src/org/eclipse/jgit/internal/transport/parser/FirstWant.java
  4. 24
      org.eclipse.jgit/src/org/eclipse/jgit/transport/FetchRequest.java
  5. 20
      org.eclipse.jgit/src/org/eclipse/jgit/transport/FetchV0Request.java
  6. 15
      org.eclipse.jgit/src/org/eclipse/jgit/transport/FetchV2Request.java
  7. 1
      org.eclipse.jgit/src/org/eclipse/jgit/transport/ProtocolV0Parser.java
  8. 12
      org.eclipse.jgit/src/org/eclipse/jgit/transport/UploadPack.java

15
org.eclipse.jgit.test/tst/org/eclipse/jgit/internal/transport/parser/FirstWantTest.java

@ -43,6 +43,7 @@
package org.eclipse.jgit.internal.transport.parser; package org.eclipse.jgit.internal.transport.parser;
import static org.junit.Assert.assertEquals; import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue; import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail; import static org.junit.Assert.fail;
@ -66,9 +67,9 @@ public class FirstWantTest {
r.getLine()); r.getLine());
Set<String> capabilities = r.getCapabilities(); Set<String> capabilities = r.getCapabilities();
Set<String> expectedCapabilities = new HashSet<>( Set<String> expectedCapabilities = new HashSet<>(
Arrays.asList("no-progress", "include-tag", "ofs-delta", Arrays.asList("no-progress", "include-tag", "ofs-delta"));
"agent=JGit/unknown"));
assertEquals(expectedCapabilities, capabilities); assertEquals(expectedCapabilities, capabilities);
assertEquals("JGit/unknown", r.getAgent());
} }
@Test @Test
@ -79,6 +80,7 @@ public class FirstWantTest {
assertEquals("want b9d4d1eb2f93058814480eae9e1b67550f46ee38", assertEquals("want b9d4d1eb2f93058814480eae9e1b67550f46ee38",
r.getLine()); r.getLine());
assertTrue(r.getCapabilities().isEmpty()); assertTrue(r.getCapabilities().isEmpty());
assertNull(r.getAgent());
} }
private String makeFirstWantLine(String capability) { private String makeFirstWantLine(String capability) {
@ -110,7 +112,7 @@ public class FirstWantTest {
List<String> validNames = Arrays.asList( List<String> validNames = Arrays.asList(
"c", "cap", "C", "CAP", "1", "1cap", "cap-64k_test", "c", "cap", "C", "CAP", "1", "1cap", "cap-64k_test",
"-", "-cap", "-", "-cap",
"_", "_cap", "agent=pack.age/Version"); "_", "_cap");
for (String capability: validNames) { for (String capability: validNames) {
FirstWant r = FirstWant.fromLine(makeFirstWantLine(capability)); FirstWant r = FirstWant.fromLine(makeFirstWantLine(capability));
@ -118,4 +120,11 @@ public class FirstWantTest {
assertTrue(r.getCapabilities().contains(capability)); assertTrue(r.getCapabilities().contains(capability));
} }
} }
@Test
public void testFirstWantValidAgentName() throws PackProtocolException {
FirstWant r = FirstWant.fromLine(makeFirstWantLine("agent=pack.age/Version"));
assertEquals(r.getCapabilities().size(), 0);
assertEquals("pack.age/Version", r.getAgent());
}
} }

70
org.eclipse.jgit.test/tst/org/eclipse/jgit/transport/UploadPackTest.java

@ -429,17 +429,7 @@ public class UploadPackTest {
RefFilter refFilter, ProtocolV2Hook hook, String... inputLines) RefFilter refFilter, ProtocolV2Hook hook, String... inputLines)
throws Exception { throws Exception {
ByteArrayOutputStream send = new ByteArrayOutputStream(); ByteArrayInputStream send = linesAsInputStream(inputLines);
PacketLineOut pckOut = new PacketLineOut(send);
for (String line : inputLines) {
if (line == PacketLineIn.END) {
pckOut.end();
} else if (line == PacketLineIn.DELIM) {
pckOut.writeDelim();
} else {
pckOut.writeString(line);
}
}
server.getConfig().setString("protocol", null, "version", "2"); server.getConfig().setString("protocol", null, "version", "2");
UploadPack up = new UploadPack(server); UploadPack up = new UploadPack(server);
@ -453,11 +443,28 @@ public class UploadPackTest {
} }
ByteArrayOutputStream recv = new ByteArrayOutputStream(); ByteArrayOutputStream recv = new ByteArrayOutputStream();
up.upload(new ByteArrayInputStream(send.toByteArray()), recv, null); up.upload(send, recv, null);
return new ByteArrayInputStream(recv.toByteArray()); return new ByteArrayInputStream(recv.toByteArray());
} }
private static ByteArrayInputStream linesAsInputStream(String... inputLines)
throws IOException {
try (ByteArrayOutputStream send = new ByteArrayOutputStream()) {
PacketLineOut pckOut = new PacketLineOut(send);
for (String line : inputLines) {
if (line == PacketLineIn.END) {
pckOut.end();
} else if (line == PacketLineIn.DELIM) {
pckOut.writeDelim();
} else {
pckOut.writeString(line);
}
}
return new ByteArrayInputStream(send.toByteArray());
}
}
/* /*
* Invokes UploadPack with protocol v2 and sends it the given lines. * Invokes UploadPack with protocol v2 and sends it the given lines.
* Returns UploadPack's output stream, not including the capability * Returns UploadPack's output stream, not including the capability
@ -1552,6 +1559,45 @@ public class UploadPackTest {
assertTrue(client.hasObject(three.toObjectId())); assertTrue(client.hasObject(three.toObjectId()));
} }
@Test
public void testGetPeerAgentProtocolV0() throws Exception {
RevCommit one = remote.commit().message("1").create();
remote.update("one", one);
UploadPack up = new UploadPack(server);
ByteArrayInputStream send = linesAsInputStream(
"want " + one.getName() + " agent=JGit-test/1.2.3\n",
PacketLineIn.END,
"have 11cedf1b796d44207da702f7d420684022fc0f09\n", "done\n");
ByteArrayOutputStream recv = new ByteArrayOutputStream();
up.upload(send, recv, null);
assertEquals(up.getPeerUserAgent(), "JGit-test/1.2.3");
}
@Test
public void testGetPeerAgentProtocolV2() throws Exception {
server.getConfig().setString("protocol", null, "version", "2");
RevCommit one = remote.commit().message("1").create();
remote.update("one", one);
UploadPack up = new UploadPack(server);
up.setExtraParameters(Sets.of("version=2"));
ByteArrayInputStream send = linesAsInputStream(
"command=fetch\n", "agent=JGit-test/1.2.4\n",
PacketLineIn.DELIM, "want " + one.getName() + "\n",
"have 11cedf1b796d44207da702f7d420684022fc0f09\n", "done\n",
PacketLineIn.END);
ByteArrayOutputStream recv = new ByteArrayOutputStream();
up.upload(send, recv, null);
assertEquals(up.getPeerUserAgent(), "JGit-test/1.2.4");
}
private static class RejectAllRefFilter implements RefFilter { private static class RejectAllRefFilter implements RefFilter {
@Override @Override
public Map<String, Ref> filter(Map<String, Ref> refs) { public Map<String, Ref> filter(Map<String, Ref> refs) {

38
org.eclipse.jgit/src/org/eclipse/jgit/internal/transport/parser/FirstWant.java

@ -42,11 +42,13 @@
*/ */
package org.eclipse.jgit.internal.transport.parser; package org.eclipse.jgit.internal.transport.parser;
import java.util.Arrays; import static org.eclipse.jgit.transport.GitProtocolConstants.OPTION_AGENT;
import java.util.Collections; import java.util.Collections;
import java.util.HashSet; import java.util.HashSet;
import java.util.Set; import java.util.Set;
import org.eclipse.jgit.annotations.Nullable;
import org.eclipse.jgit.errors.PackProtocolException; import org.eclipse.jgit.errors.PackProtocolException;
import org.eclipse.jgit.internal.JGitText; import org.eclipse.jgit.internal.JGitText;
@ -72,6 +74,11 @@ public class FirstWant {
private final Set<String> capabilities; private final Set<String> capabilities;
@Nullable
private final String agent;
private static final String AGENT_PREFIX = OPTION_AGENT + '=';
/** /**
* Parse the first want line in the protocol v0/v1 pack negotiation. * Parse the first want line in the protocol v0/v1 pack negotiation.
* *
@ -84,6 +91,7 @@ public class FirstWant {
public static FirstWant fromLine(String line) throws PackProtocolException { public static FirstWant fromLine(String line) throws PackProtocolException {
String wantLine; String wantLine;
Set<String> capabilities; Set<String> capabilities;
String agent = null;
if (line.length() > 45) { if (line.length() > 45) {
String opt = line.substring(45); String opt = line.substring(45);
@ -91,8 +99,15 @@ public class FirstWant {
throw new PackProtocolException(JGitText.get().wantNoSpaceWithCapabilities); throw new PackProtocolException(JGitText.get().wantNoSpaceWithCapabilities);
} }
opt = opt.substring(1); opt = opt.substring(1);
HashSet<String> opts = new HashSet<>(
Arrays.asList(opt.split(" "))); //$NON-NLS-1$ HashSet<String> opts = new HashSet<>();
for (String clientCapability : opt.split(" ")) { //$NON-NLS-1$
if (clientCapability.startsWith(AGENT_PREFIX)) {
agent = clientCapability.substring(AGENT_PREFIX.length());
} else {
opts.add(clientCapability);
}
}
wantLine = line.substring(0, 45); wantLine = line.substring(0, 45);
capabilities = Collections.unmodifiableSet(opts); capabilities = Collections.unmodifiableSet(opts);
} else { } else {
@ -100,12 +115,14 @@ public class FirstWant {
capabilities = Collections.emptySet(); capabilities = Collections.emptySet();
} }
return new FirstWant(wantLine, capabilities); return new FirstWant(wantLine, capabilities, agent);
} }
private FirstWant(String line, Set<String> capabilities) { private FirstWant(String line, Set<String> capabilities,
@Nullable String agent) {
this.line = line; this.line = line;
this.capabilities = capabilities; this.capabilities = capabilities;
this.agent = agent;
} }
/** @return non-capabilities part of the line. */ /** @return non-capabilities part of the line. */
@ -113,8 +130,17 @@ public class FirstWant {
return line; return line;
} }
/** @return capabilities parsed from the line as an immutable set. */ /**
* @return capabilities parsed from the line as an immutable set (excluding
* agent).
*/
public Set<String> getCapabilities() { public Set<String> getCapabilities() {
return capabilities; return capabilities;
} }
/** @return client user agent parsed from the line. */
@Nullable
public String getAgent() {
return agent;
}
} }

24
org.eclipse.jgit/src/org/eclipse/jgit/transport/FetchRequest.java

@ -48,6 +48,7 @@ import java.util.List;
import java.util.Set; import java.util.Set;
import org.eclipse.jgit.annotations.NonNull; import org.eclipse.jgit.annotations.NonNull;
import org.eclipse.jgit.annotations.Nullable;
import org.eclipse.jgit.lib.ObjectId; import org.eclipse.jgit.lib.ObjectId;
/** /**
@ -69,6 +70,9 @@ abstract class FetchRequest {
final List<String> deepenNotRefs; final List<String> deepenNotRefs;
@Nullable
final String agent;
/** /**
* Initialize the common fields of a fetch request. * Initialize the common fields of a fetch request.
* *
@ -88,11 +92,13 @@ abstract class FetchRequest {
* @param deepenSince * @param deepenSince
* Requests that the shallow clone/fetch should be cut at a * Requests that the shallow clone/fetch should be cut at a
* specific time, instead of depth * specific time, instead of depth
* @param agent
* agent as reported by the client in the request body
*/ */
FetchRequest(@NonNull Set<ObjectId> wantIds, int depth, FetchRequest(@NonNull Set<ObjectId> wantIds, int depth,
@NonNull Set<ObjectId> clientShallowCommits, long filterBlobLimit, @NonNull Set<ObjectId> clientShallowCommits, long filterBlobLimit,
@NonNull Set<String> clientCapabilities, int deepenSince, @NonNull Set<String> clientCapabilities, int deepenSince,
@NonNull List<String> deepenNotRefs) { @NonNull List<String> deepenNotRefs, @Nullable String agent) {
this.wantIds = requireNonNull(wantIds); this.wantIds = requireNonNull(wantIds);
this.depth = depth; this.depth = depth;
this.clientShallowCommits = requireNonNull(clientShallowCommits); this.clientShallowCommits = requireNonNull(clientShallowCommits);
@ -100,6 +106,7 @@ abstract class FetchRequest {
this.clientCapabilities = requireNonNull(clientCapabilities); this.clientCapabilities = requireNonNull(clientCapabilities);
this.deepenSince = deepenSince; this.deepenSince = deepenSince;
this.deepenNotRefs = requireNonNull(deepenNotRefs); this.deepenNotRefs = requireNonNull(deepenNotRefs);
this.agent = agent;
} }
/** /**
@ -146,7 +153,11 @@ abstract class FetchRequest {
* These options are listed and well-defined in the git protocol * These options are listed and well-defined in the git protocol
* specification. * specification.
* *
* @return capabilities sent by the client * The agent capability is not included in this set. It can be retrieved via
* {@link #getAgent()}.
*
* @return capabilities sent by the client (excluding the "agent"
* capability)
*/ */
@NonNull @NonNull
Set<String> getClientCapabilities() { Set<String> getClientCapabilities() {
@ -171,4 +182,13 @@ abstract class FetchRequest {
List<String> getDeepenNotRefs() { List<String> getDeepenNotRefs() {
return deepenNotRefs; return deepenNotRefs;
} }
/**
* @return string identifying the agent (as sent in the request body by the
* client)
*/
@Nullable
String getAgent() {
return agent;
}
} }

20
org.eclipse.jgit/src/org/eclipse/jgit/transport/FetchV0Request.java

@ -48,6 +48,7 @@ import java.util.HashSet;
import java.util.Set; import java.util.Set;
import org.eclipse.jgit.annotations.NonNull; import org.eclipse.jgit.annotations.NonNull;
import org.eclipse.jgit.annotations.Nullable;
import org.eclipse.jgit.lib.ObjectId; import org.eclipse.jgit.lib.ObjectId;
/** /**
@ -57,9 +58,9 @@ final class FetchV0Request extends FetchRequest {
FetchV0Request(@NonNull Set<ObjectId> wantIds, int depth, FetchV0Request(@NonNull Set<ObjectId> wantIds, int depth,
@NonNull Set<ObjectId> clientShallowCommits, long filterBlobLimit, @NonNull Set<ObjectId> clientShallowCommits, long filterBlobLimit,
@NonNull Set<String> clientCapabilities) { @NonNull Set<String> clientCapabilities, @Nullable String agent) {
super(wantIds, depth, clientShallowCommits, filterBlobLimit, super(wantIds, depth, clientShallowCommits, filterBlobLimit,
clientCapabilities, 0, Collections.emptyList()); clientCapabilities, 0, Collections.emptyList(), agent);
} }
static final class Builder { static final class Builder {
@ -74,6 +75,8 @@ final class FetchV0Request extends FetchRequest {
final Set<String> clientCaps = new HashSet<>(); final Set<String> clientCaps = new HashSet<>();
String agent;
/** /**
* @param objectId * @param objectId
* object id received in a "want" line * object id received in a "want" line
@ -115,6 +118,16 @@ final class FetchV0Request extends FetchRequest {
return this; return this;
} }
/**
* @param clientAgent
* agent line sent by the client in the request body
* @return this builder
*/
Builder setAgent(String clientAgent) {
agent = clientAgent;
return this;
}
/** /**
* @param filterBlobLim * @param filterBlobLim
* blob limit set in a "filter" line * blob limit set in a "filter" line
@ -127,7 +140,8 @@ final class FetchV0Request extends FetchRequest {
FetchV0Request build() { FetchV0Request build() {
return new FetchV0Request(wantIds, depth, clientShallowCommits, return new FetchV0Request(wantIds, depth, clientShallowCommits,
filterBlobLimit, clientCaps); filterBlobLimit, clientCaps, agent);
} }
} }
} }

15
org.eclipse.jgit/src/org/eclipse/jgit/transport/FetchV2Request.java

@ -71,9 +71,6 @@ public final class FetchV2Request extends FetchRequest {
private final boolean doneReceived; private final boolean doneReceived;
@Nullable
private final String agent;
@NonNull @NonNull
private final List<String> serverOptions; private final List<String> serverOptions;
@ -86,11 +83,10 @@ public final class FetchV2Request extends FetchRequest {
boolean doneReceived, @NonNull Set<String> clientCapabilities, boolean doneReceived, @NonNull Set<String> clientCapabilities,
@Nullable String agent, @NonNull List<String> serverOptions) { @Nullable String agent, @NonNull List<String> serverOptions) {
super(wantIds, depth, clientShallowCommits, filterBlobLimit, super(wantIds, depth, clientShallowCommits, filterBlobLimit,
clientCapabilities, deepenSince, deepenNotRefs); clientCapabilities, deepenSince, deepenNotRefs, agent);
this.peerHas = requireNonNull(peerHas); this.peerHas = requireNonNull(peerHas);
this.wantedRefs = requireNonNull(wantedRefs); this.wantedRefs = requireNonNull(wantedRefs);
this.doneReceived = doneReceived; this.doneReceived = doneReceived;
this.agent = agent;
this.serverOptions = requireNonNull(serverOptions); this.serverOptions = requireNonNull(serverOptions);
} }
@ -117,15 +113,6 @@ public final class FetchV2Request extends FetchRequest {
return doneReceived; return doneReceived;
} }
/**
* @return string identifying the agent (as sent in the request body by the
* client)
*/
@Nullable
String getAgent() {
return agent;
}
/** /**
* Options received in server-option lines. The caller can choose to act on * Options received in server-option lines. The caller can choose to act on
* these in an application-specific way * these in an application-specific way

1
org.eclipse.jgit/src/org/eclipse/jgit/transport/ProtocolV0Parser.java

@ -143,6 +143,7 @@ final class ProtocolV0Parser {
if (line.length() > 45) { if (line.length() > 45) {
FirstWant firstLine = FirstWant.fromLine(line); FirstWant firstLine = FirstWant.fromLine(line);
reqBuilder.addClientCapabilities(firstLine.getCapabilities()); reqBuilder.addClientCapabilities(firstLine.getCapabilities());
reqBuilder.setAgent(firstLine.getAgent());
line = firstLine.getLine(); line = firstLine.getLine();
} }
} }

12
org.eclipse.jgit/src/org/eclipse/jgit/transport/UploadPack.java

@ -209,6 +209,11 @@ public class UploadPack {
/** @return capabilities parsed from the line. */ /** @return capabilities parsed from the line. */
public Set<String> getOptions() { public Set<String> getOptions() {
if (firstWant.getAgent() != null) {
Set<String> caps = new HashSet<>(firstWant.getCapabilities());
caps.add(OPTION_AGENT + '=' + firstWant.getAgent());
return caps;
}
return firstWant.getCapabilities(); return firstWant.getCapabilities();
} }
} }
@ -1369,12 +1374,11 @@ public class UploadPack {
* @since 4.0 * @since 4.0
*/ */
public String getPeerUserAgent() { public String getPeerUserAgent() {
if (currentRequest == null) { if (currentRequest != null && currentRequest.getAgent() != null) {
return userAgent; return currentRequest.getAgent();
} }
return UserAgent.getAgent(currentRequest.getClientCapabilities(), return userAgent;
userAgent);
} }
private boolean negotiate(FetchRequest req, private boolean negotiate(FetchRequest req,

Loading…
Cancel
Save