#include <stdio.h>

#include <string.h>

#include <errno.h>

#include <unistd.h>

#include <task.h>

#include <stdlib.h>

#include <sys/types.h> /* See NOTES */

#include <sys/socket.h>

enum

{

STACK = 32768

};

char *server;

char *url;

static int sucess,fail;

void fetchtask(void*);

void

taskmain(int argc, char **argv)

{

int i, n;

if(argc != 4){

fprintf(stderr, "usage: httpload n server url\n");

taskexitall(1);

}

n = atoi(argv[1]);

server = argv[2];

url = argv[3];

for(i=0; i<n; i++){

taskcreate(fetchtask, 0, STACK);

//while(taskyield() > 1)

// ;

//taskdelay(500);

}

for(;;){

taskdelay(1000);

fprintf(stderr,"sucess:%d, fail:%d,rate:%f%%\n",sucess,fail,((float)sucess)/(sucess+fail)*100);

}

}

void

fetchtask(void *v)

{

int fd, n;

char buf[512];

struct timeval stTv;

//fprintf(stderr, "starting...\n");

for(;;){

if((fd = netdial(TCP, server, 80)) < 0){

fprintf(stderr, "dial %s: %s (%s)\n", server, strerror(errno), taskgetstate());

continue;

}

stTv.tv_sec = 60;

stTv.tv_usec = 0;

setsockopt(fd, SOL_SOCKET, SO_RCVTIMEO, (const char*)&stTv, sizeof(stTv));

snprintf(buf, sizeof buf, "GET %s HTTP/1.0\r\nHost: %s\r\n\r\n", url, server);

fdwrite(fd, buf, strlen(buf));

while((n = fdread(fd, buf, sizeof buf)) > 0)

;

if (n < 0)

{

fail++;

}else{

sucess++;

}

close(fd);

}

}