1 files changed, 239 insertions, 0 deletions
diff --git a/dpi/vsource.c b/dpi/vsource.c
new file mode 100644
index 00000000..113aaa99
--- /dev/null
+++ b/dpi/vsource.c
@@ -0,0 +1,239 @@
+/*
+ * Dpi for "View source".
+ *
+ * This server is an example. Play with it and modify to your taste.
+ *
+ * Copyright 2010 Jorge Arellano Cid <jcid@dillo.org>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ */
+
+#include <unistd.h>
+#include <sys/types.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <errno.h>
+#include "../dpip/dpip.h"
+#include "dpiutil.h"
+
+/*
+ * Debugging macros
+ */
+#define _MSG(...)
+#define MSG(...)  printf("[vsource dpi]: " __VA_ARGS__)
+
+/*---------------------------------------------------------------------------*/
+
+const char *DOCTYPE=
+ "<!DOCTYPE HTML PUBLIC '-//W3C//DTD HTML 4.01 Transitional//EN'>";
+
+
+void send_dpip_tag(Dsh *sh, char *dpip_tag)
+{
+   a_Dpip_dsh_write_str(sh, 0, "\nDpip tag received: ");
+   a_Dpip_dsh_write_str(sh, 0, dpip_tag ? dpip_tag : "None");
+   a_Dpip_dsh_write_str(sh, 1, "\n\n");
+}
+
+/*
+ * Send source as plain text
+ */
+void send_plain_text(Dsh *sh, int data_size)
+{
+   int bytes_read = 0;
+   char *src_str;
+
+   /* Send HTTP header for plain text MIME type */
+   a_Dpip_dsh_write_str(sh, 0, "Content-type: text/plain\n\n");
+
+   while (bytes_read < data_size &&
+          (src_str = a_Dpip_dsh_read_token(sh, 1))) {
+      bytes_read += strlen(src_str);
+      a_Dpip_dsh_write_str(sh, 1, src_str);
+      dFree(src_str);
+   }
+}
+
+/*
+ * Send source as plain text with line numbers
+ */
+void send_numbered_text(Dsh *sh, int data_size)
+{
+   int bytes_read = 0, line = 1;
+   char *p, *q, *src_str, line_str[32];
+
+   /* Send HTTP header for plain text MIME type */
+   a_Dpip_dsh_write_str(sh, 0, "Content-type: text/plain\n\n");
+
+   while (bytes_read < data_size &&
+          (src_str = a_Dpip_dsh_read_token(sh, 1))) {
+      bytes_read += strlen(src_str);
+      p = q = src_str;
+
+      while (*p) {
+         snprintf(line_str, 32, "%2d: ", line);
+         a_Dpip_dsh_write_str(sh, 0, line_str);
+         if ((p = strchr(q, '\n'))) {
+            a_Dpip_dsh_write(sh, 0, q, p - q + 1);
+            if (p[1] == '\r')
+               ++p;
+            ++line;
+         } else {
+            a_Dpip_dsh_write_str(sh, 1, q);
+            break;
+         }
+         q = ++p;
+      }
+      dFree(src_str);
+   }
+}
+
+/*
+ * Send source as html text with line numbers
+ */
+void send_html_text(Dsh *sh, const char *url, int data_size)
+{
+   int bytes_read = 0, old_line = 0, line = 1;
+   char *p, *q, *src_str, line_str[128];
+
+   if (strncmp(url, "dpi:/vsource/:", 14) == 0)
+      url += 14;
+
+   /* Send HTTP header for plain text MIME type */
+   a_Dpip_dsh_write_str(sh, 0, "Content-type: text/html\n\n");
+
+   a_Dpip_dsh_write_str(sh, 0, DOCTYPE);
+   a_Dpip_dsh_printf(sh, 0,
+                     "\n"
+                     "<html><head>\n"
+                     "<title>Source for %s</title>\n"
+                     "<style type=\"text/css\">PRE {white-space: pre-wrap}\n"
+                     "</style>\n"
+                     "</head>\n"
+                     "<body id=\"dillo_vs\">\n<table cellpadding='0'>\n", url);
+
+   while (bytes_read < data_size &&
+          (src_str = a_Dpip_dsh_read_token(sh, 1))) {
+      bytes_read += strlen(src_str);
+      p = q = src_str;
+
+      while (*p) {
+         if (line > old_line) {
+            snprintf(line_str, 128,
+                     "%s<tr><td bgcolor='%s'>%d%s<td><pre>",
+                     (line > 1) ? "</pre>" : "",
+                     (line & 1) ? "#B87333" : "#DD7F32", line,
+                     (line == 1 || (line % 10) == 0) ? "&nbsp;&nbsp;" : "");
+            a_Dpip_dsh_write_str(sh, 0, line_str);
+            old_line = line;
+         }
+         if ((p = strpbrk(q, "\r\n<&"))) {
+            if (*p == '\r' || *p == '\n') {
+               a_Dpip_dsh_write(sh, 0, q, p - q + 1);
+               if (*p == '\r' && p[1] == '\n')
+                  p++;
+               ++line;
+            } else {
+               a_Dpip_dsh_write(sh, 0, q, p - q);
+               a_Dpip_dsh_write_str(sh, 0, (*p == '<') ? "&lt;" : "&amp;");
+            }
+         } else {
+            a_Dpip_dsh_write_str(sh, 1, q);
+            break;
+         }
+         q = ++p;
+      }
+      dFree(src_str);
+   }
+
+   if (data_size > 0)
+      a_Dpip_dsh_write_str(sh, 0, "</pre>");
+   a_Dpip_dsh_write_str(sh, 1, "</table></body></html>");
+}
+
+/*
+ *
+ */
+int main(void)
+{
+   Dsh *sh;
+   int data_size;
+   char *dpip_tag, *cmd = NULL, *cmd2 = NULL, *url = NULL, *size_str = NULL;
+   char *d_cmd;
+
+   _MSG("starting...\n");
+   //sleep(20);
+
+   /* Initialize the SockHandler.
+    * This means we'll use stdin for input and stdout for output.
+    * In case of a server dpi, we'd use a socket and pass its file descriptor
+    * twice (e.g. a_Dpip_dsh_new(sock_fd, sock_fd, 1024).
+    * (Note: by now the last parameter is not used) */
+   sh = a_Dpip_dsh_new(STDIN_FILENO, STDOUT_FILENO, 2*1024);
+
+   /* Authenticate our client...
+    * As we're using Internet domain sockets, DPIP checks whether the client
+    * runs with the user's ID, by means of a shared secret. The DPIP API does
+    * the work for us. */
+   if (!(dpip_tag = a_Dpip_dsh_read_token(sh, 1)) ||
+       a_Dpip_check_auth(dpip_tag) < 0) {
+      MSG("can't authenticate request: %s\n", dStrerror(errno));
+      a_Dpip_dsh_close(sh);
+      return 1;
+   }
+   dFree(dpip_tag);
+
+   /* Read the dpi command from STDIN
+    * Now we're past the authentication phase, let's see what's dillo
+    * asking from us. a_Dpip_dsh_read_token() will block and return
+    * a full dpip token or null on error (it's commented in dpip.c) */
+   dpip_tag = a_Dpip_dsh_read_token(sh, 1);
+   MSG("tag = [%s]\n", dpip_tag);
+
+   /* Now that we have the dpip_tag, let's isolate the command and url */
+   cmd = a_Dpip_get_attr(dpip_tag, "cmd");
+   url = a_Dpip_get_attr(dpip_tag, "url");
+
+   /* Start sending our answer.
+    * (You can read the comments for DPIP API functions in dpip/dpip.c) */
+   d_cmd = a_Dpip_build_cmd("cmd=%s url=%s", "start_send_page", url);
+   a_Dpip_dsh_write_str(sh, 0, d_cmd);
+   dFree(d_cmd);
+   dFree(dpip_tag);
+
+   dpip_tag = a_Dpip_dsh_read_token(sh, 1);
+   cmd2 = a_Dpip_get_attr(dpip_tag, "cmd");
+   if (cmd2) {
+      if (strcmp(cmd2, "start_send_page") == 0 &&
+          (size_str = a_Dpip_get_attr(dpip_tag, "data_size"))) {
+         data_size = strtol(size_str, NULL, 10);
+         /* Choose your flavour */
+         //send_plain_text(sh, data_size);
+         //send_numbered_text(sh, data_size);
+         send_html_text(sh, url, data_size);
+      } else if (strcmp(cmd2, "DpiError") == 0) {
+         /* Dillo detected an error (other failures just close the socket) */
+         a_Dpip_dsh_write_str(sh, 0, "Content-type: text/plain\n\n");
+         a_Dpip_dsh_write_str(sh, 1, "[vsource dpi]: "
+                                     "ERROR: Page not cached.\n");
+      }
+      dFree(cmd2);
+   }
+
+   dFree(cmd);
+   dFree(url);
+   dFree(size_str);
+   dFree(dpip_tag);
+
+   /* Finish the SockHandler */
+   a_Dpip_dsh_close(sh);
+   a_Dpip_dsh_free(sh);
+
+   return 0;
+}
+