2007-10-16 15:34:28 +00:00
|
|
|
/*
|
|
|
|
* UNIX SOCK_STREAM protocol layer (uxst)
|
|
|
|
*
|
2009-08-16 17:12:36 +00:00
|
|
|
* Copyright 2000-2009 Willy Tarreau <w@1wt.eu>
|
2007-10-16 15:34:28 +00:00
|
|
|
*
|
|
|
|
* This program is free software; you can redistribute it and/or
|
|
|
|
* modify it under the terms of the GNU General Public License
|
|
|
|
* as published by the Free Software Foundation; either version
|
|
|
|
* 2 of the License, or (at your option) any later version.
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
|
|
|
|
#include <ctype.h>
|
|
|
|
#include <errno.h>
|
|
|
|
#include <fcntl.h>
|
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <syslog.h>
|
|
|
|
#include <time.h>
|
|
|
|
|
|
|
|
#include <sys/param.h>
|
|
|
|
#include <sys/socket.h>
|
|
|
|
#include <sys/stat.h>
|
|
|
|
#include <sys/types.h>
|
|
|
|
#include <sys/un.h>
|
|
|
|
|
|
|
|
#include <common/compat.h>
|
|
|
|
#include <common/config.h>
|
|
|
|
#include <common/debug.h>
|
2007-10-28 10:14:07 +00:00
|
|
|
#include <common/errors.h>
|
2007-10-16 15:34:28 +00:00
|
|
|
#include <common/memory.h>
|
|
|
|
#include <common/mini-clist.h>
|
|
|
|
#include <common/standard.h>
|
2008-07-06 22:09:58 +00:00
|
|
|
#include <common/ticks.h>
|
2007-10-16 15:34:28 +00:00
|
|
|
#include <common/time.h>
|
|
|
|
#include <common/version.h>
|
|
|
|
|
|
|
|
#include <types/global.h>
|
|
|
|
|
|
|
|
#include <proto/acl.h>
|
|
|
|
#include <proto/backend.h>
|
|
|
|
#include <proto/buffers.h>
|
|
|
|
#include <proto/fd.h>
|
|
|
|
#include <proto/log.h>
|
|
|
|
#include <proto/protocols.h>
|
|
|
|
#include <proto/proto_uxst.h>
|
|
|
|
#include <proto/queue.h>
|
|
|
|
#include <proto/session.h>
|
2008-12-07 15:06:43 +00:00
|
|
|
#include <proto/stream_interface.h>
|
2007-10-16 15:34:28 +00:00
|
|
|
#include <proto/stream_sock.h>
|
|
|
|
#include <proto/task.h>
|
|
|
|
|
|
|
|
#ifndef MAXPATHLEN
|
|
|
|
#define MAXPATHLEN 128
|
|
|
|
#endif
|
|
|
|
|
2007-10-28 20:59:24 +00:00
|
|
|
static int uxst_bind_listeners(struct protocol *proto);
|
|
|
|
static int uxst_unbind_listeners(struct protocol *proto);
|
|
|
|
|
|
|
|
/* Note: must not be declared <const> as its list will be overwritten */
|
|
|
|
static struct protocol proto_unix = {
|
|
|
|
.name = "unix_stream",
|
|
|
|
.sock_domain = PF_UNIX,
|
|
|
|
.sock_type = SOCK_STREAM,
|
|
|
|
.sock_prot = 0,
|
|
|
|
.sock_family = AF_UNIX,
|
|
|
|
.sock_addrlen = sizeof(struct sockaddr_un),
|
|
|
|
.l3_addrlen = sizeof(((struct sockaddr_un*)0)->sun_path),/* path len */
|
|
|
|
.read = &stream_sock_read,
|
|
|
|
.write = &stream_sock_write,
|
|
|
|
.bind_all = uxst_bind_listeners,
|
|
|
|
.unbind_all = uxst_unbind_listeners,
|
|
|
|
.enable_all = enable_all_listeners,
|
|
|
|
.disable_all = disable_all_listeners,
|
|
|
|
.listeners = LIST_HEAD_INIT(proto_unix.listeners),
|
|
|
|
.nb_listeners = 0,
|
|
|
|
};
|
|
|
|
|
|
|
|
/********************************
|
|
|
|
* 1) low-level socket functions
|
|
|
|
********************************/
|
|
|
|
|
|
|
|
|
2007-10-16 15:34:28 +00:00
|
|
|
/* This function creates a named PF_UNIX stream socket at address <path>. Note
|
2007-10-18 10:45:54 +00:00
|
|
|
* that the path cannot be NULL nor empty. <uid> and <gid> different of -1 will
|
|
|
|
* be used to change the socket owner. If <mode> is not 0, it will be used to
|
|
|
|
* restrict access to the socket. While it is known not to be portable on every
|
|
|
|
* OS, it's still useful where it works.
|
2007-10-16 15:34:28 +00:00
|
|
|
* It returns the assigned file descriptor, or -1 in the event of an error.
|
|
|
|
*/
|
2007-10-18 10:45:54 +00:00
|
|
|
static int create_uxst_socket(const char *path, uid_t uid, gid_t gid, mode_t mode)
|
2007-10-16 15:34:28 +00:00
|
|
|
{
|
|
|
|
char tempname[MAXPATHLEN];
|
|
|
|
char backname[MAXPATHLEN];
|
|
|
|
struct sockaddr_un addr;
|
|
|
|
|
|
|
|
int ret, sock;
|
|
|
|
|
|
|
|
/* 1. create socket names */
|
|
|
|
if (!path[0]) {
|
|
|
|
Alert("Invalid name for a UNIX socket. Aborting.\n");
|
|
|
|
goto err_return;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = snprintf(tempname, MAXPATHLEN, "%s.%d.tmp", path, pid);
|
|
|
|
if (ret < 0 || ret >= MAXPATHLEN) {
|
|
|
|
Alert("name too long for UNIX socket. Aborting.\n");
|
|
|
|
goto err_return;
|
|
|
|
}
|
|
|
|
|
|
|
|
ret = snprintf(backname, MAXPATHLEN, "%s.%d.bak", path, pid);
|
|
|
|
if (ret < 0 || ret >= MAXPATHLEN) {
|
|
|
|
Alert("name too long for UNIX socket. Aborting.\n");
|
|
|
|
goto err_return;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* 2. clean existing orphaned entries */
|
|
|
|
if (unlink(tempname) < 0 && errno != ENOENT) {
|
|
|
|
Alert("error when trying to unlink previous UNIX socket. Aborting.\n");
|
|
|
|
goto err_return;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (unlink(backname) < 0 && errno != ENOENT) {
|
|
|
|
Alert("error when trying to unlink previous UNIX socket. Aborting.\n");
|
|
|
|
goto err_return;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* 3. backup existing socket */
|
|
|
|
if (link(path, backname) < 0 && errno != ENOENT) {
|
|
|
|
Alert("error when trying to preserve previous UNIX socket. Aborting.\n");
|
|
|
|
goto err_return;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* 4. prepare new socket */
|
|
|
|
addr.sun_family = AF_UNIX;
|
|
|
|
strncpy(addr.sun_path, tempname, sizeof(addr.sun_path));
|
|
|
|
addr.sun_path[sizeof(addr.sun_path) - 1] = 0;
|
|
|
|
|
|
|
|
sock = socket(PF_UNIX, SOCK_STREAM, 0);
|
|
|
|
if (sock < 0) {
|
|
|
|
Alert("cannot create socket for UNIX listener. Aborting.\n");
|
|
|
|
goto err_unlink_back;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (sock >= global.maxsock) {
|
|
|
|
Alert("socket(): not enough free sockets for UNIX listener. Raise -n argument. Aborting.\n");
|
|
|
|
goto err_unlink_temp;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (fcntl(sock, F_SETFL, O_NONBLOCK) == -1) {
|
|
|
|
Alert("cannot make UNIX socket non-blocking. Aborting.\n");
|
|
|
|
goto err_unlink_temp;
|
|
|
|
}
|
|
|
|
|
|
|
|
if (bind(sock, (struct sockaddr *)&addr, sizeof(addr)) < 0) {
|
|
|
|
/* note that bind() creates the socket <tempname> on the file system */
|
|
|
|
Alert("cannot bind socket for UNIX listener. Aborting.\n");
|
|
|
|
goto err_unlink_temp;
|
|
|
|
}
|
|
|
|
|
2007-10-18 10:45:54 +00:00
|
|
|
if (((uid != -1 || gid != -1) && (chown(tempname, uid, gid) == -1)) ||
|
|
|
|
(mode != 0 && chmod(tempname, mode) == -1)) {
|
|
|
|
Alert("cannot change UNIX socket ownership. Aborting.\n");
|
|
|
|
goto err_unlink_temp;
|
|
|
|
}
|
|
|
|
|
2007-10-16 15:34:28 +00:00
|
|
|
if (listen(sock, 0) < 0) {
|
|
|
|
Alert("cannot listen to socket for UNIX listener. Aborting.\n");
|
|
|
|
goto err_unlink_temp;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* 5. install.
|
|
|
|
* Point of no return: we are ready, we'll switch the sockets. We don't
|
|
|
|
* fear loosing the socket <path> because we have a copy of it in
|
|
|
|
* backname.
|
|
|
|
*/
|
|
|
|
if (rename(tempname, path) < 0) {
|
|
|
|
Alert("cannot switch final and temporary sockets for UNIX listener. Aborting.\n");
|
|
|
|
goto err_rename;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* 6. cleanup */
|
|
|
|
unlink(backname); /* no need to keep this one either */
|
|
|
|
|
|
|
|
return sock;
|
|
|
|
|
|
|
|
err_rename:
|
|
|
|
ret = rename(backname, path);
|
|
|
|
if (ret < 0 && errno == ENOENT)
|
|
|
|
unlink(path);
|
|
|
|
err_unlink_temp:
|
|
|
|
unlink(tempname);
|
|
|
|
close(sock);
|
|
|
|
err_unlink_back:
|
|
|
|
unlink(backname);
|
|
|
|
err_return:
|
|
|
|
return -1;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Tries to destroy the UNIX stream socket <path>. The socket must not be used
|
|
|
|
* anymore. It practises best effort, and no error is returned.
|
|
|
|
*/
|
|
|
|
static void destroy_uxst_socket(const char *path)
|
|
|
|
{
|
|
|
|
struct sockaddr_un addr;
|
|
|
|
int sock, ret;
|
|
|
|
|
|
|
|
/* We might have been chrooted, so we may not be able to access the
|
|
|
|
* socket. In order to avoid bothering the other end, we connect with a
|
|
|
|
* wrong protocol, namely SOCK_DGRAM. The return code from connect()
|
|
|
|
* is enough to know if the socket is still live or not. If it's live
|
|
|
|
* in mode SOCK_STREAM, we get EPROTOTYPE or anything else but not
|
|
|
|
* ECONNREFUSED. In this case, we do not touch it because it's used
|
|
|
|
* by some other process.
|
|
|
|
*/
|
|
|
|
sock = socket(PF_UNIX, SOCK_DGRAM, 0);
|
|
|
|
if (sock < 0)
|
|
|
|
return;
|
|
|
|
|
|
|
|
addr.sun_family = AF_UNIX;
|
|
|
|
strncpy(addr.sun_path, path, sizeof(addr.sun_path));
|
2007-10-18 14:15:52 +00:00
|
|
|
addr.sun_path[sizeof(addr.sun_path) - 1] = 0;
|
2007-10-16 15:34:28 +00:00
|
|
|
ret = connect(sock, (struct sockaddr *)&addr, sizeof(addr));
|
|
|
|
if (ret < 0 && errno == ECONNREFUSED) {
|
|
|
|
/* Connect failed: the socket still exists but is not used
|
|
|
|
* anymore. Let's remove this socket now.
|
|
|
|
*/
|
|
|
|
unlink(path);
|
|
|
|
}
|
|
|
|
close(sock);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
2007-10-28 20:59:24 +00:00
|
|
|
/********************************
|
|
|
|
* 2) listener-oriented functions
|
|
|
|
********************************/
|
|
|
|
|
|
|
|
|
|
|
|
/* This function creates the UNIX socket associated to the listener. It changes
|
|
|
|
* the state from ASSIGNED to LISTEN. The socket is NOT enabled for polling.
|
|
|
|
* The return value is composed from ERR_NONE, ERR_RETRYABLE and ERR_FATAL.
|
|
|
|
*/
|
|
|
|
static int uxst_bind_listener(struct listener *listener)
|
|
|
|
{
|
|
|
|
int fd;
|
2008-12-07 15:06:43 +00:00
|
|
|
|
2007-10-28 20:59:24 +00:00
|
|
|
if (listener->state != LI_ASSIGNED)
|
|
|
|
return ERR_NONE; /* already bound */
|
|
|
|
|
|
|
|
fd = create_uxst_socket(((struct sockaddr_un *)&listener->addr)->sun_path,
|
|
|
|
listener->perm.ux.uid,
|
|
|
|
listener->perm.ux.gid,
|
|
|
|
listener->perm.ux.mode);
|
|
|
|
if (fd == -1)
|
|
|
|
return ERR_FATAL;
|
2008-12-07 15:06:43 +00:00
|
|
|
|
2007-10-28 20:59:24 +00:00
|
|
|
/* the socket is now listening */
|
|
|
|
listener->fd = fd;
|
|
|
|
listener->state = LI_LISTEN;
|
|
|
|
|
|
|
|
/* the function for the accept() event */
|
|
|
|
fd_insert(fd);
|
|
|
|
fdtab[fd].cb[DIR_RD].f = listener->accept;
|
|
|
|
fdtab[fd].cb[DIR_WR].f = NULL; /* never called */
|
|
|
|
fdtab[fd].cb[DIR_RD].b = fdtab[fd].cb[DIR_WR].b = NULL;
|
2008-08-29 21:36:51 +00:00
|
|
|
fdtab[fd].owner = listener; /* reference the listener instead of a task */
|
2007-10-28 20:59:24 +00:00
|
|
|
fdtab[fd].state = FD_STLISTEN;
|
|
|
|
fdtab[fd].peeraddr = NULL;
|
|
|
|
fdtab[fd].peerlen = 0;
|
|
|
|
return ERR_NONE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* This function closes the UNIX sockets for the specified listener.
|
|
|
|
* The listener enters the LI_ASSIGNED state. It always returns ERR_NONE.
|
|
|
|
*/
|
|
|
|
static int uxst_unbind_listener(struct listener *listener)
|
|
|
|
{
|
|
|
|
if (listener->state == LI_READY)
|
|
|
|
EV_FD_CLR(listener->fd, DIR_RD);
|
|
|
|
|
|
|
|
if (listener->state >= LI_LISTEN) {
|
2007-10-28 21:07:08 +00:00
|
|
|
fd_delete(listener->fd);
|
2007-10-28 20:59:24 +00:00
|
|
|
listener->state = LI_ASSIGNED;
|
|
|
|
destroy_uxst_socket(((struct sockaddr_un *)&listener->addr)->sun_path);
|
|
|
|
}
|
|
|
|
return ERR_NONE;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* Add a listener to the list of unix stream listeners. The listener's state
|
|
|
|
* is automatically updated from LI_INIT to LI_ASSIGNED. The number of
|
|
|
|
* listeners is updated. This is the function to use to add a new listener.
|
|
|
|
*/
|
|
|
|
void uxst_add_listener(struct listener *listener)
|
|
|
|
{
|
|
|
|
if (listener->state != LI_INIT)
|
|
|
|
return;
|
|
|
|
listener->state = LI_ASSIGNED;
|
|
|
|
listener->proto = &proto_unix;
|
|
|
|
LIST_ADDQ(&proto_unix.listeners, &listener->proto_list);
|
|
|
|
proto_unix.nb_listeners++;
|
|
|
|
}
|
|
|
|
|
|
|
|
/********************************
|
|
|
|
* 3) protocol-oriented functions
|
|
|
|
********************************/
|
|
|
|
|
|
|
|
|
2007-10-16 15:34:28 +00:00
|
|
|
/* This function creates all UNIX sockets bound to the protocol entry <proto>.
|
|
|
|
* It is intended to be used as the protocol's bind_all() function.
|
|
|
|
* The sockets will be registered but not added to any fd_set, in order not to
|
|
|
|
* loose them across the fork(). A call to uxst_enable_listeners() is needed
|
|
|
|
* to complete initialization.
|
|
|
|
*
|
|
|
|
* The return value is composed from ERR_NONE, ERR_RETRYABLE and ERR_FATAL.
|
|
|
|
*/
|
|
|
|
static int uxst_bind_listeners(struct protocol *proto)
|
|
|
|
{
|
|
|
|
struct listener *listener;
|
|
|
|
int err = ERR_NONE;
|
|
|
|
|
|
|
|
list_for_each_entry(listener, &proto->listeners, proto_list) {
|
2007-10-28 20:59:24 +00:00
|
|
|
err |= uxst_bind_listener(listener);
|
|
|
|
if (err != ERR_NONE)
|
2007-10-16 15:34:28 +00:00
|
|
|
continue;
|
|
|
|
}
|
|
|
|
return err;
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/* This function stops all listening UNIX sockets bound to the protocol
|
|
|
|
* <proto>. It does not detaches them from the protocol.
|
|
|
|
* It always returns ERR_NONE.
|
|
|
|
*/
|
|
|
|
static int uxst_unbind_listeners(struct protocol *proto)
|
|
|
|
{
|
|
|
|
struct listener *listener;
|
|
|
|
|
2007-10-28 20:59:24 +00:00
|
|
|
list_for_each_entry(listener, &proto->listeners, proto_list)
|
|
|
|
uxst_unbind_listener(listener);
|
2007-10-16 15:34:28 +00:00
|
|
|
return ERR_NONE;
|
|
|
|
}
|
|
|
|
|
2007-10-28 20:59:24 +00:00
|
|
|
|
|
|
|
/********************************
|
|
|
|
* 4) high-level functions
|
|
|
|
********************************/
|
|
|
|
|
|
|
|
|
2007-10-16 15:34:28 +00:00
|
|
|
/*
|
|
|
|
* This function is called on a read event from a listen socket, corresponding
|
|
|
|
* to an accept. It tries to accept as many connections as possible.
|
|
|
|
* It returns 0. Since we use UNIX sockets on the local system for monitoring
|
|
|
|
* purposes and other related things, we do not need to output as many messages
|
|
|
|
* as with TCP which can fall under attack.
|
|
|
|
*/
|
|
|
|
int uxst_event_accept(int fd) {
|
2008-08-29 21:36:51 +00:00
|
|
|
struct listener *l = fdtab[fd].owner;
|
2007-10-16 15:34:28 +00:00
|
|
|
struct session *s;
|
|
|
|
struct task *t;
|
|
|
|
int cfd;
|
|
|
|
int max_accept;
|
|
|
|
|
|
|
|
if (global.nbproc > 1)
|
|
|
|
max_accept = 8; /* let other processes catch some connections too */
|
|
|
|
else
|
|
|
|
max_accept = -1;
|
|
|
|
|
2009-03-28 10:02:18 +00:00
|
|
|
while (max_accept--) {
|
2007-10-16 15:34:28 +00:00
|
|
|
struct sockaddr_storage addr;
|
|
|
|
socklen_t laddr = sizeof(addr);
|
|
|
|
|
|
|
|
if ((cfd = accept(fd, (struct sockaddr *)&addr, &laddr)) == -1) {
|
|
|
|
switch (errno) {
|
|
|
|
case EAGAIN:
|
|
|
|
case EINTR:
|
|
|
|
case ECONNABORTED:
|
|
|
|
return 0; /* nothing more to accept */
|
|
|
|
case ENFILE:
|
|
|
|
/* Process reached system FD limit. Check system tunables. */
|
|
|
|
return 0;
|
|
|
|
case EMFILE:
|
|
|
|
/* Process reached process FD limit. Check 'ulimit-n'. */
|
|
|
|
return 0;
|
|
|
|
case ENOBUFS:
|
|
|
|
case ENOMEM:
|
|
|
|
/* Process reached system memory limit. Check system tunables. */
|
|
|
|
return 0;
|
|
|
|
default:
|
|
|
|
return 0;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2009-03-28 10:02:18 +00:00
|
|
|
if (l->nbconn >= l->maxconn || actconn >= global.maxconn) {
|
2007-10-16 15:34:28 +00:00
|
|
|
/* too many connections, we shoot this one and return.
|
|
|
|
* FIXME: it would be better to simply switch the listener's
|
|
|
|
* state to LI_FULL and disable the FD. We could re-enable
|
|
|
|
* it upon fd_delete(), but this requires all protocols to
|
|
|
|
* be switched.
|
|
|
|
*/
|
2008-12-01 00:44:25 +00:00
|
|
|
goto out_close;
|
2007-10-16 15:34:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if ((s = pool_alloc2(pool2_session)) == NULL) {
|
|
|
|
Alert("out of memory in uxst_event_accept().\n");
|
2008-12-01 00:44:25 +00:00
|
|
|
goto out_close;
|
2007-10-16 15:34:28 +00:00
|
|
|
}
|
|
|
|
|
2008-11-23 18:53:55 +00:00
|
|
|
LIST_ADDQ(&sessions, &s->list);
|
2008-12-07 19:16:23 +00:00
|
|
|
LIST_INIT(&s->back_refs);
|
2008-11-23 18:53:55 +00:00
|
|
|
|
[MAJOR] proto_uxst rework -> SNMP support
Currently there is a ~16KB limit for a data size passed via unix socket.
It is caused by a trivial bug ttat is going to fixed soon, however
in most cases there is no need to dump a full stats.
This patch makes possible to select a scope of dumped data by extending
current "show stat" to "show stat [<iid> <type> <sid>]":
- iid is a proxy id, -1 to dump all proxies
- type selects type of dumpable objects: 1 for frontend, 2 for backend, 4 for
server, -1 for all types. Values can be ORed, for example:
1+2=3 -> frontend+backend.
1+2+4=7 -> frontend+backend+server.
- sid is a service id, -1 to dump everything from the selected proxy.
To do this I implemented a new session flag (SN_STAT_BOUND), added three
variables in data_ctx.stats (iid, type, sid), modified dumpstats.c and
completely revorked the process_uxst_stats: now it waits for a "\n"
terminated string, splits args and uses them. BTW: It should be quite easy
to add new commands, for example to enable/disable servers, the only problem
I can see is a not very lucky config name (*stats* socket). :|
During the work I also fixed two bug:
- s->flags were not initialized for proto_uxst
- missing comma if throttling not enabled (caused by a stupid change in
"Implement persistent id for proxies and servers")
Other changes:
- No more magic type valuse, use STATS_TYPE_FE/STATS_TYPE_BE/STATS_TYPE_SV
- Don't memset full s->data_ctx (it was clearing s->data_ctx.stats.{iid/type/sid},
instead initialize stats.sv & stats.sv_st (stats.px and stats.px_st were already
initialized)
With all that changes it was extremely easy to write a short perl plugin
for a perl-enabled net-snmp (also included in this patch).
29385 is my PEN (Private Enterprise Number) and I'm willing to donate
the SNMPv2-SMI::enterprises.29385.106.* OIDs for HAProxy if there
is nothing assigned already.
2008-03-02 01:42:14 +00:00
|
|
|
s->flags = 0;
|
2008-08-16 12:55:08 +00:00
|
|
|
s->term_trace = 0;
|
[MAJOR] proto_uxst rework -> SNMP support
Currently there is a ~16KB limit for a data size passed via unix socket.
It is caused by a trivial bug ttat is going to fixed soon, however
in most cases there is no need to dump a full stats.
This patch makes possible to select a scope of dumped data by extending
current "show stat" to "show stat [<iid> <type> <sid>]":
- iid is a proxy id, -1 to dump all proxies
- type selects type of dumpable objects: 1 for frontend, 2 for backend, 4 for
server, -1 for all types. Values can be ORed, for example:
1+2=3 -> frontend+backend.
1+2+4=7 -> frontend+backend+server.
- sid is a service id, -1 to dump everything from the selected proxy.
To do this I implemented a new session flag (SN_STAT_BOUND), added three
variables in data_ctx.stats (iid, type, sid), modified dumpstats.c and
completely revorked the process_uxst_stats: now it waits for a "\n"
terminated string, splits args and uses them. BTW: It should be quite easy
to add new commands, for example to enable/disable servers, the only problem
I can see is a not very lucky config name (*stats* socket). :|
During the work I also fixed two bug:
- s->flags were not initialized for proto_uxst
- missing comma if throttling not enabled (caused by a stupid change in
"Implement persistent id for proxies and servers")
Other changes:
- No more magic type valuse, use STATS_TYPE_FE/STATS_TYPE_BE/STATS_TYPE_SV
- Don't memset full s->data_ctx (it was clearing s->data_ctx.stats.{iid/type/sid},
instead initialize stats.sv & stats.sv_st (stats.px and stats.px_st were already
initialized)
With all that changes it was extremely easy to write a short perl plugin
for a perl-enabled net-snmp (also included in this patch).
29385 is my PEN (Private Enterprise Number) and I'm willing to donate
the SNMPv2-SMI::enterprises.29385.106.* OIDs for HAProxy if there
is nothing assigned already.
2008-03-02 01:42:14 +00:00
|
|
|
|
2009-03-21 17:13:21 +00:00
|
|
|
if ((t = task_new()) == NULL) {
|
2007-10-16 15:34:28 +00:00
|
|
|
Alert("out of memory in uxst_event_accept().\n");
|
2008-12-01 00:44:25 +00:00
|
|
|
goto out_free_session;
|
2007-10-16 15:34:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
s->cli_addr = addr;
|
|
|
|
|
|
|
|
/* FIXME: should be checked earlier */
|
|
|
|
if (cfd >= global.maxsock) {
|
|
|
|
Alert("accept(): not enough free sockets. Raise -n argument. Giving up.\n");
|
2008-12-01 00:44:25 +00:00
|
|
|
goto out_free_task;
|
2007-10-16 15:34:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
if (fcntl(cfd, F_SETFL, O_NONBLOCK) == -1) {
|
|
|
|
Alert("accept(): cannot set the socket in non blocking mode. Giving up\n");
|
2008-12-01 00:44:25 +00:00
|
|
|
goto out_free_task;
|
2007-10-16 15:34:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
t->process = l->handler;
|
|
|
|
t->context = s;
|
2009-08-16 17:12:36 +00:00
|
|
|
t->nice = l->nice;
|
2007-10-16 15:34:28 +00:00
|
|
|
|
|
|
|
s->task = t;
|
2008-12-07 15:45:10 +00:00
|
|
|
s->listener = l;
|
2009-08-16 15:41:45 +00:00
|
|
|
s->fe = s->be = l->private;
|
2007-10-16 15:34:28 +00:00
|
|
|
|
2008-12-07 15:06:43 +00:00
|
|
|
s->ana_state = 0;
|
2007-10-16 15:34:28 +00:00
|
|
|
s->req = s->rep = NULL; /* will be allocated later */
|
|
|
|
|
2008-12-01 00:44:25 +00:00
|
|
|
s->si[0].state = s->si[0].prev_state = SI_ST_EST;
|
|
|
|
s->si[0].err_type = SI_ET_NONE;
|
|
|
|
s->si[0].err_loc = NULL;
|
|
|
|
s->si[0].owner = t;
|
2009-08-16 12:02:45 +00:00
|
|
|
s->si[0].connect = NULL;
|
2008-12-01 00:44:25 +00:00
|
|
|
s->si[0].shutr = stream_sock_shutr;
|
|
|
|
s->si[0].shutw = stream_sock_shutw;
|
2008-12-14 13:42:35 +00:00
|
|
|
s->si[0].chk_rcv = stream_sock_chk_rcv;
|
|
|
|
s->si[0].chk_snd = stream_sock_chk_snd;
|
2008-12-01 00:44:25 +00:00
|
|
|
s->si[0].fd = cfd;
|
|
|
|
s->si[0].flags = SI_FL_NONE;
|
|
|
|
s->si[0].exp = TICK_ETERNITY;
|
|
|
|
|
|
|
|
s->si[1].state = s->si[1].prev_state = SI_ST_INI;
|
|
|
|
s->si[1].err_type = SI_ET_NONE;
|
|
|
|
s->si[1].err_loc = NULL;
|
|
|
|
s->si[1].owner = t;
|
2009-08-16 12:02:45 +00:00
|
|
|
s->si[1].connect = NULL;
|
2008-12-01 00:44:25 +00:00
|
|
|
s->si[1].shutr = stream_sock_shutr;
|
|
|
|
s->si[1].shutw = stream_sock_shutw;
|
2008-12-14 13:42:35 +00:00
|
|
|
s->si[1].chk_rcv = stream_sock_chk_rcv;
|
|
|
|
s->si[1].chk_snd = stream_sock_chk_snd;
|
2008-12-01 00:44:25 +00:00
|
|
|
s->si[1].exp = TICK_ETERNITY;
|
|
|
|
s->si[1].fd = -1; /* just to help with debugging */
|
|
|
|
s->si[1].flags = SI_FL_NONE;
|
|
|
|
|
|
|
|
s->srv = s->prev_srv = s->srv_conn = NULL;
|
2007-10-16 15:34:28 +00:00
|
|
|
s->pend_pos = NULL;
|
|
|
|
|
|
|
|
memset(&s->logs, 0, sizeof(s->logs));
|
|
|
|
memset(&s->txn, 0, sizeof(s->txn));
|
|
|
|
|
2008-12-07 21:29:48 +00:00
|
|
|
s->logs.tv_accept = now; /* corrected date for internal use */
|
|
|
|
|
2007-10-17 16:57:38 +00:00
|
|
|
s->data_state = DATA_ST_INIT;
|
2007-10-16 15:34:28 +00:00
|
|
|
s->data_source = DATA_SRC_NONE;
|
|
|
|
s->uniq_id = totalconn;
|
|
|
|
|
2008-12-01 00:44:25 +00:00
|
|
|
if ((s->req = pool_alloc2(pool2_buffer)) == NULL)
|
|
|
|
goto out_free_task;
|
2007-10-16 15:34:28 +00:00
|
|
|
|
2008-12-07 15:06:43 +00:00
|
|
|
buffer_init(s->req);
|
2008-12-01 00:44:25 +00:00
|
|
|
s->req->prod = &s->si[0];
|
|
|
|
s->req->cons = &s->si[1];
|
|
|
|
s->si[0].ib = s->si[1].ob = s->req;
|
|
|
|
s->req->flags |= BF_READ_ATTACHED; /* the producer is already connected */
|
2009-03-21 20:10:04 +00:00
|
|
|
s->req->flags |= BF_READ_DONTWAIT; /* we plan to read small requests */
|
2008-12-01 00:44:25 +00:00
|
|
|
|
2008-12-07 15:06:43 +00:00
|
|
|
s->req->analysers = l->analysers;
|
2008-12-01 00:44:25 +00:00
|
|
|
|
|
|
|
s->req->wto = TICK_ETERNITY;
|
|
|
|
s->req->cto = TICK_ETERNITY;
|
|
|
|
s->req->rto = TICK_ETERNITY;
|
|
|
|
|
|
|
|
if ((s->rep = pool_alloc2(pool2_buffer)) == NULL)
|
|
|
|
goto out_free_req;
|
2007-10-16 15:34:28 +00:00
|
|
|
|
|
|
|
buffer_init(s->rep);
|
|
|
|
|
2008-12-01 00:44:25 +00:00
|
|
|
s->rep->prod = &s->si[1];
|
|
|
|
s->rep->cons = &s->si[0];
|
|
|
|
s->si[0].ob = s->si[1].ib = s->rep;
|
|
|
|
|
|
|
|
s->rep->rto = TICK_ETERNITY;
|
|
|
|
s->rep->cto = TICK_ETERNITY;
|
|
|
|
s->rep->wto = TICK_ETERNITY;
|
2007-10-16 15:34:28 +00:00
|
|
|
|
2008-07-06 22:09:58 +00:00
|
|
|
s->req->rex = TICK_ETERNITY;
|
|
|
|
s->req->wex = TICK_ETERNITY;
|
2008-08-17 16:03:28 +00:00
|
|
|
s->req->analyse_exp = TICK_ETERNITY;
|
2008-07-06 22:09:58 +00:00
|
|
|
s->rep->rex = TICK_ETERNITY;
|
|
|
|
s->rep->wex = TICK_ETERNITY;
|
2008-08-17 16:03:28 +00:00
|
|
|
s->rep->analyse_exp = TICK_ETERNITY;
|
2007-10-16 15:34:28 +00:00
|
|
|
|
2008-12-01 00:44:25 +00:00
|
|
|
t->expire = TICK_ETERNITY;
|
2007-10-16 15:34:28 +00:00
|
|
|
|
2008-12-01 00:44:25 +00:00
|
|
|
if (l->timeout) {
|
2007-10-16 15:34:28 +00:00
|
|
|
s->req->rto = *l->timeout;
|
|
|
|
s->rep->wto = *l->timeout;
|
|
|
|
}
|
|
|
|
|
2008-12-01 00:44:25 +00:00
|
|
|
fd_insert(cfd);
|
|
|
|
fdtab[cfd].owner = &s->si[0];
|
|
|
|
fdtab[cfd].state = FD_STREADY;
|
|
|
|
fdtab[cfd].cb[DIR_RD].f = l->proto->read;
|
|
|
|
fdtab[cfd].cb[DIR_RD].b = s->req;
|
|
|
|
fdtab[cfd].cb[DIR_WR].f = l->proto->write;
|
|
|
|
fdtab[cfd].cb[DIR_WR].b = s->rep;
|
|
|
|
fdtab[cfd].peeraddr = (struct sockaddr *)&s->cli_addr;
|
|
|
|
fdtab[cfd].peerlen = sizeof(s->cli_addr);
|
|
|
|
|
|
|
|
EV_FD_SET(cfd, DIR_RD);
|
|
|
|
|
2008-08-29 16:19:04 +00:00
|
|
|
task_wakeup(t, TASK_WOKEN_INIT);
|
2007-10-16 15:34:28 +00:00
|
|
|
|
|
|
|
l->nbconn++; /* warning! right now, it's up to the handler to decrease this */
|
|
|
|
if (l->nbconn >= l->maxconn) {
|
|
|
|
EV_FD_CLR(l->fd, DIR_RD);
|
|
|
|
l->state = LI_FULL;
|
|
|
|
}
|
|
|
|
actconn++;
|
|
|
|
totalconn++;
|
2008-12-07 15:06:43 +00:00
|
|
|
}
|
2007-10-16 15:34:28 +00:00
|
|
|
return 0;
|
2008-12-01 00:44:25 +00:00
|
|
|
|
|
|
|
out_free_req:
|
|
|
|
pool_free2(pool2_buffer, s->req);
|
|
|
|
out_free_task:
|
2009-03-21 17:13:21 +00:00
|
|
|
task_free(t);
|
2008-12-01 00:44:25 +00:00
|
|
|
out_free_session:
|
|
|
|
LIST_DEL(&s->list);
|
|
|
|
pool_free2(pool2_session, s);
|
|
|
|
out_close:
|
|
|
|
close(cfd);
|
|
|
|
return 0;
|
2007-10-16 15:34:28 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
__attribute__((constructor))
|
|
|
|
static void __uxst_protocol_init(void)
|
|
|
|
{
|
|
|
|
protocol_register(&proto_unix);
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
* Local variables:
|
|
|
|
* c-indent-level: 8
|
|
|
|
* c-basic-offset: 8
|
|
|
|
* End:
|
|
|
|
*/
|