Rewrite using xmlparse

expat/xmlwf/xmlwf.c

@@ -1,65 +1,259 @@
-#include <stdio.h>
+#include "xmlparse.h"
-#include <string.h>
-#include "wfcheck.h"
 #include "filemap.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <fcntl.h>
+
+#ifdef _MSC_VER
+#include <io.h>
+#endif
+
+#ifndef O_BINARY
+#ifdef _O_BINARY
+#define O_BINARY _O_BINARY
+#else
+#define O_BINARY 0
+#endif
+#endif
+
 #ifdef _MSC_VER
 #include <crtdbg.h>
 #endif
 
-struct ProcessFileArg {
+#ifdef _DEBUG
-  enum EntityType entityType;
+#define READ_SIZE 16
-  int result;
+#else
-};
+#define READ_SIZE (1024*8)
+#endif
+
+static void characterData(void *userData, const char *s, size_t len)
+{
+  FILE *fp = userData;
+  for (; len > 0; --len, ++s) {
+    switch (*s) {
+    case '&':
+      fputs("&amp;", fp);
+      break;
+    case '<':
+      fputs("&lt;", fp);
+      break;
+    case '>':
+      fputs("&gt;", fp);
+      break;
+    case '"':
+      fputs("&quot;", fp);
+      break;
+    case '\r':
+      fputs("&#13;", fp);
+      break;
+    default:
+      putc(*s, fp);
+      break;
+    }
+  }
+}
+
+/* Lexicographically comparing UTF-8 encoded attribute values,
+is equivalent to lexicographically comparing based on the character number. */
+
+static int attcmp(const void *att1, const void *att2)
+{
+  return strcmp(*(const char **)att1, *(const char **)att2);
+}
+
+static void startElement(void *userData, const char *name, const char **atts)
+{
+  int nAtts;
+  const char **p;
+  FILE *fp = userData;
+  putc('<', fp);
+  fputs(name, fp);
+
+  p = atts;
+  while (*p)
+    ++p;
+  nAtts = (p - atts) >> 1;
+  if (nAtts > 1)
+    qsort((void *)atts, nAtts, sizeof(char *) * 2, attcmp);
+  while (*atts) {
+    putc(' ', fp);
+    fputs(*atts++, fp);
+    putc('=', fp);
+    putc('"', fp);
+    characterData(userData, *atts, strlen(*atts));
+    putc('"', fp);
+    atts++;
+  }
+  putc('>', fp);
+}
+
+static void endElement(void *userData, const char *name)
+{
+  FILE *fp = userData;
+  putc('<', fp);
+  putc('/', fp);
+  fputs(name, fp);
+  putc('>', fp);
+}
+
+static void processingInstruction(void *userData, const char *target, const char *data)
+{
+  FILE *fp = userData;
+  fprintf(fp, "<?%s %s?>", target, data);
+}
+
+typedef struct {
+  XML_Parser parser;
+  int *retPtr;
+} PROCESS_ARGS;
 
 static
-void processFile(const void *data, size_t size, const char *filename, void *p)
+void reportError(XML_Parser parser, const char *filename)
 {
-  const char *badPtr = 0;
+  int code = XML_GetErrorCode(parser);
-  unsigned long badLine = 0;
+  const char *message = XML_ErrorString(code);
-  unsigned long badCol = 0;
+  if (message)
-  struct ProcessFileArg *arg = p;
+    fprintf(stderr, "%s:%d:%ld: %s\n",
-  enum WfCheckResult result;
+	    filename,
+	    XML_GetErrorLineNumber(parser),
+	    XML_GetErrorColumnNumber(parser),
+	  message);
+  else
+    fprintf(stderr, "%s: (unknown message %d)\n", filename, code);
+}
 
-  result = wfCheck(arg->entityType, data, size, &badPtr, &badLine, &badCol);
+static
-  if (result) {
+void processFile(const void *data, size_t size, const char *filename, void *args)
-    const char *msg = wfCheckMessage(result);
+{
-    fprintf(stdout, "%s:", filename);
+  XML_Parser parser = ((PROCESS_ARGS *)args)->parser;
-    if (badPtr != 0)
+  int *retPtr = ((PROCESS_ARGS *)args)->retPtr;
-      fprintf(stdout, "%lu:%lu:", badLine+1, badCol);
+  if (!XML_Parse(parser, data, size, 1)) {
-    fprintf(stdout, "E: %s", msg ? msg : "(unknown message)");
+    reportError(parser, filename);
-    putc('\n', stdout);
+    *retPtr = 0;
-    arg->result = 1;
   }
   else
-    arg->result = 0;
+    *retPtr = 1;
+}
+
+static
+int processStream(const char *filename, XML_Parser parser)
+{
+  int fd = open(filename, O_BINARY|O_RDONLY);
+  if (fd < 0) {
+    perror(filename);
+    return 0;
+  }
+  for (;;) {
+    int nread;
+    char *buf = XML_GetBuffer(parser, READ_SIZE);
+    if (!buf) {
+      close(fd);
+      fprintf(stderr, "%s: out of memory\n", filename);
+      return 0;
+    }
+    nread = read(fd, buf, READ_SIZE);
+    if (nread < 0) {
+      perror(filename);
+      close(fd);
+      return 0;
+    }
+    if (!XML_ParseBuffer(parser, nread, nread == 0)) {
+      reportError(parser, filename);
+      close(fd);
+      return 0;
+    }
+    if (nread == 0) {
+      close(fd);
+      break;;
+    }
+  }
+  return 1;
+}
+
+static
+void usage(const char *prog)
+{
+  fprintf(stderr, "usage: %s [-d output-dir] file ...\n", prog);
+  exit(1);
 }
 
 int main(int argc, char **argv)
 {
-  int i = 1;
+  int i;
-  int ret = 0;
+  const char *outputDir = 0;
-  struct ProcessFileArg arg;
+  int useFilemap = 1;
 
 #ifdef _MSC_VER
   _CrtSetDbgFlag(_CRTDBG_ALLOC_MEM_DF|_CRTDBG_LEAK_CHECK_DF);
 #endif
-  arg.entityType = documentEntity;
 
-  if (i < argc && strcmp(argv[i], "-g") == 0) {
+  i = 1;
+  while(i < argc && argv[i][0] == '-') {
+    int j;
+    if (argv[i][1] == '-' && argv[i][2] == '\0') {
       i++;
-    arg.entityType = generalTextEntity;
+      break;
     }
-  if (i < argc && strcmp(argv[i], "--") == 0)
+    j = 1;
+    if (argv[i][j] == 'r') {
+      useFilemap = 0;
+      j++;
+    }
+    if (argv[i][j] == 'd') {
+      if (argv[i][j + 1] == '\0') {
+	if (++i == argc)
+	  usage(argv[0]);
+	outputDir = argv[i];
+      }
+      else
+	outputDir = argv[i] + j + 1;
       i++;
-  if (i == argc) {
-    fprintf(stderr, "usage: %s [-g] filename ...\n", argv[0]);
-    return 1;
     }
+    else if (argv[i][j] == '\0' && j > 1)
+      i++;
+    else
+      usage(argv[0]);
+  }
+  if (i == argc)
+    usage(argv[0]);
   for (; i < argc; i++) {
-    if (!filemap(argv[i], processFile, &arg))
+    FILE *fp = 0;
-      ret = 2;
+    char *outName = 0;
-    else if (arg.result && !ret)
+    int result;
-      ret = 1;
+    XML_Parser parser = XML_ParserCreate(0);
+    if (outputDir) {
+      outName = malloc(strlen(outputDir) + strlen(argv[i]) + 2);
+      strcpy(outName, outputDir);
+      strcat(outName, "/");
+      strcat(outName, argv[i]);
+      fp = fopen(outName, "wb");
+      if (!fp) {
+	perror(outName);
+	exit(1);
       }
-  return ret;
+      XML_SetUserData(parser, fp);
+      XML_SetElementHandler(parser, startElement, endElement);
+      XML_SetCharacterDataHandler(parser, characterData);
+      XML_SetProcessingInstructionHandler(parser, processingInstruction);
+    }
+    if (useFilemap) {
+      PROCESS_ARGS args;
+      args.retPtr = &result;
+      args.parser = parser;
+      if (!filemap(argv[i], processFile, &args))
+	result = 0;
+    }
+    else
+      result = processStream(argv[i], parser);
+    if (outputDir) {
+      fclose(fp);
+      if (!result)
+	remove(outName);
+      free(outName);
+    }
+    XML_ParserFree(parser);
+  }
+  return 0;
 }