razzor/torrent-stats
1
0
Fork 0
Code Issues Pull Requests Releases Activity

added httpd

Browse Source
This commit is contained in:
Olaf Rempel 2007-06-18 15:44:05 +02:00
parent d169f4ab47
commit 503b837763
7 changed files with 441 additions and 201 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Makefile
View File

@ -1,6 +1,6 @@
CFLAGS := -O2 -pipe -Wall CFLAGS := -O2 -pipe -Wall
OBJS := configfile.o event.o linebuffer.o logging.o sockaddr.o tcpsocket.o OBJS := configfile.o connection.o event.o httpd.o linebuffer.o logging.o sockaddr.o tcpsocket.o
all: torrent-stats all: torrent-stats

155
connection.c Normal file
View File

@ -0,0 +1,155 @@
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include "event.h"
#include "httpd.h"
#include "linebuffer.h"
#include "list.h"
#include "logging.h"
#include "sockaddr.h"
#include "tcpsocket.h"
static LIST_HEAD(client_list);
struct client_con {
struct list_head list;
struct sockaddr_in addr;
struct event_timeout *trigger;
struct event_fd *event;
struct linebuffer *lbuf;
int bw_up;
int bw_dn;
unsigned long long total_up;
unsigned long long total_dn;
int chunk_total;
int chunk_avail;
int chunk_have;
};
static void free_client(struct client_con *con)
{
close(event_get_fd(con->event));
event_remove_fd(con->event);
event_remove_timeout(con->trigger);
free(con->lbuf);
free(con);
}
static int trigger_status(void *privdata)
{
struct client_con *con = (struct client_con *)privdata;
write(event_get_fd(con->event), "SENDSTATUS\n", 11);
return 0;
}
static int data_cb(int fd, void *privdata)
{
struct client_con *con = (struct client_con *)privdata;
if (linebuffer_read(con->lbuf, fd) < 0) {
log_print(LOG_WARN, "data_cb(): client %s read", get_sockaddr_buf(&con->addr));
list_del(&con->list);
free_client(con);
return -1;
}
char *line;
while ((line = linebuffer_get(con->lbuf)) != NULL) {
if (strncmp(line, "CTBW ", 5) == 0) {
int bwup, bwdn, liup, lidn;
if (sscanf(line +5, "%d,%d %d,%d", &bwdn, &bwup, &lidn, &liup) == 4) {
con->bw_up = bwup;
con->bw_dn = bwdn;
}
} else if (strncmp(line, "CTSTATUS ", 9) == 0) {
int seeds1 = 0, seeds2 = 0, leech1 = 0, leech2 = 0, count = 0;
int chunk1 = 0, chunk2 = 0, chunk3 = 0, bwdn = 0, bwup = 0;
int lidn = 0, liup = 0, cache = 0;
unsigned long long totdn = 0, totup = 0;
if (sscanf(line +9, "%d:%d/%d:%d/%d %d/%d/%d %d,%d %llu,%llu %d,%d %d",
&seeds1, &seeds2, &leech1, &leech2, &count,
&chunk1, &chunk2, &chunk3,
&bwdn, &bwup, &totdn, &totup,
&lidn, &liup, &cache) == 15) {
con->total_up = totup;
con->total_dn = totdn;
con->chunk_have = chunk1;
con->chunk_total = chunk2;
con->chunk_avail = chunk3;
}
}
free(line);
}
return 0;
}
int ctcs_accept_handler(int fd, void *privdata)
{
struct client_con *con = malloc(sizeof(struct client_con));
if (con == NULL) {
log_print(LOG_WARN, "accept_cb(): out of memory");
return 0;
}
memset(con, 0, sizeof(struct client_con));
con->lbuf = create_linebuffer(1024);
if (con->lbuf == NULL) {
log_print(LOG_WARN, "accept_cb(): out of memory");
free(con);
return 0;
}
unsigned int i = sizeof(con->addr);
int sockfd = accept(fd, (struct sockaddr *)&con->addr, &i);
if (sockfd < 0) {
log_print(LOG_WARN, "accept_cb(): accept()");
free(con->lbuf);
free(con);
return 0;
}
struct timeval tv = { .tv_sec = 10, .tv_usec = 0 };
con->trigger = event_add_timeout(&tv, trigger_status, con);
con->event = event_add_readfd(NULL, sockfd, data_cb, con);
list_add(&con->list, &client_list);
return 0;
}
int ctcs_httpd_handler(struct httpd_con *con, void *privdata)
{
int size = 4096, len = 0;
char *text = malloc(size);
if (text == NULL) {
httpd_send_error(con, "500 ERROR", "Out of Memory");
return -1;
}
len += snprintf(text + len, size - len, "<html><body><h1>ctorrent stats</h1>\n<table border=\"1\">\n");
len += snprintf(text + len, size - len, "<tr><td><b>Client IP:Port</b></td><td><b>Chunks (have/total/avail)</b></td><td><b>Download total(current)</b></td><td><b>Upload total(current)</b></td></tr>\n");
struct client_con *tmp;
list_for_each_entry(tmp, &client_list, list) {
len += snprintf(text + len, size - len, "<tr><td align=\"right\">%s</td><td align=\"right\">%3.2lf%% (%d/%d/%d)</td><td align=\"right\">%llu (%d)</td><td align=\"right\">%llu (%d)</td></tr>\n",
get_sockaddr_buf(&tmp->addr),
(double)tmp->chunk_have / (double)tmp->chunk_total * 100.0, tmp->chunk_have, tmp->chunk_total, tmp->chunk_avail,
tmp->total_dn, tmp->bw_dn, tmp->total_up, tmp->bw_up);
}
len += snprintf(text + len, size - len, "</table>\n</body></html>\n");
httpd_send_header(con, "200 OK", "text/html");
write(con->fd, text, len);
return 0;
}

9
connection.h Normal file
View File

@ -0,0 +1,9 @@
#ifndef _CONNECTION_H_
#define _CONNECTION_H_
#include "httpd.h"
int ctcs_accept_handler(int fd, void *privdata);
int ctcs_httpd_handler(struct httpd_con *con, void *privdata);
#endif /* _CONNECTION_H_ */

226
httpd.c Normal file
View File

@ -0,0 +1,226 @@
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/ioctl.h>
#include "event.h"
#include "httpd.h"
#include "list.h"
#include "logging.h"
#include "sockaddr.h"
#include "tcpsocket.h"
static LIST_HEAD(httpd_cb_list);
static void destroy_httpd_con(struct httpd_con *con)
{
close(event_get_fd(con->event));
event_remove_fd(con->event);
if (con->req_data != NULL)
free(con->req_data);
if (con->req_headers != NULL)
free(con->req_headers);
if (con->req_args != NULL)
free(con->req_args);
free(con);
}
int httpd_send_header(struct httpd_con *con, char *code, char *type)
{
char buf[256];
int len = snprintf(buf, sizeof(buf), "HTTP/1.0 %s\r\nContent-Type: %s\r\nConnection: close\r\n\r\n", code, type);
write(con->fd, buf, len);
return 0;
}
int httpd_send_error(struct httpd_con *con, char *code, char *msg)
{
httpd_send_header(con, code, "text/plain");
write(con->fd, msg, strlen(msg));
return 0;
}
static int parse_headers(struct httpd_con *con)
{
char *data = con->req_data;
while ((data = strstr(data, "\r\n")) && data < con->req_data + con->req_size) {
con->req_header_cnt++;
data += 2;
}
con->req_headers = malloc(con->req_header_cnt * sizeof(char *));
if (con->req_headers == NULL)
return -1;
int i = 0;
data = con->req_data;
do {
con->req_headers[i++] = data;
data = strstr(data, "\r\n");
if (data == NULL)
break;
*data++ = '\0';
*data++ = '\0';
} while (data < con->req_data + con->req_size);
return 0;
}
static int parse_args(struct httpd_con *con)
{
/* seperate GET (in req->data) */
char *req = strchr(con->req_data, ' ');
*req++ = '\0';
/* HTTP/1.1 will go to header[0] */
char *tmp = strchr(req, ' ');
*tmp++ = '\0';
con->req_headers[0] = tmp;
/* count args */
tmp = req;
while (*tmp != '\0' && tmp < con->req_data + con->req_size) {
if (*tmp == '?' || *tmp == '&')
con->req_arg_cnt++;
tmp++;
}
con->req_arg_cnt++;
/* alloc args */
con->req_args = malloc(con->req_arg_cnt * sizeof(char *));
if (con->req_args == NULL)
return -1;
int i = 0;
tmp = req;
con->req_args[i++] = tmp;
while (*tmp != '\0' && tmp < con->req_data + con->req_size) {
if (*tmp == '?' || *tmp == '&') {
*tmp++ = '\0';
con->req_args[i++] = tmp;
}
tmp++;
}
return 0;
}
static int httpd_content_handler(int fd, void *privdata)
{
struct httpd_con *con = (struct httpd_con *)privdata;
if (ioctl(fd, FIONREAD, &con->req_size) == -1) {
log_print(LOG_WARN, "http_content_handler(): ioctl(FIONREAD)");
destroy_httpd_con(con);
return -1;
}
con->req_data = malloc(con->req_size);
if (con->req_data == NULL) {
log_print(LOG_WARN, "http_content_handler(): out of memory");
destroy_httpd_con(con);
return -1;
}
con->req_size = read(fd, con->req_data, con->req_size);
if (con->req_size <= 0) {
log_print(LOG_WARN, "http_content_handler(): read()");
destroy_httpd_con(con);
return -1;
}
if (strncmp(con->req_data, "GET ", 4) != 0) {
httpd_send_error(con, "400 Invalid Request", "Not a GET request\r\n");
destroy_httpd_con(con);
return -1;
}
if (parse_headers(con) < 0) {
httpd_send_error(con, "400 Invalid Request", "Error parsing headers\r\n");
destroy_httpd_con(con);
return -1;
}
if (parse_args(con) < 0) {
httpd_send_error(con, "400 Invalid Request", "Error parsing arguments\r\n");
destroy_httpd_con(con);
return -1;
}
struct httpd_callback *entry;
list_for_each_entry(entry, &httpd_cb_list, list) {
if ((entry->wildcard == 1 && strncmp(entry->name, con->req_args[0], strlen(entry->name)) == 0) ||
(entry->wildcard == 0 && strcmp(entry->name, con->req_args[0]) == 0)) {
entry->callback(con, entry->privdata);
destroy_httpd_con(con);
return 0;
}
}
httpd_send_error(con, "404 Not Found", "File not found\r\n");
destroy_httpd_con(con);
return 0;
}
int httpd_accept_handler(int fd, void *privdata)
{
struct httpd_con *con = malloc(sizeof(struct httpd_con));
if (con == NULL) {
log_print(LOG_WARN, "httpd_accept_handler(): out of memory");
return 0;
}
memset(con, 0, sizeof(struct httpd_con));
unsigned int i = sizeof(con->addr);
con->fd = accept(fd, (struct sockaddr *)&con->addr, &i);
if (con->fd < 0) {
free(con);
return 0;
}
con->event = event_add_readfd(NULL, con->fd, httpd_content_handler, con);
return 0;
}
struct httpd_callback * httpd_add_cb(const char *name,
int wildcard,
int (* callback)(struct httpd_con *con, void *privdata),
void *privdata)
{
struct httpd_callback *hcb = malloc(sizeof(struct httpd_callback));
if (hcb == NULL) {
log_print(LOG_WARN, "httpd_create_cb(): out of memory");
return NULL;
}
hcb->name = strdup(name);
hcb->wildcard = wildcard;
hcb->callback = callback;
hcb->privdata = privdata;
list_add_tail(&hcb->list, &httpd_cb_list);
return hcb;
}
int httpd_remove_cb(struct httpd_callback *hcb)
{
list_del(&hcb->list);
free(hcb->name);
free(hcb);
return 0;
}

42
httpd.h Normal file
View File

@ -0,0 +1,42 @@
#ifndef _HTTPD_H_
#define _HTTPD_H_
#include <netinet/in.h>
#include "event.h"
#include "list.h"
struct httpd_con {
struct sockaddr_in addr;
struct event_fd *event;
int fd;
char *req_data;
unsigned int req_size;
char **req_headers;
unsigned int req_header_cnt;
char **req_args;
unsigned int req_arg_cnt;
};
struct httpd_callback {
struct list_head list;
char *name;
int wildcard;
int (* callback)(struct httpd_con *con, void *privdata);
void *privdata;
};
int httpd_send_header(struct httpd_con *con, char *code, char *type);
int httpd_send_error(struct httpd_con *con, char *code, char *msg);
struct httpd_callback * httpd_add_cb(const char *name, int flags, int (* cb)(struct httpd_con *con, void *privdata), void *privdata);
int httpd_remove_cb(struct httpd_callback *cb);
int httpd_accept_handler(int fd, void *privdata);
#endif // _HTTP_H_

207
torrent-stats.c
View File

@ -1,212 +1,16 @@
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <unistd.h> #include <unistd.h>
#include <string.h>
#include "configfile.h" #include "configfile.h"
#include "connection.h"
#include "event.h" #include "event.h"
#include "linebuffer.h" #include "httpd.h"
#include "list.h" #include "list.h"
#include "logging.h" #include "logging.h"
#include "sockaddr.h" #include "sockaddr.h"
#include "tcpsocket.h" #include "tcpsocket.h"
/*
startup:
PROTOCOL 0002
CTORRENT -CD0300-0xECFCD73A6132281994AF75C3 1179237904 1179239410 debian-40r0-i386-netinst.iso.torrent
CTSTATUS 0:174/1:5/0 636/636/636 0,1070 169914368,638976 0,0 32
periodic:
CTBW 0,1070 0,0
-------------------------
"SENDDETAIL":
CTDETAIL 166621184 262144 1179239468 1179238848
CTFILES
CTFILE 1 636 636 636 166621184 debian-40r0-i386-netinst.iso
CTFDONE
"CTDETAIL %lld %d %ld %ld",
BTCONTENT.GetTotalFilesLength(),
(int)(BTCONTENT.GetPieceLength()),
(long)now,
(long)(BTCONTENT.GetSeedTime()) );
"CTFILE %d %d %d %d %llu %s",
(int)n,
(int)(BTCONTENT.getFilePieces(n)),
(int)(tmpBitField.Count()),
(int)(tmpavail.Count()),
(unsigned long long)(file->bf_length),
file->bf_filename );
-------------------------
"SENDSTATUS":
CTSTATUS 0:174/1:5/0 636/636/636 0,959 169914368,655360 0,0 32
"CTSTATUS %d:%d/%d:%d/%d %d/%d/%d %d,%d %llu,%llu %d,%d %d",
(int)(m_ctstatus.seeders),
(int)(Tracker.GetSeedsCount() - ((BTCONTENT.pBF->IsFull() && Tracker.GetSeedsCount() > 0) ? 1 : 0)),
(int)(m_ctstatus.leechers),
(int)(Tracker.GetPeersCount() - Tracker.GetSeedsCount() - ((!BTCONTENT.pBF->IsFull() && Tracker.GetPeersCount() > 0) ? 1 : 0)),
(int)(WORLD.GetConnCount()),
(int)(m_ctstatus.nhave),
(int)(m_ctstatus.ntotal),
(int)(WORLD.Pieces_I_Can_Get()),
(int)(m_ctstatus.dlrate),
(int)(m_ctstatus.ulrate),
(unsigned long long)(m_ctstatus.dltotal),
(unsigned long long)(m_ctstatus.ultotal),
(int)(m_ctstatus.dlimit),
(int)(m_ctstatus.ulimit),
(int)(m_ctstatus.cacheused) );
-------------------------
"SENDCONF":
CTCONFIG 0 72 0.000000 100 1 0 16 0
"SENDPEERS":
CTPEERS
CTPEER M3-4-2--bae925c0c070 159.148.184.187 UnUi 0 862 114688 688128 520
CTPDONE
*/
static LIST_HEAD(client_list);
struct client_con {
struct list_head list;
struct sockaddr_in addr;
struct event_timeout *trigger;
struct event_fd *event;
struct linebuffer *lbuf;
int bw_up;
int bw_dn;
unsigned long long total_up;
unsigned long long total_dn;
int chunk_total;
int chunk_avail;
int chunk_have;
};
static void free_client(struct client_con *con)
{
close(event_get_fd(con->event));
event_remove_fd(con->event);
event_remove_timeout(con->trigger);
free(con->lbuf);
free(con);
}
static int trigger_status(void *privdata)
{
struct client_con *con = (struct client_con *)privdata;
write(event_get_fd(con->event), "SENDSTATUS\n", 11);
return 0;
}
static int data_cb(int fd, void *privdata)
{
struct client_con *con = (struct client_con *)privdata;
if (linebuffer_read(con->lbuf, fd) < 0) {
log_print(LOG_WARN, "data_cb(): client %s read", get_sockaddr_buf(&con->addr));
list_del(&con->list);
free_client(con);
return -1;
}
char *line;
while ((line = linebuffer_get(con->lbuf)) != NULL) {
// log_print(LOG_DEBUG, "%s: %s", get_sockaddr_buf(&con->addr), line);
if (strncmp(line, "CTBW ", 5) == 0) {
int bwup, bwdn, liup, lidn;
if (sscanf(line +5, "%d,%d %d,%d", &bwdn, &bwup, &lidn, &liup) == 4) {
con->bw_up = bwup;
con->bw_dn = bwdn;
}
} else if (strncmp(line, "CTSTATUS ", 9) == 0) {
int seeds1 = 0, seeds2 = 0, leech1 = 0, leech2 = 0, count = 0;
int chunk1 = 0, chunk2 = 0, chunk3 = 0, bwdn = 0, bwup = 0;
int lidn = 0, liup = 0, cache = 0;
unsigned long long totdn = 0, totup = 0;
if (sscanf(line +9, "%d:%d/%d:%d/%d %d/%d/%d %d,%d %llu,%llu %d,%d %d",
&seeds1, &seeds2, &leech1, &leech2, &count,
&chunk1, &chunk2, &chunk3,
&bwdn, &bwup, &totdn, &totup,
&lidn, &liup, &cache) == 15) {
con->total_up = totup;
con->total_dn = totdn;
con->chunk_have = chunk1;
con->chunk_total = chunk2;
con->chunk_avail = chunk3;
}
log_print(LOG_INFO, "%s: got %3.2lf%% of %3.2lf%% [down: %llu (%d), up: %llu (%d)] ",
get_sockaddr_buf(&con->addr),
(double)con->chunk_have / (double)con->chunk_total * 100.0,
(double)con->chunk_avail / (double)con->chunk_total * 100.0,
con->total_dn, con->bw_dn, con->total_up, con->bw_up);
}
free(line);
}
return 0;
}
static int accept_cb(int fd, void *privdata)
{
struct client_con *con = malloc(sizeof(struct client_con));
if (con == NULL) {
log_print(LOG_WARN, "accept_cb(): out of memory");
return 0;
}
memset(con, 0, sizeof(struct client_con));
con->lbuf = create_linebuffer(1024);
if (con->lbuf == NULL) {
log_print(LOG_WARN, "accept_cb(): out of memory");
free(con);
return 0;
}
unsigned int i = sizeof(con->addr);
int sockfd = accept(fd, (struct sockaddr *)&con->addr, &i);
if (sockfd < 0) {
log_print(LOG_WARN, "accept_cb(): accept()");
free(con->lbuf);
free(con);
return 0;
}
log_print(LOG_INFO, "new client %s", get_sockaddr_buf(&con->addr));
struct timeval tv = { .tv_sec = 10, .tv_usec = 0 };
con->trigger = event_add_timeout(&tv, trigger_status, con);
con->event = event_add_readfd(NULL, sockfd, data_cb, con);
list_add(&con->list, &client_list);
return 0;
}
static int listen_cb(const char *parameter, void *privdata) static int listen_cb(const char *parameter, void *privdata)
{ {
struct sockaddr_in addr; struct sockaddr_in addr;
@ -223,7 +27,7 @@ static int listen_cb(const char *parameter, void *privdata)
} }
log_print(LOG_INFO, "listen on %s", get_sockaddr_buf(&addr)); log_print(LOG_INFO, "listen on %s", get_sockaddr_buf(&addr));
event_add_readfd(NULL, sockfd, accept_cb, NULL); event_add_readfd(NULL, sockfd, privdata, NULL);
return 0; return 0;
} }
@ -232,7 +36,10 @@ int main(int argc, char *argv[])
if (config_parse("torrent-stats.conf") < 0) if (config_parse("torrent-stats.conf") < 0)
return 1; return 1;
config_get_strings("global", "listen", listen_cb, NULL); config_get_strings("global", "listen", listen_cb, ctcs_accept_handler);
config_get_strings("global", "listen-http", listen_cb, httpd_accept_handler);
httpd_add_cb("/", 1, ctcs_httpd_handler, NULL);
event_loop(); event_loop();
return 0; return 0;

1
torrent-stats.conf
View File

@ -1,2 +1,3 @@
[global] [global]
listen 0.0.0.0:2780 listen 0.0.0.0:2780
listen-http 0.0.0.0:8080