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 package org.eclipse.jgit.http.test;
45
46 import static org.junit.Assert.assertEquals;
47 import static org.junit.Assert.assertTrue;
48
49 import java.io.ByteArrayOutputStream;
50 import java.io.IOException;
51 import java.io.InputStream;
52 import java.io.OutputStream;
53 import java.net.HttpURLConnection;
54 import java.net.URL;
55
56 import javax.servlet.http.HttpServletRequest;
57
58 import org.eclipse.jetty.servlet.ServletContextHandler;
59 import org.eclipse.jetty.servlet.ServletHolder;
60 import org.eclipse.jgit.errors.RepositoryNotFoundException;
61 import org.eclipse.jgit.http.server.GitServlet;
62 import org.eclipse.jgit.http.server.GitSmartHttpTools;
63 import org.eclipse.jgit.internal.JGitText;
64 import org.eclipse.jgit.junit.TestRepository;
65 import org.eclipse.jgit.junit.http.HttpTestCase;
66 import org.eclipse.jgit.lib.Constants;
67 import org.eclipse.jgit.lib.ObjectId;
68 import org.eclipse.jgit.lib.Repository;
69 import org.eclipse.jgit.lib.StoredConfig;
70 import org.eclipse.jgit.revwalk.RevBlob;
71 import org.eclipse.jgit.transport.PacketLineIn;
72 import org.eclipse.jgit.transport.PacketLineOut;
73 import org.eclipse.jgit.transport.URIish;
74 import org.eclipse.jgit.util.NB;
75 import org.junit.Before;
76 import org.junit.Test;
77
78 public class ProtocolErrorTest extends HttpTestCase {
79 private Repository remoteRepository;
80
81 private URIish remoteURI;
82
83 private RevBlob a_blob;
84
85 @Override
86 @Before
87 public void setUp() throws Exception {
88 super.setUp();
89
90 final TestRepository<Repository> src = createTestRepository();
91 final String srcName = src.getRepository().getDirectory().getName();
92
93 ServletContextHandler app = server.addContext("/git");
94 GitServlet gs = new GitServlet();
95 gs.setRepositoryResolver((HttpServletRequest req, String name) -> {
96 if (!name.equals(srcName)) {
97 throw new RepositoryNotFoundException(name);
98 }
99 final Repository db = src.getRepository();
100 db.incrementOpen();
101 return db;
102 });
103 app.addServlet(new ServletHolder(gs), "/*");
104
105 server.setUp();
106
107 remoteRepository = src.getRepository();
108 remoteURI = toURIish(app, srcName);
109
110 StoredConfig cfg = remoteRepository.getConfig();
111 cfg.setBoolean("http", null, "receivepack", true);
112 cfg.save();
113
114 a_blob = src.blob("a");
115 }
116
117 @Test
118 public void testPush_UnpackError_TruncatedPack() throws Exception {
119 StringBuilder sb = new StringBuilder();
120 sb.append(ObjectId.zeroId().name());
121 sb.append(' ');
122 sb.append(a_blob.name());
123 sb.append(' ');
124 sb.append("refs/objects/A");
125 sb.append('\0');
126 sb.append("report-status");
127
128 ByteArrayOutputStream reqbuf = new ByteArrayOutputStream();
129 PacketLineOut reqpck = new PacketLineOut(reqbuf);
130 reqpck.writeString(sb.toString());
131 reqpck.end();
132
133 packHeader(reqbuf, 1);
134
135 byte[] reqbin = reqbuf.toByteArray();
136
137 URL u = new URL(remoteURI.toString() + "/git-receive-pack");
138 HttpURLConnection c = (HttpURLConnection) u.openConnection();
139 try {
140 c.setRequestMethod("POST");
141 c.setDoOutput(true);
142 c.setRequestProperty("Content-Type",
143 GitSmartHttpTools.RECEIVE_PACK_REQUEST_TYPE);
144 c.setFixedLengthStreamingMode(reqbin.length);
145 try (OutputStream out = c.getOutputStream()) {
146 out.write(reqbin);
147 }
148
149 assertEquals(200, c.getResponseCode());
150 assertEquals(GitSmartHttpTools.RECEIVE_PACK_RESULT_TYPE,
151 c.getContentType());
152
153 try (InputStream rawin = c.getInputStream()) {
154 PacketLineIn pckin = new PacketLineIn(rawin);
155 assertEquals("unpack error "
156 + JGitText.get().packfileIsTruncatedNoParam,
157 pckin.readString());
158 assertEquals("ng refs/objects/A n/a (unpacker error)",
159 pckin.readString());
160 assertTrue(PacketLineIn.isEnd(pckin.readString()));
161 }
162 } finally {
163 c.disconnect();
164 }
165 }
166
167 private static void packHeader(ByteArrayOutputStream tinyPack, int cnt)
168 throws IOException {
169 final byte[] hdr = new byte[8];
170 NB.encodeInt32(hdr, 0, 2);
171 NB.encodeInt32(hdr, 4, cnt);
172
173 tinyPack.write(Constants.PACK_SIGNATURE);
174 tinyPack.write(hdr, 0, 8);
175 }
176 }