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 | 51 ++++++++++++++++++++++++++++++++++++++++----------- 1 files changed, 40 insertions(+), 11 deletions(-) diff --git a/src/main/java/com/gitblit/servlet/BranchGraphServlet.java b/src/main/java/com/gitblit/servlet/BranchGraphServlet.java index 3efe60d..85fbb74 100644 --- a/src/main/java/com/gitblit/servlet/BranchGraphServlet.java +++ b/src/main/java/com/gitblit/servlet/BranchGraphServlet.java @@ -36,13 +36,14 @@ import java.util.TreeSet; import javax.imageio.ImageIO; -import javax.inject.Inject; -import javax.inject.Singleton; +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; @@ -51,13 +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.Keys.web; import com.gitblit.manager.IRepositoryManager; -import com.gitblit.manager.IRuntimeManager; import com.gitblit.utils.JGitUtils; import com.gitblit.utils.StringUtils; @@ -79,24 +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); + } } /** @@ -120,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 { @@ -127,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(); @@ -144,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