From e41e8f8c3bc9f5edab1d271464364f95620ece8c Mon Sep 17 00:00:00 2001 From: James Moger <james.moger@gitblit.com> Date: Thu, 19 Nov 2015 17:55:38 -0500 Subject: [PATCH] Create filestore directory on startup --- src/main/java/com/gitblit/servlet/BranchGraphServlet.java | 50 ++++++++++++++++++++++++++++++++++++++++++-------- 1 files changed, 42 insertions(+), 8 deletions(-) diff --git a/src/main/java/com/gitblit/servlet/BranchGraphServlet.java b/src/main/java/com/gitblit/servlet/BranchGraphServlet.java index 1f6d3a8..85fbb74 100644 --- a/src/main/java/com/gitblit/servlet/BranchGraphServlet.java +++ b/src/main/java/com/gitblit/servlet/BranchGraphServlet.java @@ -36,11 +36,14 @@ import java.util.TreeSet; import javax.imageio.ImageIO; +import com.google.inject.Inject; +import com.google.inject.Singleton; import javax.servlet.ServletException; import javax.servlet.http.HttpServlet; import javax.servlet.http.HttpServletRequest; import javax.servlet.http.HttpServletResponse; +import org.eclipse.jgit.lib.ObjectId; import org.eclipse.jgit.lib.Ref; import org.eclipse.jgit.lib.Repository; import org.eclipse.jgit.revplot.AbstractPlotRenderer; @@ -49,12 +52,13 @@ import org.eclipse.jgit.revplot.PlotLane; import org.eclipse.jgit.revplot.PlotWalk; import org.eclipse.jgit.revwalk.RevCommit; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; import com.gitblit.Constants; import com.gitblit.IStoredSettings; import com.gitblit.Keys; import com.gitblit.manager.IRepositoryManager; -import com.gitblit.manager.IRuntimeManager; import com.gitblit.utils.JGitUtils; import com.gitblit.utils.StringUtils; @@ -64,6 +68,7 @@ * @author James Moger * */ +@Singleton public class BranchGraphServlet extends HttpServlet { private static final long serialVersionUID = 1L; @@ -75,23 +80,26 @@ private static final int RIGHT_PAD = 2; + private final Logger log = LoggerFactory.getLogger(getClass()); + private final Stroke[] strokeCache; - private final IStoredSettings settings; + private IStoredSettings settings; - private final IRepositoryManager repositoryManager; + private IRepositoryManager repositoryManager; + @Inject public BranchGraphServlet( - IRuntimeManager runtimeManager, + IStoredSettings settings, IRepositoryManager repositoryManager) { - super(); - this.settings = runtimeManager.getSettings(); + this.settings = settings; this.repositoryManager = repositoryManager; strokeCache = new Stroke[4]; - for (int i = 1; i < strokeCache.length; i++) + for (int i = 1; i < strokeCache.length; i++) { strokeCache[i] = new BasicStroke(i); + } } /** @@ -115,6 +123,9 @@ @Override protected long getLastModified(HttpServletRequest req) { String repository = req.getParameter("r"); + if (StringUtils.isEmpty(repository)) { + return 0; + } String objectId = req.getParameter("h"); Repository r = null; try { @@ -122,8 +133,15 @@ if (StringUtils.isEmpty(objectId)) { objectId = JGitUtils.getHEADRef(r); } + ObjectId id = r.resolve(objectId); + if (id == null) { + return 0; + } RevCommit commit = JGitUtils.getCommit(r, objectId); return JGitUtils.getCommitDate(commit).getTime(); + } catch (Exception e) { + log.error("Failed to determine last modified", e); + return 0; } finally { if (r != null) { r.close(); @@ -139,17 +157,33 @@ PlotWalk rw = null; try { String repository = request.getParameter("r"); + if (StringUtils.isEmpty(repository)) { + response.setStatus(HttpServletResponse.SC_BAD_REQUEST); + response.getWriter().append("Bad request"); + return; + } String objectId = request.getParameter("h"); String length = request.getParameter("l"); r = repositoryManager.getRepository(repository); + if (r == null) { + response.setStatus(HttpServletResponse.SC_BAD_REQUEST); + response.getWriter().append("Bad request"); + return; + } rw = new PlotWalk(r); if (StringUtils.isEmpty(objectId)) { objectId = JGitUtils.getHEADRef(r); } - rw.markStart(rw.lookupCommit(r.resolve(objectId))); + ObjectId id = r.resolve(objectId); + if (id == null) { + response.setStatus(HttpServletResponse.SC_BAD_REQUEST); + response.getWriter().append("Bad request"); + return; + } + rw.markStart(rw.lookupCommit(id)); // default to the items-per-page setting, unless specified int maxCommits = settings.getInteger(Keys.web.itemsPerPage, 50); -- Gitblit v1.9.1