Subversion Repositories planix.SVN

#include <u.h>
#include <libc.h>
#include <bio.h>
#include <thread.h>
#include <ctype.h>
#include <plumb.h>
#include "dat.h"


char*
formathtml(char *body, int *np)
{
        int i, j, p[2], q[2];
        Exec *e;
        char buf[1024];
        Channel *sync;

        e = emalloc(sizeof(struct Exec));
        if(pipe(p) < 0 || pipe(q) < 0)
                error("can't create pipe: %r");

        e->p[0] = p[0];
        e->p[1] = p[1];
        e->q[0] = q[0];
        e->q[1] = q[1];
        e->argv = emalloc(3*sizeof(char*));
        e->argv[0] = estrdup("htmlfmt");
        e->argv[1] = estrdup("-cutf-8");
        e->argv[2] = nil;
        e->prog = "/bin/htmlfmt";
        sync = chancreate(sizeof(int), 0);
        e->sync = sync;
        proccreate(execproc, e, EXECSTACK);
        recvul(sync);
        close(p[0]);
        close(q[1]);

        if((i=write(p[1], body, *np)) != *np){
                fprint(2, "Mail: warning: htmlfmt failed: wrote %d of %d: %r\n", i, *np);
                close(p[1]);
                close(q[0]);
                return body;
        }
        close(p[1]);

        free(body);
        body = nil;
        i = 0;
        for(;;){
                j = read(q[0], buf, sizeof buf);
                if(j <= 0)
                        break;
                body = realloc(body, i+j+1);
                if(body == nil)
                        error("realloc failed: %r");
                memmove(body+i, buf, j);
                i += j;
                body[i] = '\0';
        }
        close(q[0]);

        *np = i;
        return body;
}

char*
readbody(char *type, char *dir, int *np)
{
        char *body;
        
        body = readfile(dir, "body", np);
        if(body != nil && strcmp(type, "text/html") == 0)
                return formathtml(body, np);
        return body;
}