admin mode

This commit is contained in:
Olaf Rempel 2006-12-03 22:13:20 +01:00
parent 5bdb9d85fb
commit 48f18df080
2 changed files with 37 additions and 12 deletions

View File

@ -1,6 +1,7 @@
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
#include <unistd.h> #include <unistd.h>
#include <string.h>
#include <time.h> #include <time.h>
#include <netinet/ip.h> #include <netinet/ip.h>
@ -20,7 +21,7 @@ struct connection {
int src_fd; int src_fd;
int dst_fd; int dst_fd;
unsigned long login_time; time_t login_time;
}; };
static LIST_HEAD(connection_list); static LIST_HEAD(connection_list);
@ -34,6 +35,8 @@ static struct connection * create_connection()
} }
con->src_fd = con->dst_fd = -1; con->src_fd = con->dst_fd = -1;
con->src_addr.sin_addr.s_addr = con->src_addr.sin_addr.s_addr = 0;
con->src_addr.sin_port = con->src_addr.sin_port = 0;
con->login_time = time(NULL); con->login_time = time(NULL);
return con; return con;
@ -114,7 +117,37 @@ static int client_handler(int fd, void *privdata)
static int admin_handler(int fd, void *privdata) static int admin_handler(int fd, void *privdata)
{ {
return -1; struct connection *con = (struct connection *)privdata;
char buf[256];
int len = read(fd, buf, sizeof(buf));
if (len <= 0) {
list_del(&con->list);
destroy_connection(con);
return -1;
}
if (strncmp(buf, "list", 4) == 0) {
struct connection *entry;
list_for_each_entry(entry, &connection_list, list) {
len = snprintf(buf, sizeof(buf), "%d,", entry->src_fd);
len += get_sockaddr(&entry->src_addr, buf + len, sizeof(buf) - len);
len += snprintf(buf + len, sizeof(buf) - len, ",%d,", entry->dst_fd);
len += get_sockaddr(&entry->dst_addr, buf + len, sizeof(buf) - len);
len += snprintf(buf + len, sizeof(buf) - len, ",%ld\n", (time(NULL) - entry->login_time));
write(fd, buf, len);
}
} else if (strncmp(buf, "kill", 4) == 0) {
} else if (strncmp(buf, "quit", 4) == 0) {
list_del(&con->list);
destroy_connection(con);
return -1;
}
return 0;
} }
int listen_handler(int fd, void *privdata) int listen_handler(int fd, void *privdata)
@ -139,11 +172,9 @@ int listen_handler(int fd, void *privdata)
list_add_tail(&con->list, &connection_list); list_add_tail(&con->list, &connection_list);
if (mode == CON_ADMIN) if (mode == CON_ADMIN)
event_add_readfd(con->src_fd, client_handler, con);
else
event_add_readfd(con->src_fd, admin_handler, con); event_add_readfd(con->src_fd, admin_handler, con);
else
event_add_readfd(con->src_fd, client_handler, con);
return 0; return 0;
} }

View File

@ -164,7 +164,6 @@ int event_loop(void)
entry->flags &= ~FD_NEW; entry->flags &= ~FD_NEW;
if (entry->flags & FD_DELETE) { if (entry->flags & FD_DELETE) {
log_print(LOG_DEBUG, "event_loop: delete FD: %d (%p/%p)", entry->fd, entry->callback, entry->privdata);
list_del(&entry->list); list_del(&entry->list);
free(entry); free(entry);
@ -173,7 +172,6 @@ int event_loop(void)
readfds = &fdsets[0]; readfds = &fdsets[0];
FD_ZERO(readfds); FD_ZERO(readfds);
} }
log_print(LOG_DEBUG, "event_loop: read FD: %d (%p/%p)", entry->fd, entry->callback, entry->privdata);
FD_SET(entry->fd, readfds); FD_SET(entry->fd, readfds);
} else if ((entry->flags & FD_WRITE) != 0) { } else if ((entry->flags & FD_WRITE) != 0) {
@ -181,7 +179,6 @@ int event_loop(void)
writefds = &fdsets[1]; writefds = &fdsets[1];
FD_ZERO(writefds); FD_ZERO(writefds);
} }
log_print(LOG_DEBUG, "event_loop: write FD: %d (%p/%p)", entry->fd, entry->callback, entry->privdata);
FD_SET(entry->fd, writefds); FD_SET(entry->fd, writefds);
} else if ((entry->flags & FD_EXCEPT) != 0) { } else if ((entry->flags & FD_EXCEPT) != 0) {
@ -189,7 +186,6 @@ int event_loop(void)
exceptfds = &fdsets[2]; exceptfds = &fdsets[2];
FD_ZERO(exceptfds); FD_ZERO(exceptfds);
} }
log_print(LOG_DEBUG, "event_loop: except FD: %d (%p/%p)", entry->fd, entry->callback, entry->privdata);
FD_SET(entry->fd, exceptfds); FD_SET(entry->fd, exceptfds);
} }
} }
@ -218,8 +214,6 @@ int event_loop(void)
} }
} }
log_print(LOG_DEBUG, "");
int i = select(FD_SETSIZE, readfds, writefds, exceptfds, timeout_p); int i = select(FD_SETSIZE, readfds, writefds, exceptfds, timeout_p);
if (i < 0) { if (i < 0) {
/* On error, -1 is returned, and errno is set /* On error, -1 is returned, and errno is set