X-Git-Url: https://gitweb.ps.run/ps-cgit/blobdiff_plain/9fb53af215639fcd3bfb876fa9c8bac221244bdf..0928d8827a714f3908efa7eb9eb4cde28761af26:/ui-tree.c

diff --git a/ui-tree.c b/ui-tree.c
index db63e13..006ca47 100644
--- a/ui-tree.c
+++ b/ui-tree.c
@@ -50,7 +50,7 @@ static int print_entry(const unsigned char *sha1, const char *base,
 					  cgit_query_path ? cgit_query_path : "",
 					  pathname)));
 	}
-	htmlf("'>%s</a></div></td>", name);
+	htmlf("'>%s</a></td>", name);
 	htmlf("<td class='filesize'>%li</td>", size);
 
 	html("<td class='links'><a href='");
@@ -69,8 +69,18 @@ void cgit_print_tree(const char *rev, const char *hex, char *path)
 {
 	struct tree *tree;
 	unsigned char sha1[20];
+	struct commit *commit;
 
 	curr_rev = xstrdup(rev);
+	get_sha1(rev, sha1);
+	commit = lookup_commit_reference(sha1);
+	if (!commit || parse_commit(commit)) {
+		cgit_print_error(fmt("Invalid head: %s", rev));
+		return;
+	}
+	if (!hex)
+		hex = sha1_to_hex(commit->tree->object.sha1);
+
 	if (get_sha1_hex(hex, sha1)) {
 		cgit_print_error(fmt("Invalid object id: %s", hex));
 		return;