#include <u.h>#include <libc.h>#include <bio.h>#include "xml.h" static char *escmap[] ={	"\x06\"&quot;",	"\x06\'&apos;",	"\x04<&lt;",	"\x04>&gt;",	"\x05&&amp;",}; enum{	Xmlvalue = 2,}; static char*unxml(char *orig){	char *s;	Rune r;	int i, n, rsz; 	n = 0;	for(s = orig; *s != 0; s += rsz, n += rsz){		if(*s == '\r'){			if(s[1] == '\n'){				n--;				rsz = 1;				continue;			}			*s = '\n';		} 		rsz = chartorune(&r, s);		memmove(orig+n, s, rsz);		if(*s == '&'){			for(i = 0; i < nelem(escmap); i++){				if(strncmp(s, &escmap[i][2], escmap[i][0]) == 0){					orig[n] = escmap[i][1];					s += escmap[i][0] - 1;					break;				}			}		}	} 	orig[n] = 0;	return orig;} static voidxmlprint_(Xelem *x, int fd, int off){	Xattr *a; 	for(; x != nil; x = x->next){		fprint(fd, "%*c%q", off, ' ', x->n);		if(x->v != nil)			fprint(fd, "=%q", x->v);		for(a = x->a; a != nil; a = a->next)			fprint(fd, " %q=%q", a->n, a->v);		fprint(fd, "\n");		off += 4;		xmlprint_(x->ch, fd, off);		off -= 4;	}} static Xattr*xmlattr(char *s, int *err){	Xattr *a, *attrs;	char *p; 	attrs = nil; 	for(; *s;){		a = mallocz(sizeof(*a), 1);		a->n = s;		for(; *s && *s != '='; s++);		if(*s != '='){			werrstr("xml sucks (%d)", *s);			goto error;		}		*s++ = 0;		if(*s != '\'' && *s != '\"'){			werrstr("xml is complicated (%d)", *s);			goto error;		}		a->v = s+1;		s = utfrune(a->v, *s);		if(s == nil){			werrstr("xml is broken");			goto error;		}		*s++ = 0;		a->next = attrs;		a->n = unxml(a->n);		a->v = unxml(a->v);		attrs = a;		if(*s == ' ')			s++;		if((p = strchr(a->n, ':')) != nil && strncmp(p, ":zdef", 5) == 0)			*p = 0;	} 	return attrs;error:	*err = 1;	free(a);	for(; attrs != nil; attrs = a){		a = attrs->next;		free(attrs);	}	return nil;} static Xelem*xmlread_(Biobufhdr *h, Xelem *par, int flags, int *err){	char *s, *t;	Xelem *x, *ch;	int r, closed, len; 	x = nil; 	for(;;){		r = Bgetrune(h);		if(r < 0){			werrstr("xmlread: %r");			goto error;		}		if(r == '<')			break;		if(isspacerune(r))			continue;		if(flags & Xmlvalue && par != nil){			Bungetrune(h);			if((s = Brdstr(h, '<', 1)) == nil){				werrstr("xmlread: %r");				goto error;			}			par->v = unxml(s);			if((s = Brdstr(h, '>', 1)) == nil){				free(par->v);				par->v = nil;				werrstr("xmlread: %r");			}			free(s);			return nil;		}		werrstr("xmlread: unexpected rune (%C)", r);		goto error;	} 	s = Brdstr(h, '>', 1);	if(s == nil){		werrstr("xmlread: %r");		goto error;	}	if(s[0] == '/'){		free(s);		return nil;	}	if(s[0] == '?'){		free(s);		return xmlread_(h, par, flags, err);	} 	x = mallocz(sizeof(*x), 1);	x->priv = s;	x->n = s; 	if(strncmp(x->n, "zdef", 4) == 0){		if((x->n = strchr(x->n, ':')) == nil){			werrstr("xmlread: zdef without ':'");			goto error;		}		x->n += 1;	} 	len = strlen(s);	if(s[len-1] == '/' || s[len-1] == '?'){		closed = 1;		s[len-1] = 0;	}else		closed = flags & Xmlstartonly; 	for(; *s && *s != ' '; s++);	if(*s){		*s++ = 0;		x->a = xmlattr(s, err);	} 	if(strcmp(x->n, "html") == 0){		for(len = 0;; len += r){			s = Brdstr(h, '>', 0);			if(s == nil){				werrstr("xmlread: %r");				goto error;			} 			r = strlen(s);			x->v = realloc(x->v, len + r + 1);			if(x->v == nil){				werrstr("xmlread: %r");				goto error;			}			strcpy(x->v+len, s);			free(s);			t = strstr(x->v+len, "</html>");			if(t != nil){				*t = 0;				return x;			}		}	} 	if(!closed){		for(;;){			flags = Xmlvalue;			ch = xmlread_(h, x, flags, err);			if(ch == nil)				break;			ch->next = x->ch;			x->ch = ch;		}	} 	if(!*err)		return x; error:	*err = 2;	xmlfree(x);	return nil;} Xelem*xmlread(Biobuf *b, int flags, int *err){	*err = 0;	return xmlread_(b, nil, flags & Xmlstartonly, err);} voidxmlfree(Xelem *x){	Xattr *a, *ta;	Xelem *n, *n2; 	if(x == nil)		return; 	xmlfree(x->ch);	free(x->v);	x->ch = nil;	x->v = nil;	free(x->priv);	for(a = x->a; a != nil; a = ta){		ta = a->next;		free(a);	} 	for(n = x->next; n != nil; n = n2){		n2 = n->next;		n->next = nil;		xmlfree(n);	} 	free(x);} Xelem*xmlget(Xelem *x, char *name){	for(; x != nil; x = x->next)		if(strcmp(x->n, name) == 0)			return x;	return nil;} Xattr*xmlgetattr(Xattr *a, char *name){	for(; a != nil; a = a->next)		if(strcmp(a->n, name) == 0)			return a;	return nil;} voidxmlprint(Xelem *x, int fd){	xmlprint_(x, fd, 0);} intxmlstrfmt(Fmt *f){	char *s, *orig, *new;	int i, sz, n; 	orig = va_arg(f->args, char*);	for(s = orig; *s; s++){		for(i = 0; i < nelem(escmap); i++){			if(escmap[i][1] == *s)				goto escape;		}	}	return fmtprint(f, "%s", orig); escape:	n = s-orig;	sz = n*2 + 1;	new = malloc(sz);	memcpy(new, orig, n); 	for(; *s; s++, n++){		if(sz <= n+6){			sz = (n+6)*2;			new = realloc(new, sz);		}		new[n] = *s; 		for(i = 0; i < nelem(escmap); i++){			if(escmap[i][1] == *s){				memcpy(new+n, &escmap[i][2], escmap[i][0]);				n += escmap[i][0] - 1;				break;			}		}	} 	new[n] = 0;	n = fmtprint(f, "%s", new);	free(new);	return n;}