common/httest.c

   1 /*
   2  *  Dolda Connect - Modular multiuser Direct Connect-style client
   3  *  Copyright (C) 2007 Fredrik Tolf <fredrik@dolda2000.com>
   4  *
   5  *  This program is free software; you can redistribute it and/or modify
   6  *  it under the terms of the GNU General Public License as published by
   7  *  the Free Software Foundation; either version 2 of the License, or
   8  *  (at your option) any later version.
   9  *
  10  *  This program is distributed in the hope that it will be useful,
  11  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13  *  GNU General Public License for more details.
  14  *
  15  *  You should have received a copy of the GNU General Public License
  16  *  along with this program; if not, write to the Free Software
  17  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18 */
  19
  20 #include <stdlib.h>
  21 #include <stdio.h>
  22 #include <unistd.h>
  23 #include <string.h>
  24 #include <sys/poll.h>
  25 #include <errno.h>
  26
  27 #include <utils.h>
  28 #include <http.h>
  29
  30 void parse(char *url)
  31 {
  32     struct hturlinfo *u;
  33
  34     if((u = parseurl(url)) == NULL) {
  35         fprintf(stderr, "httest: %s: invalid url\n", url);
  36         return;
  37     }
  38     printf("host: %s\n", u->host);
  39     printf("port: %i\n", u->port);
  40     printf("path: %s\n", u->path);
  41     printf("query: %s\n", u->query);
  42     printf("fragment: %s\n", u->fragment);
  43     freeurl(u);
  44 }
  45
  46 void head(char *url)
  47 {
  48     struct hturlinfo *u;
  49     struct htconn *c;
  50     struct pollfd pfd;
  51     struct strpair *p;
  52     int ret;
  53
  54     if((u = parseurl(url)) == NULL) {
  55         fprintf(stderr, "httest: %s: invalid url\n", url);
  56         return;
  57     }
  58     c = htconnect(u);
  59     freeurl(u);
  60     while(1) {
  61         pfd.fd = c->fd;
  62         pfd.events = htpollflags(c);
  63         if(poll(&pfd, 1, -1) < 0) {
  64             fprintf(stderr, "httest: %s: %s\n", url, strerror(errno));
  65             freehtconn(c);
  66             return;
  67         }
  68         if((ret = htprocess(c, pfd.revents)) < 0) {
  69             fprintf(stderr, "httest: %s: %s\n", url, strerror(errno));
  70             freehtconn(c);
  71             return;
  72         }
  73         c->databufdata = 0;
  74         if(ret)
  75             break;
  76     }
  77     printf("%i %s\n", c->rescode, c->resstr);
  78     for(p = c->headers; p != NULL; p = p->next)
  79         printf("%s: %s\n", p->key, p->val);
  80     freehtconn(c);
  81 }
  82
  83 void get(char *url)
  84 {
  85     struct hturlinfo *u;
  86     struct htconn *c;
  87     struct pollfd pfd;
  88     struct strpair *p;
  89     int ret, ret2;
  90
  91     if((u = parseurl(url)) == NULL) {
  92         fprintf(stderr, "httest: %s: invalid url\n", url);
  93         return;
  94     }
  95     c = htconnect(u);
  96     freeurl(u);
  97     while(1) {
  98         pfd.fd = c->fd;
  99         pfd.events = htpollflags(c);
 100         if(poll(&pfd, 1, -1) < 0) {
 101             fprintf(stderr, "httest: %s: %s\n", url, strerror(errno));
 102             freehtconn(c);
 103             return;
 104         }
 105         if((ret = htprocess(c, pfd.revents)) < 0) {
 106             fprintf(stderr, "httest: %s: %s\n", url, strerror(errno));
 107             freehtconn(c);
 108             return;
 109         }
 110         while(c->databufdata > 0) {
 111             ret2 = write(1, c->databuf, c->databufdata);
 112             memmove(c->databuf, c->databuf + ret2, c->databufdata -= ret2);
 113         }
 114         if(ret)
 115             break;
 116     }
 117     printf("%i %s\n", c->rescode, c->resstr);
 118     for(p = c->headers; p != NULL; p = p->next)
 119         printf("%s: %s\n", p->key, p->val);
 120     freehtconn(c);
 121 }
 122
 123 int main(int argc, char **argv) {
 124     int i;
 125
 126     for(i = 1; i < argc; i++) {
 127         if(!strcmp(argv[i], "parse")) {
 128             parse(argv[++i]);
 129         } else if(!strcmp(argv[i], "head")) {
 130             head(argv[++i]);
 131         } else if(!strcmp(argv[i], "get")) {
 132             get(argv[++i]);
 133         }
 134     }
 135     return(0);
 136 }