UploadPack: Fix NPE when traversing a tag chain

Always parse RevTags including their body before getting their object
to ensure that non-cached objects are handled correctly when traversing
a tag chain. An NPE in UploadPack#addTagChain will occur on a depth=1
fetch of a branch containing a tag chain and the ref to one of the
middle tags in the chain is deleted.

Change-Id: Ifd8fe868869070b365df926fec5dcd8e64d4f521
Signed-off-by: Kaushik Lingarkar <quic_kaushikl@quicinc.com>
This commit is contained in:
Kaushik Lingarkar 2023-04-04 18:05:53 -07:00 committed by Matthias Sohn
parent 40daa780ef
commit 064691e90c
2 changed files with 83 additions and 13 deletions

View File

@ -2538,6 +2538,75 @@ public void onSendPack(UploadPack up,
}
}
@Test
public void testSingleBranchShallowCloneTagChainWithReflessTag() throws Exception {
RevCommit one = remote.commit().message("1").create();
remote.update("master", one);
RevTag tag1 = remote.tag("t1", one);
remote.lightweightTag("t1", tag1);
RevTag tag2 = remote.tag("t2", tag1);
RevTag tag3 = remote.tag("t3", tag2);
remote.lightweightTag("t3", tag3);
UploadPack uploadPack = new UploadPack(remote.getRepository());
ByteArrayOutputStream cli = new ByteArrayOutputStream();
PacketLineOut clientWant = new PacketLineOut(cli);
clientWant.writeString("want " + one.name() + " include-tag");
clientWant.writeString("deepen 1\n");
clientWant.end();
clientWant.writeString("done\n");
try (ByteArrayOutputStream serverResponse = new ByteArrayOutputStream()) {
uploadPack.setPreUploadHook(new PreUploadHook() {
@Override
public void onBeginNegotiateRound(UploadPack up,
Collection<? extends ObjectId> wants, int cntOffered)
throws ServiceMayNotContinueException {
// Do nothing.
}
@Override
public void onEndNegotiateRound(UploadPack up,
Collection<? extends ObjectId> wants, int cntCommon,
int cntNotFound, boolean ready)
throws ServiceMayNotContinueException {
// Do nothing.
}
@Override
public void onSendPack(UploadPack up,
Collection<? extends ObjectId> wants,
Collection<? extends ObjectId> haves)
throws ServiceMayNotContinueException {
// collect pack data
serverResponse.reset();
}
});
uploadPack.upload(new ByteArrayInputStream(cli.toByteArray()),
serverResponse, System.err);
ByteArrayInputStream packReceived = new ByteArrayInputStream(
serverResponse.toByteArray());
PackLock lock = null;
try (ObjectInserter ins = client.newObjectInserter()) {
PackParser parser = ins.newPackParser(packReceived);
parser.setAllowThin(true);
parser.setLockMessage("receive-tag-chain");
ProgressMonitor mlc = NullProgressMonitor.INSTANCE;
lock = parser.parse(mlc, mlc);
ins.flush();
} finally {
if (lock != null) {
lock.unlock();
}
}
InMemoryRepository.MemObjDatabase objDb = client
.getObjectDatabase();
assertTrue(objDb.has(one.toObjectId()));
}
}
@Test
public void testSafeToClearRefsInFetchV0() throws Exception {
server =

View File

@ -2405,11 +2405,11 @@ else if (ref.getName().startsWith(Constants.R_HEADS))
if (peeledId == null || objectId == null)
continue;
objectId = ref.getObjectId();
if (pw.willInclude(peeledId) && !pw.willInclude(objectId)) {
RevObject o = rw.parseAny(objectId);
addTagChain(o, pw);
pw.addObject(o);
if (pw.willInclude(peeledId)) {
// We don't need to handle parseTag throwing an
// IncorrectObjectTypeException as we only reach
// here when ref is an annotated tag
addTagChain(rw.parseTag(objectId), pw);
}
}
}
@ -2459,15 +2459,16 @@ private static void findSymrefs(
}
private void addTagChain(
RevObject o, PackWriter pw) throws IOException {
while (Constants.OBJ_TAG == o.getType()) {
RevTag t = (RevTag) o;
o = t.getObject();
if (o.getType() == Constants.OBJ_TAG && !pw.willInclude(o.getId())) {
walk.parseBody(o);
pw.addObject(o);
RevTag tag, PackWriter pw) throws IOException {
RevObject o = tag;
do {
tag = (RevTag) o;
walk.parseBody(tag);
if (!pw.willInclude(tag.getId())) {
pw.addObject(tag);
}
}
o = tag.getObject();
} while (Constants.OBJ_TAG == o.getType());
}
private static class ResponseBufferedOutputStream extends OutputStream {