1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
/*
* Copyright (C) 2023, Google LLC. 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 java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.security.MessageDigest;
import java.util.zip.Deflater;
import org.eclipse.jgit.lib.AnyObjectId;
import org.eclipse.jgit.lib.Constants;
import org.eclipse.jgit.util.NB;
import org.eclipse.jgit.util.TemporaryBuffer;
import org.eclipse.jgit.util.TemporaryBuffer.Heap;
/**
* Helper class to create packs for tests.
*/
public class InMemoryPack {
private final Heap tinyPack;
public InMemoryPack() {
this(1024);
}
public InMemoryPack(int size) {
this.tinyPack = new TemporaryBuffer.Heap(size);
}
public InMemoryPack header(int cnt)
throws IOException {
final byte[] hdr = new byte[8];
NB.encodeInt32(hdr, 0, 2);
NB.encodeInt32(hdr, 4, cnt);
tinyPack.write(Constants.PACK_SIGNATURE);
tinyPack.write(hdr, 0, 8);
return this;
}
public InMemoryPack write(int i) throws IOException {
tinyPack.write(i);
return this;
}
public InMemoryPack deflate(byte[] content)
throws IOException {
Deflater deflater = new Deflater();
byte[] buf = new byte[128];
deflater.setInput(content, 0, content.length);
deflater.finish();
do {
final int n = deflater.deflate(buf, 0, buf.length);
if (n > 0)
tinyPack.write(buf, 0, n);
} while (!deflater.finished());
return this;
}
public InMemoryPack copyRaw(AnyObjectId o) throws IOException {
o.copyRawTo(tinyPack);
return this;
}
public InMemoryPack digest() throws IOException {
MessageDigest md = Constants.newMessageDigest();
md.update(tinyPack.toByteArray());
tinyPack.write(md.digest());
return this;
}
public InputStream toInputStream() throws IOException {
return new ByteArrayInputStream(tinyPack.toByteArray());
}
public byte[] toByteArray() throws IOException {
return tinyPack.toByteArray();
}
}
|
