7 files changed, 46 insertions, 4 deletions

diff --git a/cgit.c b/cgit.c
index 2afc598..513ea12 100644
--- a/cgit.c
+++ b/cgit.c
@@ -209,4 +209,5 @@ static void prepare_context(struct cgit_context *ctx)
         ctx->page.modified = time(NULL);
         ctx->page.expires = ctx->page.modified;
+        ctx->page.etag = NULL;
 }
 
@@ -288,4 +289,6 @@ static int prepare_repo_cmd(struct cgit_context *ctx)
                 tmp = xstrdup(ctx->qry.head);
                 ctx->qry.head = ctx->repo->defbranch;
+                ctx->page.status = 404;
+                ctx->page.statusmsg = "not found";
                 cgit_print_http_headers(ctx);
                 cgit_print_docstart(ctx);
@@ -432,4 +435,5 @@ int main(int argc, const char **argv)
 {
         const char *cgit_config_env = getenv("CGIT_CONFIG");
+        const char *method = getenv("REQUEST_METHOD");
         const char *path;
         char *qry;
@@ -478,4 +482,6 @@ int main(int argc, const char **argv)
         ttl = calc_ttl();
         ctx.page.expires += ttl*60;
+        if (method && !strcmp(method, "HEAD"))
+                ctx.cfg.nocache = 1;
         if (ctx.cfg.nocache)
                 ctx.cfg.cache_size = 0;
diff --git a/cgit.h b/cgit.h
index aed826a..78b30ba 100644
--- a/cgit.h
+++ b/cgit.h
@@ -182,5 +182,8 @@ struct cgit_page {
         char *charset;
         char *filename;
+        char *etag;
         char *title;
+        int status;
+        char *statusmsg;
 };
 
diff --git a/ui-atom.c b/ui-atom.c
index a6ea3ee..e5c31d9 100644
--- a/ui-atom.c
+++ b/ui-atom.c
@@ -53,5 +53,6 @@ void add_entry(struct commit *commit, char *host)
         html("</published>\n");
         if (host) {
-                html("<link rel='alternate' type='text/html' href='http://");
+                html("<link rel='alternate' type='text/html' href='");
+                html(cgit_httpscheme());
                 html_attr(host);
                 html_attr(cgit_pageurl(ctx.repo->url, "commit", NULL));
@@ -114,5 +115,6 @@ void cgit_print_atom(char *tip, char *path, int max_count)
         html("</subtitle>\n");
         if (host) {
-                html("<link rel='alternate' type='text/html' href='http://");
+                html("<link rel='alternate' type='text/html' href='");
+                html(cgit_httpscheme());
                 html_attr(host);
                 html_attr(cgit_repourl(ctx.repo->url));
diff --git a/ui-blob.c b/ui-blob.c
index 3cda03d..2ccd31d 100644
--- a/ui-blob.c
+++ b/ui-blob.c
@@ -28,5 +28,5 @@ void cgit_print_blob(const char *hex, char *path, const char *head)
         unsigned char sha1[20];
         enum object_type type;
-        unsigned char *buf;
+        char *buf;
         unsigned long size;
         struct commit *commit;
@@ -68,4 +68,10 @@ void cgit_print_blob(const char *hex, char *path, const char *head)
         buf[size] = '\0';
         ctx.page.mimetype = ctx.qry.mimetype;
+        if (!ctx.page.mimetype) {
+                if (buffer_is_binary(buf, size))
+                        ctx.page.mimetype = "application/octet-stream";
+                else
+                        ctx.page.mimetype = "text/plain";
+        }
         ctx.page.filename = path;
         cgit_print_http_headers(&ctx);
diff --git a/ui-plain.c b/ui-plain.c
index 5addd9e..93a3a05 100644
--- a/ui-plain.c
+++ b/ui-plain.c
@@ -32,7 +32,11 @@ static void print_object(const unsigned char *sha1, const char *path)
                 return;
         }
+        if (buffer_is_binary(buf, size))
+                ctx.page.mimetype = "application/octet-stream";
+        else
         ctx.page.mimetype = "text/plain";
         ctx.page.filename = fmt("%s", path);
         ctx.page.size = size;
+        ctx.page.etag = sha1_to_hex(sha1);
         cgit_print_http_headers(&ctx);
         html_raw(buf, size);
diff --git a/ui-shared.c b/ui-shared.c
index fea2c40..66d5b82 100644
--- a/ui-shared.c
+++ b/ui-shared.c
@@ -35,4 +35,15 @@ void cgit_print_error(char *msg)
 }
 
+char *cgit_httpscheme()
+{
+        char *https;
+
+        https = getenv("HTTPS");
+        if (https != NULL && strcmp(https, "on") == 0)
+                return "https://";
+        else
+                return "http://";
+}
+
 char *cgit_hosturl()
 {
@@ -457,4 +468,8 @@ void cgit_print_age(time_t t, time_t max_relative, char *format)
 void cgit_print_http_headers(struct cgit_context *ctx)
 {
+        const char *method = getenv("REQUEST_METHOD");
+
+        if (ctx->page.status)
+                htmlf("Status: %d %s\n", ctx->page.status, ctx->page.statusmsg);
         if (ctx->page.mimetype && ctx->page.charset)
                 htmlf("Content-Type: %s; charset=%s\n", ctx->page.mimetype,
@@ -469,5 +484,9 @@ void cgit_print_http_headers(struct cgit_context *ctx)
         htmlf("Last-Modified: %s\n", http_date(ctx->page.modified));
         htmlf("Expires: %s\n", http_date(ctx->page.expires));
+        if (ctx->page.etag)
+                htmlf("ETag: \"%s\"\n", ctx->page.etag);
         html("\n");
+        if (method && !strcmp(method, "HEAD"))
+                exit(0);
 }
 
@@ -493,5 +512,6 @@ void cgit_print_docstart(struct cgit_context *ctx)
         }
         if (host && ctx->repo) {
-                html("<link rel='alternate' title='Atom feed' href='http://");
+                html("<link rel='alternate' title='Atom feed' href='");
+                html(cgit_httpscheme());
                 html_attr(cgit_hosturl());
                 html_attr(cgit_fileurl(ctx->repo->url, "atom", ctx->qry.path,
diff --git a/ui-shared.h b/ui-shared.h
index 5a3821f..bff4826 100644
--- a/ui-shared.h
+++ b/ui-shared.h
@@ -2,4 +2,5 @@
 #define UI_SHARED_H
 
+extern char *cgit_httpscheme();
 extern char *cgit_hosturl();
 extern char *cgit_repourl(const char *reponame);