1 files changed, 24 insertions, 18 deletions

diff --git a/ui-shared.c b/ui-shared.c
index a03661a..88dd6b8 100644
--- a/ui-shared.c
+++ b/ui-shared.c
@@ -62,15 +62,17 @@ char *cgit_fileurl(const char *reponame, const char *pagename,
 {
+        char *tmp;
+        char *delim;
+
         if (cgit_virtual_root) {
-                if (query)
-                        return fmt("%s/%s/%s/%s?%s", cgit_virtual_root, reponame,
-                                   pagename, filename?filename:"", query);
-                else
-                        return fmt("%s/%s/%s/", cgit_virtual_root, reponame,
-                                   pagename);
+                tmp = fmt("%s/%s/%s/%s", cgit_virtual_root, reponame,
+                          pagename, (filename ? filename:""));
+                delim = "?";
         } else {
-                if (query)
-                        return fmt("?r=%s&p=%s&%s", reponame, pagename, query);
-                else
-                        return fmt("?r=%s&p=%s", reponame, pagename);
+                tmp = fmt("?url=%s/%s/%s", reponame, pagename,
+                          (filename ? filename : ""));
+                delim = "&";
         }
+        if (query)
+                tmp = fmt("%s%s%s", tmp, delim, query);
+        return tmp;
 }
@@ -423,9 +425,11 @@ int print_archive_ref(const char *refname, const unsigned char *sha1,
 
-void add_hidden_formfields(int incl_head, int incl_search)
+void add_hidden_formfields(int incl_head, int incl_search, char *page)
 {
+        char *url;
+
         if (!cgit_virtual_root) {
-                if (cgit_query_repo)
-                        html_hidden("r", cgit_query_repo);
-                if (cgit_query_page)
-                        html_hidden("p", cgit_query_page);
+                url = fmt("%s/%s", cgit_query_repo, page);
+                if (cgit_query_path)
+                        url = fmt("%s/%s", url, cgit_query_path);
+                html_hidden("url", url);
         }
@@ -485,3 +489,3 @@ void cgit_print_pageheader(char *title, int show_search)
                 html("<form method='get' action=''>\n");
-                add_hidden_formfields(0, 1);
+                add_hidden_formfields(0, 1, cgit_query_page);
                 html("<select name='h' onchange='this.form.submit();'>\n");
@@ -493,5 +497,7 @@ void cgit_print_pageheader(char *title, int show_search)
                 html("<form method='get' action='");
-                html_attr(cgit_pageurl(cgit_query_repo, "log", NULL));
+                if (cgit_virtual_root)
+                        html_attr(cgit_fileurl(cgit_query_repo, "log",
+                                               cgit_query_path, NULL));
                 html("'>\n");
-                add_hidden_formfields(1, 0);
+                add_hidden_formfields(1, 0, "log");
                 html("<select name='qt'>\n");