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

multi torrent support

Browse Source
This commit is contained in:
Olaf Rempel 2007-07-07 13:11:51 +02:00
parent 419ce7766e
commit 2e21d27d10
1 changed files with 96 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

124
connection.c
View File

@ -12,7 +12,13 @@
#include "sockaddr.h" #include "sockaddr.h"
#include "tcpsocket.h" #include "tcpsocket.h"
static LIST_HEAD(client_list); static LIST_HEAD(torrent_list);
struct torrent_file {
struct list_head list;
struct list_head client_list;
char *name;
};
struct client_con { struct client_con {
struct list_head list; struct list_head list;
@ -32,10 +38,47 @@ struct client_con {
int chunk_have; int chunk_have;
long completed; long completed;
struct torrent_file *torrent;
}; };
static struct torrent_file * find_create_torrent(const char *filename)
{
struct torrent_file *torrent;
list_for_each_entry(torrent, &torrent_list, list) {
if (strcmp(torrent->name, filename) == 0)
return torrent;
}
torrent = malloc(sizeof(struct torrent_file) + strlen(filename));
if (torrent == NULL) {
log_print(LOG_WARN, "find_create_torrent(): out of memory");
return NULL;
}
INIT_LIST_HEAD(&torrent->client_list);
torrent->name = strdup(filename);
struct torrent_file *search;
list_for_each_entry(search, &torrent_list, list)
if (strcmp(search->name, torrent->name) > 0)
break;
list_add_tail(&torrent->list, &search->list);
return torrent;
}
static void free_client(struct client_con *con) static void free_client(struct client_con *con)
{ {
list_del(&con->list);
/* remove torrents without clients */
if (list_empty(&con->torrent->client_list)) {
list_del(&con->torrent->list);
free(con->torrent->name);
free(con->torrent);
}
close(event_get_fd(con->event)); close(event_get_fd(con->event));
event_remove_fd(con->event); event_remove_fd(con->event);
free(con->lbuf); free(con->lbuf);
@ -47,8 +90,6 @@ static int data_cb(int fd, void *privdata)
struct client_con *con = (struct client_con *)privdata; struct client_con *con = (struct client_con *)privdata;
if (linebuffer_readfd(con->lbuf, fd) < 0) { if (linebuffer_readfd(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); free_client(con);
return -1; return -1;
} }
@ -80,6 +121,17 @@ static int data_cb(int fd, void *privdata)
con->chunk_total = chunk2; con->chunk_total = chunk2;
con->chunk_avail = chunk3; con->chunk_avail = chunk3;
} }
} else if (strncmp(line, "CTORRENT ", 9) == 0) {
char *filename = strrchr(line +9, ' ');
if (filename != NULL) {
struct torrent_file *torrent = find_create_torrent(filename +1);
if (torrent != NULL) {
list_del(&con->list);
con->torrent = torrent;
list_add_tail(&con->list, &con->torrent->client_list);
}
}
} }
linebuffer_freeline(con->lbuf); linebuffer_freeline(con->lbuf);
} }
@ -91,7 +143,7 @@ static int data_cb(int fd, void *privdata)
list_del(&con->list); list_del(&con->list);
struct client_con *search; struct client_con *search;
list_for_each_entry(search, &client_list, list) { list_for_each_entry(search, &con->torrent->client_list, list) {
if (search->completed == 0) if (search->completed == 0)
break; break;
@ -106,26 +158,30 @@ static int data_cb(int fd, void *privdata)
int ctcs_trigger_status(void *privdata) int ctcs_trigger_status(void *privdata)
{ {
int delete = 0;
long timeout = time(NULL) - 300; long timeout = time(NULL) - 300;
struct client_con *con; struct torrent_file *torrent;
list_for_each_entry(con, &client_list, list) { list_for_each_entry(torrent, &torrent_list, list) {
write(event_get_fd(con->event), "SENDSTATUS\n", 11); int delete = 0;
delete += (con->completed == 0) ? -1 : 1; struct client_con *con;
} list_for_each_entry(con, &torrent->client_list, list) {
write(event_get_fd(con->event), "SENDSTATUS\n", 11);
/* delete holds the number of clients to quit */ delete += (con->completed == 0) ? -1 : 1;
list_for_each_entry(con, &client_list, list) { }
if (delete <= 0)
break;
if (con->completed == 0 || con->completed > timeout) /* delete holds the number of clients to quit */
continue; list_for_each_entry(con, &torrent->client_list, list) {
if (delete <= 0)
break;
write(event_get_fd(con->event), "CTQUIT\n", 7); if (con->completed == 0 || con->completed > timeout)
delete--; continue;
write(event_get_fd(con->event), "CTQUIT\n", 7);
delete--;
}
} }
return 0; return 0;
} }
@ -158,32 +214,44 @@ int ctcs_accept_handler(int fd, void *privdata)
con->event = event_add_readfd(NULL, sockfd, data_cb, con); con->event = event_add_readfd(NULL, sockfd, data_cb, con);
list_add_tail(&con->list, &client_list); con->torrent = find_create_torrent("[unknown]");
list_add_tail(&con->list, &con->torrent->client_list);
return 0; return 0;
} }
int ctcs_httpd_handler(struct httpd_con *con, void *privdata) int ctcs_httpd_handler(struct httpd_con *con, void *privdata)
{ {
struct linebuffer *lbuf = create_linebuffer(4096); struct linebuffer *lbuf = create_linebuffer(16384);
if (lbuf == NULL) { if (lbuf == NULL) {
httpd_send_error(con, "500 ERROR", "Out of Memory"); httpd_send_error(con, "500 ERROR", "Out of Memory");
return -1; return -1;
} }
linebuffer_printf(lbuf, "HTTP/1.0 200 OK\r\nContent-Type: text/html\r\nConnection: close\r\n\r\n"); linebuffer_printf(lbuf, "HTTP/1.0 200 OK\r\nContent-Type: text/html\r\nConnection: close\r\n\r\n");
linebuffer_printf(lbuf, "<html><head><meta http-equiv=\"refresh\" content=\"30;\"></head><body><h1>ctorrent stats</h1>\n<table border=\"1\">\n"); linebuffer_printf(lbuf, "<html><head><meta http-equiv=\"refresh\" content=\"30;\"></head><body><h1>ctorrent stats</h1>\n");
linebuffer_printf(lbuf, "<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><td><b>Completed since</b></td></tr>\n");
struct client_con *tmp; struct torrent_file *torrent;
list_for_each_entry(tmp, &client_list, list) { list_for_each_entry(torrent, &torrent_list, list) {
linebuffer_printf(lbuf, "<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><td align=\"right\">%s</td></tr>\n", if (list_empty(&torrent->client_list))
continue;
linebuffer_printf(lbuf, "<table border=\"1\">\n<tr><td colspan=\"5\" align=\"center\">%s</td></tr>\n", torrent->name);
linebuffer_printf(lbuf, "<tr></td><td><b>Client IP:Port</b></td><td><b>Chunks (have/total/avail)</b></td>");
linebuffer_printf(lbuf, "<td><b>Download total(current)</b></td><td><b>Upload total(current)</b></td><td><b>Completed since</b></td></tr>\n");
struct client_con *tmp;
list_for_each_entry(tmp, &torrent->client_list, list) {
linebuffer_printf(lbuf, "<tr><td align=\"right\">%s</td><td align=\"right\">%3.2lf%% (%d/%d/%d)</td>",
get_sockaddr_buf(&tmp->addr), get_sockaddr_buf(&tmp->addr),
(double)tmp->chunk_have / (double)tmp->chunk_total * 100.0, tmp->chunk_have, tmp->chunk_total, tmp->chunk_avail, (double)tmp->chunk_have / (double)tmp->chunk_total * 100.0, tmp->chunk_have, tmp->chunk_total, tmp->chunk_avail);
linebuffer_printf(lbuf, "<td align=\"right\">%llu (%d)</td><td align=\"right\">%llu (%d)</td><td align=\"right\">%s</td></tr>\n",
tmp->total_dn, tmp->bw_dn, tmp->total_up, tmp->bw_up, tmp->total_dn, tmp->bw_dn, tmp->total_up, tmp->bw_up,
(tmp->completed != 0) ? ctime(&tmp->completed) : "-"); (tmp->completed != 0) ? ctime(&tmp->completed) : "-");
}
linebuffer_printf(lbuf, "</table>\n<br><br>\n");
} }
linebuffer_printf(lbuf, "</body></html>\n");
linebuffer_printf(lbuf, "</table>\n</body></html>\n");
linebuffer_writefd(lbuf, con->fd); linebuffer_writefd(lbuf, con->fd);
linebuffer_free(lbuf); linebuffer_free(lbuf);