mirror of https://github.com/0intro/libtask
61 lines
952 B
C
61 lines
952 B
C
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <errno.h>
|
|
#include <unistd.h>
|
|
#include <task.h>
|
|
#include <stdlib.h>
|
|
|
|
enum
|
|
{
|
|
STACK = 32768
|
|
};
|
|
|
|
char *server;
|
|
char *url;
|
|
|
|
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)
|
|
;
|
|
sleep(1);
|
|
}
|
|
}
|
|
|
|
void
|
|
fetchtask(void *v)
|
|
{
|
|
int fd, n;
|
|
char buf[512];
|
|
|
|
(void)v;
|
|
|
|
fprintf(stderr, "starting...\n");
|
|
for(;;){
|
|
if((fd = netdial(TCP, server, 80)) < 0){
|
|
fprintf(stderr, "dial %s: %s (%s)\n", server, strerror(errno), taskgetstate());
|
|
continue;
|
|
}
|
|
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)
|
|
;
|
|
close(fd);
|
|
write(1, ".", 1);
|
|
}
|
|
}
|