mothra: changes toward multipart/formdata

1 files changed, 150 insertions, 0 deletions

diff --git a/sys/src/cmd/mothra/url.c b/sys/src/cmd/mothra/url.c
new file mode 100644
index 000000000..9773f15e9
--- /dev/null
+++ b/sys/src/cmd/mothra/url.c
@@ -0,0 +1,150 @@
+#include <u.h>
+#include <libc.h>
+#include <draw.h>
+#include <event.h>
+#include <panel.h>
+#include "mothra.h"
+
+static int
+fileget(Url *url)
+{
+	char *rel, *base, *x;
+
+	rel = base = nil;
+	if(cistrncmp(url->basename, "file:", 5) == 0)
+		base = url->basename+5;
+	if(cistrncmp(url->reltext, "file:", 5) == 0)
+		rel = url->reltext+5;
+	if(rel == nil && base == nil)
+		return -1;
+	if(rel == nil)
+		rel = url->reltext;
+	if(base && base[0] == '/' && rel[0] != '/'){
+		if(x = strrchr(base, '/'))
+			*x = 0;
+		snprint(url->fullname, sizeof(url->fullname), "%s/%s", base, rel);
+		if(x)	*x = '/';
+	}else
+		snprint(url->fullname, sizeof(url->fullname), "%s", rel);
+	url->tag[0] = 0;
+	if(x = strrchr(url->fullname, '#')){
+		*x++ = 0;
+		nstrcpy(url->tag, x, sizeof(url->tag));
+	}
+	base = cleanname(url->fullname);
+	x = base + strlen(base)+1;
+	if((x - base) > sizeof(url->fullname)-5)
+		return -1;
+	memmove(url->fullname+5, base, x - base);
+	memmove(url->fullname, "file:", 5);
+	return open(url->fullname+5, OREAD);
+}
+
+char *mtpt="/mnt/web";
+
+static int
+webclone(Url *url, char *buf, int nbuf)
+{
+	int n, fd;
+
+	snprint(buf, nbuf, "%s/clone", mtpt);
+	if((fd = open(buf, ORDWR)) < 0)
+		return -1;
+	if((n = read(fd, buf, nbuf-1)) <= 0){
+		close(fd);
+		return -1;
+	}
+	buf[n] = 0;
+	n = atoi(buf);
+	snprint(buf, nbuf, "%s/%d", mtpt, n);
+	if(url && url->reltext[0]){
+		if(url->basename[0])
+			fprint(fd, "baseurl %s", url->basename);
+		if(fprint(fd, "url %s", url->reltext) < 0){
+			close(fd);
+			return -1;
+		}
+	}
+	return fd;
+}
+
+static int
+readstr(char *path, char *buf, int nbuf){
+	int n, fd;
+
+	n = 0;
+	if((fd = open(path, OREAD)) >= 0){
+		if((n = read(fd, buf, nbuf-1)) < 0)
+			n = 0;
+		close(fd);
+	}
+	buf[n] = 0;
+	return n;
+}
+
+int
+urlpost(Url *url, char *ctype)
+{
+	char buf[1024];
+	int n, fd;
+
+	if((fd = webclone(url, buf, sizeof(buf))) < 0)
+		return -1;
+	if(ctype && *ctype)
+		fprint(fd, "contenttype %s", ctype);
+	n = strlen(buf);
+	snprint(buf+n, sizeof(buf)-n, "/postbody");
+	n = open(buf, OWRITE);
+	close(fd);
+	return n;
+}
+
+int
+urlget(Url *url, int body)
+{
+	char buf[1024];
+	int n, fd;
+
+	if(body < 0){
+		if((fd = fileget(url)) >= 0)
+			return fd;
+		if((fd = webclone(url, buf, sizeof(buf))) < 0)
+			return -1;
+	}else{
+		char *x;
+
+		if(fd2path(body, buf, sizeof(buf))){
+			close(body);
+			return -1;
+		}
+		if(x = strrchr(buf, '/'))
+			*x = 0;
+		fd = open(buf, OREAD);
+		close(body);
+	}
+	n = strlen(buf);
+	snprint(buf+n, sizeof(buf)-n, "/body");
+	body = open(buf, OREAD);
+	close(fd);
+	fd = body;
+	if(fd < 0)
+		return -1;
+
+	snprint(buf+n, sizeof(buf)-n, "/parsed/url");
+	readstr(buf, url->fullname, sizeof(url->fullname));
+
+	snprint(buf+n, sizeof(buf)-n, "/parsed/fragment");
+	readstr(buf, url->tag, sizeof(url->tag));
+
+	snprint(buf+n, sizeof(buf)-n, "/contentencoding");
+	readstr(buf, buf, sizeof(buf));
+
+	if(!cistrcmp(buf, "compress"))
+		fd = pipeline("/bin/uncompress", fd);
+	else if(!cistrcmp(buf, "gzip"))
+		fd = pipeline("/bin/gunzip", fd);
+	else if(!cistrcmp(buf, "bzip2"))
+		fd = pipeline("/bin/bunzip2", fd);
+
+	return fd;
+}