1 /*
2 * Canonical XML implementation test program
3 * (http://www.w3.org/TR/2001/REC-xml-c14n-20010315)
4 *
5 * See Copyright for the status of this software.
6 *
7 * Author: Aleksey Sanin <aleksey@aleksey.com>
8 */
9 #include "libxml.h"
10 #if defined(LIBXML_C14N_ENABLED) && defined(LIBXML_OUTPUT_ENABLED)
11
12 #include <stdio.h>
13 #include <string.h>
14 #ifndef STDOUT_FILENO
15 #ifdef HAVE_UNISTD_H
16 #include <unistd.h>
17 #else
18 #define STDOUT_FILENO fileno(stdout)
19 #endif /* HAVE_UNISTD_H */
20 #endif
21 #ifdef HAVE_STDLIB_H
22 #include <stdlib.h>
23 #endif
24
25 #include <libxml/xmlmemory.h>
26 #include <libxml/parser.h>
27 #include <libxml/xpath.h>
28 #include <libxml/xpathInternals.h>
29
30 #include <libxml/c14n.h>
31
32
usage(const char * name)33 static void usage(const char *name) {
34 fprintf(stderr,
35 "Usage: %s <mode> <xml-file> [<xpath-expr>] [<inclusive-ns-list>]\n",
36 name);
37 fprintf(stderr, "where <mode> is one of following:\n");
38 fprintf(stderr,
39 "--with-comments \t XML file canonicalization v1.0 w comments \n");
40 fprintf(stderr,
41 "--without-comments \t XML file canonicalization v1.0 w/o comments\n");
42 fprintf(stderr,
43 "--1-1-with-comments \t XML file canonicalization v1.1 w comments\n");
44 fprintf(stderr,
45 "--1-1-without-comments \t XML file canonicalization v1.1 w/o comments\n");
46 fprintf(stderr,
47 "--exc-with-comments \t Exclusive XML file canonicalization v1.0 w comments\n");
48 fprintf(stderr,
49 "--exc-without-comments\t Exclusive XML file canonicalization v1.0 w/o comments\n");
50 }
51
52 static xmlXPathObjectPtr
53 load_xpath_expr (xmlDocPtr parent_doc, const char* filename);
54
55 static xmlChar **parse_list(xmlChar *str);
56
57 /* static void print_xpath_nodes(xmlNodeSetPtr nodes); */
58
59 static int
test_c14n(const char * xml_filename,int with_comments,int mode,const char * xpath_filename,xmlChar ** inclusive_namespaces)60 test_c14n(const char* xml_filename, int with_comments, int mode,
61 const char* xpath_filename, xmlChar **inclusive_namespaces) {
62 xmlDocPtr doc;
63 xmlXPathObjectPtr xpath = NULL;
64 xmlChar *result = NULL;
65 int ret;
66
67 /*
68 * build an XML tree from a the file; we need to add default
69 * attributes and resolve all character and entities references
70 */
71 xmlLoadExtDtdDefaultValue = XML_DETECT_IDS | XML_COMPLETE_ATTRS;
72 xmlSubstituteEntitiesDefault(1);
73
74 doc = xmlReadFile(xml_filename, NULL, XML_PARSE_DTDATTR | XML_PARSE_NOENT);
75 if (doc == NULL) {
76 fprintf(stderr, "Error: unable to parse file \"%s\"\n", xml_filename);
77 return(-1);
78 }
79
80 /*
81 * Check the document is of the right kind
82 */
83 if(xmlDocGetRootElement(doc) == NULL) {
84 fprintf(stderr,"Error: empty document for file \"%s\"\n", xml_filename);
85 xmlFreeDoc(doc);
86 return(-1);
87 }
88
89 /*
90 * load xpath file if specified
91 */
92 if(xpath_filename) {
93 xpath = load_xpath_expr(doc, xpath_filename);
94 if(xpath == NULL) {
95 fprintf(stderr,"Error: unable to evaluate xpath expression\n");
96 xmlFreeDoc(doc);
97 return(-1);
98 }
99 }
100
101 /*
102 * Canonical form
103 */
104 /* fprintf(stderr,"File \"%s\" loaded: start canonization\n", xml_filename); */
105 ret = xmlC14NDocDumpMemory(doc,
106 (xpath) ? xpath->nodesetval : NULL,
107 mode, inclusive_namespaces,
108 with_comments, &result);
109 if(ret >= 0) {
110 if(result != NULL) {
111 if (write(STDOUT_FILENO, result, ret) == -1) {
112 fprintf(stderr, "Can't write data\n");
113 }
114 xmlFree(result);
115 }
116 } else {
117 fprintf(stderr,"Error: failed to canonicalize XML file \"%s\" (ret=%d)\n", xml_filename, ret);
118 if(result != NULL) xmlFree(result);
119 xmlFreeDoc(doc);
120 return(-1);
121 }
122
123 /*
124 * Cleanup
125 */
126 if(xpath != NULL) xmlXPathFreeObject(xpath);
127 xmlFreeDoc(doc);
128
129 return(ret);
130 }
131
main(int argc,char ** argv)132 int main(int argc, char **argv) {
133 int ret = -1;
134
135 /*
136 * Init libxml
137 */
138 xmlInitParser();
139 LIBXML_TEST_VERSION
140
141 /*
142 * Parse command line and process file
143 */
144 if( argc < 3 ) {
145 fprintf(stderr, "Error: wrong number of arguments.\n");
146 usage(argv[0]);
147 } else if(strcmp(argv[1], "--with-comments") == 0) {
148 ret = test_c14n(argv[2], 1, XML_C14N_1_0, (argc > 3) ? argv[3] : NULL, NULL);
149 } else if(strcmp(argv[1], "--without-comments") == 0) {
150 ret = test_c14n(argv[2], 0, XML_C14N_1_0, (argc > 3) ? argv[3] : NULL, NULL);
151 } else if(strcmp(argv[1], "--1-1-with-comments") == 0) {
152 ret = test_c14n(argv[2], 1, XML_C14N_1_1, (argc > 3) ? argv[3] : NULL, NULL);
153 } else if(strcmp(argv[1], "--1-1-without-comments") == 0) {
154 ret = test_c14n(argv[2], 0, XML_C14N_1_1, (argc > 3) ? argv[3] : NULL, NULL);
155 } else if(strcmp(argv[1], "--exc-with-comments") == 0) {
156 xmlChar **list;
157
158 /* load exclusive namespace from command line */
159 list = (argc > 4) ? parse_list((xmlChar *)argv[4]) : NULL;
160 ret = test_c14n(argv[2], 1, XML_C14N_EXCLUSIVE_1_0, (argc > 3) ? argv[3] : NULL, list);
161 if(list != NULL) xmlFree(list);
162 } else if(strcmp(argv[1], "--exc-without-comments") == 0) {
163 xmlChar **list;
164
165 /* load exclusive namespace from command line */
166 list = (argc > 4) ? parse_list((xmlChar *)argv[4]) : NULL;
167 ret = test_c14n(argv[2], 0, XML_C14N_EXCLUSIVE_1_0, (argc > 3) ? argv[3] : NULL, list);
168 if(list != NULL) xmlFree(list);
169 } else {
170 fprintf(stderr, "Error: bad option.\n");
171 usage(argv[0]);
172 }
173
174 /*
175 * Shutdown libxml
176 */
177 xmlCleanupParser();
178 xmlMemoryDump();
179
180 return((ret >= 0) ? 0 : 1);
181 }
182
183 /*
184 * Macro used to grow the current buffer.
185 */
186 #define growBufferReentrant() { \
187 buffer_size *= 2; \
188 buffer = (xmlChar **) \
189 xmlRealloc(buffer, buffer_size * sizeof(xmlChar*)); \
190 if (buffer == NULL) { \
191 perror("realloc failed"); \
192 return(NULL); \
193 } \
194 }
195
196 static xmlChar **
parse_list(xmlChar * str)197 parse_list(xmlChar *str) {
198 xmlChar **buffer;
199 xmlChar **out = NULL;
200 int buffer_size = 0;
201 int len;
202
203 if(str == NULL) {
204 return(NULL);
205 }
206
207 len = xmlStrlen(str);
208 if((str[0] == '\'') && (str[len - 1] == '\'')) {
209 str[len - 1] = '\0';
210 str++;
211 }
212 /*
213 * allocate an translation buffer.
214 */
215 buffer_size = 1000;
216 buffer = (xmlChar **) xmlMalloc(buffer_size * sizeof(xmlChar*));
217 if (buffer == NULL) {
218 perror("malloc failed");
219 return(NULL);
220 }
221 out = buffer;
222
223 while(*str != '\0') {
224 if (out - buffer > buffer_size - 10) {
225 int indx = out - buffer;
226
227 growBufferReentrant();
228 out = &buffer[indx];
229 }
230 (*out++) = str;
231 while(*str != ',' && *str != '\0') ++str;
232 if(*str == ',') *(str++) = '\0';
233 }
234 (*out) = NULL;
235 return buffer;
236 }
237
238 static xmlXPathObjectPtr
load_xpath_expr(xmlDocPtr parent_doc,const char * filename)239 load_xpath_expr (xmlDocPtr parent_doc, const char* filename) {
240 xmlXPathObjectPtr xpath;
241 xmlDocPtr doc;
242 xmlChar *expr;
243 xmlXPathContextPtr ctx;
244 xmlNodePtr node;
245 xmlNsPtr ns;
246
247 /*
248 * load XPath expr as a file
249 */
250 xmlLoadExtDtdDefaultValue = XML_DETECT_IDS | XML_COMPLETE_ATTRS;
251 xmlSubstituteEntitiesDefault(1);
252
253 doc = xmlReadFile(filename, NULL, XML_PARSE_DTDATTR | XML_PARSE_NOENT);
254 if (doc == NULL) {
255 fprintf(stderr, "Error: unable to parse file \"%s\"\n", filename);
256 return(NULL);
257 }
258
259 /*
260 * Check the document is of the right kind
261 */
262 if(xmlDocGetRootElement(doc) == NULL) {
263 fprintf(stderr,"Error: empty document for file \"%s\"\n", filename);
264 xmlFreeDoc(doc);
265 return(NULL);
266 }
267
268 node = doc->children;
269 while(node != NULL && !xmlStrEqual(node->name, (const xmlChar *)"XPath")) {
270 node = node->next;
271 }
272
273 if(node == NULL) {
274 fprintf(stderr,"Error: XPath element expected in the file \"%s\"\n", filename);
275 xmlFreeDoc(doc);
276 return(NULL);
277 }
278
279 expr = xmlNodeGetContent(node);
280 if(expr == NULL) {
281 fprintf(stderr,"Error: XPath content element is NULL \"%s\"\n", filename);
282 xmlFreeDoc(doc);
283 return(NULL);
284 }
285
286 ctx = xmlXPathNewContext(parent_doc);
287 if(ctx == NULL) {
288 fprintf(stderr,"Error: unable to create new context\n");
289 xmlFree(expr);
290 xmlFreeDoc(doc);
291 return(NULL);
292 }
293
294 /*
295 * Register namespaces
296 */
297 ns = node->nsDef;
298 while(ns != NULL) {
299 if(xmlXPathRegisterNs(ctx, ns->prefix, ns->href) != 0) {
300 fprintf(stderr,"Error: unable to register NS with prefix=\"%s\" and href=\"%s\"\n", ns->prefix, ns->href);
301 xmlFree(expr);
302 xmlXPathFreeContext(ctx);
303 xmlFreeDoc(doc);
304 return(NULL);
305 }
306 ns = ns->next;
307 }
308
309 /*
310 * Evaluate xpath
311 */
312 xpath = xmlXPathEvalExpression(expr, ctx);
313 if(xpath == NULL) {
314 fprintf(stderr,"Error: unable to evaluate xpath expression\n");
315 xmlFree(expr);
316 xmlXPathFreeContext(ctx);
317 xmlFreeDoc(doc);
318 return(NULL);
319 }
320
321 /* print_xpath_nodes(xpath->nodesetval); */
322
323 xmlFree(expr);
324 xmlXPathFreeContext(ctx);
325 xmlFreeDoc(doc);
326 return(xpath);
327 }
328
329 /*
330 static void
331 print_xpath_nodes(xmlNodeSetPtr nodes) {
332 xmlNodePtr cur;
333 int i;
334
335 if(nodes == NULL ){
336 fprintf(stderr, "Error: no nodes set defined\n");
337 return;
338 }
339
340 fprintf(stderr, "Nodes Set:\n-----\n");
341 for(i = 0; i < nodes->nodeNr; ++i) {
342 if(nodes->nodeTab[i]->type == XML_NAMESPACE_DECL) {
343 xmlNsPtr ns;
344
345 ns = (xmlNsPtr)nodes->nodeTab[i];
346 cur = (xmlNodePtr)ns->next;
347 fprintf(stderr, "namespace \"%s\"=\"%s\" for node %s:%s\n",
348 ns->prefix, ns->href,
349 (cur->ns) ? cur->ns->prefix : BAD_CAST "", cur->name);
350 } else if(nodes->nodeTab[i]->type == XML_ELEMENT_NODE) {
351 cur = nodes->nodeTab[i];
352 fprintf(stderr, "element node \"%s:%s\"\n",
353 (cur->ns) ? cur->ns->prefix : BAD_CAST "", cur->name);
354 } else {
355 cur = nodes->nodeTab[i];
356 fprintf(stderr, "node \"%s\": type %d\n", cur->name, cur->type);
357 }
358 }
359 }
360 */
361
362 #else
363 #include <stdio.h>
main(int argc ATTRIBUTE_UNUSED,char ** argv ATTRIBUTE_UNUSED)364 int main(int argc ATTRIBUTE_UNUSED, char **argv ATTRIBUTE_UNUSED) {
365 printf("%s : XPath/Canonicalization and output support not compiled in\n", argv[0]);
366 return(0);
367 }
368 #endif /* LIBXML_C14N_ENABLED */
369
370
371