| /* |
| * Listener management functions. |
| * |
| * Copyright 2000-2013 Willy Tarreau <w@1wt.eu> |
| * |
| * 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. |
| * |
| */ |
| |
| #define _GNU_SOURCE |
| #include <ctype.h> |
| #include <errno.h> |
| #include <stdio.h> |
| #include <string.h> |
| #include <unistd.h> |
| #include <fcntl.h> |
| |
| #include <common/accept4.h> |
| #include <common/config.h> |
| #include <common/errors.h> |
| #include <common/mini-clist.h> |
| #include <common/standard.h> |
| #include <common/time.h> |
| |
| #include <types/global.h> |
| #include <types/protocol.h> |
| |
| #include <proto/acl.h> |
| #include <proto/fd.h> |
| #include <proto/freq_ctr.h> |
| #include <proto/log.h> |
| #include <proto/listener.h> |
| #include <proto/protocol.h> |
| #include <proto/sample.h> |
| #include <proto/stream.h> |
| #include <proto/task.h> |
| |
| /* List head of all known bind keywords */ |
| static struct bind_kw_list bind_keywords = { |
| .list = LIST_HEAD_INIT(bind_keywords.list) |
| }; |
| |
| struct xfer_sock_list *xfer_sock_list = NULL; |
| |
| /* This function adds the specified listener's file descriptor to the polling |
| * lists if it is in the LI_LISTEN state. The listener enters LI_READY or |
| * LI_FULL state depending on its number of connections. In deamon mode, we |
| * also support binding only the relevant processes to their respective |
| * listeners. We don't do that in debug mode however. |
| */ |
| static void enable_listener(struct listener *listener) |
| { |
| if (listener->state == LI_LISTEN) { |
| if ((global.mode & (MODE_DAEMON | MODE_MWORKER)) && |
| listener->bind_conf->bind_proc && |
| !(listener->bind_conf->bind_proc & (1UL << (relative_pid - 1)))) { |
| /* we don't want to enable this listener and don't |
| * want any fd event to reach it. |
| */ |
| if (!(global.tune.options & GTUNE_SOCKET_TRANSFER)) |
| unbind_listener(listener); |
| else { |
| unbind_listener_no_close(listener); |
| listener->state = LI_LISTEN; |
| } |
| } |
| else if (listener->nbconn < listener->maxconn) { |
| fd_want_recv(listener->fd); |
| listener->state = LI_READY; |
| } |
| else { |
| listener->state = LI_FULL; |
| } |
| } |
| } |
| |
| /* This function removes the specified listener's file descriptor from the |
| * polling lists if it is in the LI_READY or in the LI_FULL state. The listener |
| * enters LI_LISTEN. |
| */ |
| static void disable_listener(struct listener *listener) |
| { |
| if (listener->state < LI_READY) |
| return; |
| if (listener->state == LI_READY) |
| fd_stop_recv(listener->fd); |
| if (listener->state == LI_LIMITED) |
| LIST_DEL(&listener->wait_queue); |
| listener->state = LI_LISTEN; |
| } |
| |
| /* This function tries to temporarily disable a listener, depending on the OS |
| * capabilities. Linux unbinds the listen socket after a SHUT_RD, and ignores |
| * SHUT_WR. Solaris refuses either shutdown(). OpenBSD ignores SHUT_RD but |
| * closes upon SHUT_WR and refuses to rebind. So a common validation path |
| * involves SHUT_WR && listen && SHUT_RD. In case of success, the FD's polling |
| * is disabled. It normally returns non-zero, unless an error is reported. |
| */ |
| int pause_listener(struct listener *l) |
| { |
| if (l->state <= LI_ZOMBIE) |
| return 1; |
| |
| if (l->proto->pause) { |
| /* Returns < 0 in case of failure, 0 if the listener |
| * was totally stopped, or > 0 if correctly paused. |
| */ |
| int ret = l->proto->pause(l); |
| |
| if (ret < 0) |
| return 0; |
| else if (ret == 0) |
| return 1; |
| } |
| |
| if (l->state == LI_LIMITED) |
| LIST_DEL(&l->wait_queue); |
| |
| fd_stop_recv(l->fd); |
| l->state = LI_PAUSED; |
| return 1; |
| } |
| |
| /* This function tries to resume a temporarily disabled listener. Paused, full, |
| * limited and disabled listeners are handled, which means that this function |
| * may replace enable_listener(). The resulting state will either be LI_READY |
| * or LI_FULL. 0 is returned in case of failure to resume (eg: dead socket). |
| * Listeners bound to a different process are not woken up unless we're in |
| * foreground mode, and are ignored. If the listener was only in the assigned |
| * state, it's totally rebound. This can happen if a pause() has completely |
| * stopped it. If the resume fails, 0 is returned and an error might be |
| * displayed. |
| */ |
| int resume_listener(struct listener *l) |
| { |
| if ((global.mode & (MODE_DAEMON | MODE_MWORKER)) && |
| l->bind_conf->bind_proc && |
| !(l->bind_conf->bind_proc & (1UL << (relative_pid - 1)))) |
| return 1; |
| |
| if (l->state == LI_ASSIGNED) { |
| char msg[100]; |
| int err; |
| |
| err = l->proto->bind(l, msg, sizeof(msg)); |
| if (err & ERR_ALERT) |
| Alert("Resuming listener: %s\n", msg); |
| else if (err & ERR_WARN) |
| Warning("Resuming listener: %s\n", msg); |
| |
| if (err & (ERR_FATAL | ERR_ABORT)) |
| return 0; |
| } |
| |
| if (l->state < LI_PAUSED || l->state == LI_ZOMBIE) |
| return 0; |
| |
| if (l->proto->sock_prot == IPPROTO_TCP && |
| l->state == LI_PAUSED && |
| listen(l->fd, l->backlog ? l->backlog : l->maxconn) != 0) |
| return 0; |
| |
| if (l->state == LI_READY) |
| return 1; |
| |
| if (l->state == LI_LIMITED) |
| LIST_DEL(&l->wait_queue); |
| |
| if (l->nbconn >= l->maxconn) { |
| l->state = LI_FULL; |
| return 1; |
| } |
| |
| fd_want_recv(l->fd); |
| l->state = LI_READY; |
| return 1; |
| } |
| |
| /* Marks a ready listener as full so that the stream code tries to re-enable |
| * it upon next close() using resume_listener(). |
| */ |
| static void listener_full(struct listener *l) |
| { |
| if (l->state >= LI_READY) { |
| if (l->state == LI_LIMITED) |
| LIST_DEL(&l->wait_queue); |
| |
| fd_stop_recv(l->fd); |
| l->state = LI_FULL; |
| } |
| } |
| |
| /* Marks a ready listener as limited so that we only try to re-enable it when |
| * resources are free again. It will be queued into the specified queue. |
| */ |
| static void limit_listener(struct listener *l, struct list *list) |
| { |
| if (l->state == LI_READY) { |
| LIST_ADDQ(list, &l->wait_queue); |
| fd_stop_recv(l->fd); |
| l->state = LI_LIMITED; |
| } |
| } |
| |
| /* This function adds all of the protocol's listener's file descriptors to the |
| * polling lists when they are in the LI_LISTEN state. It is intended to be |
| * used as a protocol's generic enable_all() primitive, for use after the |
| * fork(). It puts the listeners into LI_READY or LI_FULL states depending on |
| * their number of connections. It always returns ERR_NONE. |
| */ |
| int enable_all_listeners(struct protocol *proto) |
| { |
| struct listener *listener; |
| |
| list_for_each_entry(listener, &proto->listeners, proto_list) |
| enable_listener(listener); |
| return ERR_NONE; |
| } |
| |
| /* This function removes all of the protocol's listener's file descriptors from |
| * the polling lists when they are in the LI_READY or LI_FULL states. It is |
| * intended to be used as a protocol's generic disable_all() primitive. It puts |
| * the listeners into LI_LISTEN, and always returns ERR_NONE. |
| */ |
| int disable_all_listeners(struct protocol *proto) |
| { |
| struct listener *listener; |
| |
| list_for_each_entry(listener, &proto->listeners, proto_list) |
| disable_listener(listener); |
| return ERR_NONE; |
| } |
| |
| /* Dequeues all of the listeners waiting for a resource in wait queue <queue>. */ |
| void dequeue_all_listeners(struct list *list) |
| { |
| struct listener *listener, *l_back; |
| |
| list_for_each_entry_safe(listener, l_back, list, wait_queue) { |
| /* This cannot fail because the listeners are by definition in |
| * the LI_LIMITED state. The function also removes the entry |
| * from the queue. |
| */ |
| resume_listener(listener); |
| } |
| } |
| |
| static int do_unbind_listener(struct listener *listener, int do_close) |
| { |
| if (listener->state == LI_READY) |
| fd_stop_recv(listener->fd); |
| |
| if (listener->state == LI_LIMITED) |
| LIST_DEL(&listener->wait_queue); |
| |
| if (listener->state >= LI_PAUSED) { |
| if (do_close) { |
| fd_delete(listener->fd); |
| listener->fd = -1; |
| } |
| else |
| fd_remove(listener->fd); |
| listener->state = LI_ASSIGNED; |
| } |
| return ERR_NONE; |
| } |
| |
| /* This function closes the listening socket for the specified listener, |
| * provided that it's already in a listening state. The listener enters the |
| * LI_ASSIGNED state. It always returns ERR_NONE. This function is intended |
| * to be used as a generic function for standard protocols. |
| */ |
| int unbind_listener(struct listener *listener) |
| { |
| return do_unbind_listener(listener, 1); |
| } |
| |
| /* This function pretends the listener is dead, but keeps the FD opened, so |
| * that we can provide it, for conf reloading. |
| */ |
| int unbind_listener_no_close(struct listener *listener) |
| { |
| return do_unbind_listener(listener, 0); |
| } |
| |
| /* This function closes all listening sockets bound to the protocol <proto>, |
| * and the listeners end in LI_ASSIGNED state if they were higher. It does not |
| * detach them from the protocol. It always returns ERR_NONE. |
| */ |
| int unbind_all_listeners(struct protocol *proto) |
| { |
| struct listener *listener; |
| |
| list_for_each_entry(listener, &proto->listeners, proto_list) |
| unbind_listener(listener); |
| return ERR_NONE; |
| } |
| |
| /* creates one or multiple listeners for bind_conf <bc> on sockaddr <ss> on port |
| * range <portl> to <porth>, and possibly attached to fd <fd> (or -1 for auto |
| * allocation). The address family is taken from ss->ss_family. The number of |
| * jobs and listeners is automatically increased by the number of listeners |
| * created. It returns non-zero on success, zero on error with the error message |
| * set in <err>. |
| */ |
| int create_listeners(struct bind_conf *bc, const struct sockaddr_storage *ss, |
| int portl, int porth, int fd, char **err) |
| { |
| struct protocol *proto = protocol_by_family(ss->ss_family); |
| struct listener *l; |
| int port; |
| |
| if (!proto) { |
| memprintf(err, "unsupported protocol family %d", ss->ss_family); |
| return 0; |
| } |
| |
| for (port = portl; port <= porth; port++) { |
| l = calloc(1, sizeof(*l)); |
| if (!l) { |
| memprintf(err, "out of memory"); |
| return 0; |
| } |
| l->obj_type = OBJ_TYPE_LISTENER; |
| LIST_ADDQ(&bc->frontend->conf.listeners, &l->by_fe); |
| LIST_ADDQ(&bc->listeners, &l->by_bind); |
| l->bind_conf = bc; |
| |
| l->fd = fd; |
| memcpy(&l->addr, ss, sizeof(*ss)); |
| l->state = LI_INIT; |
| |
| proto->add(l, port); |
| |
| jobs++; |
| listeners++; |
| } |
| return 1; |
| } |
| |
| /* Delete a listener from its protocol's list of listeners. The listener's |
| * state is automatically updated from LI_ASSIGNED to LI_INIT. The protocol's |
| * number of listeners is updated, as well as the global number of listeners |
| * and jobs. Note that the listener must have previously been unbound. This |
| * is the generic function to use to remove a listener. |
| */ |
| void delete_listener(struct listener *listener) |
| { |
| if (listener->state != LI_ASSIGNED) |
| return; |
| listener->state = LI_INIT; |
| LIST_DEL(&listener->proto_list); |
| listener->proto->nb_listeners--; |
| listeners--; |
| jobs--; |
| } |
| |
| /* This function is called on a read event from a listening socket, corresponding |
| * to an accept. It tries to accept as many connections as possible, and for each |
| * calls the listener's accept handler (generally the frontend's accept handler). |
| */ |
| void listener_accept(int fd) |
| { |
| struct listener *l = fdtab[fd].owner; |
| struct proxy *p = l->bind_conf->frontend; |
| int max_accept = l->maxaccept ? l->maxaccept : 1; |
| int expire; |
| int cfd; |
| int ret; |
| #ifdef USE_ACCEPT4 |
| static int accept4_broken; |
| #endif |
| |
| if (unlikely(l->nbconn >= l->maxconn)) { |
| listener_full(l); |
| return; |
| } |
| |
| if (!(l->options & LI_O_UNLIMITED) && global.sps_lim) { |
| int max = freq_ctr_remain(&global.sess_per_sec, global.sps_lim, 0); |
| |
| if (unlikely(!max)) { |
| /* frontend accept rate limit was reached */ |
| expire = tick_add(now_ms, next_event_delay(&global.sess_per_sec, global.sps_lim, 0)); |
| goto wait_expire; |
| } |
| |
| if (max_accept > max) |
| max_accept = max; |
| } |
| |
| if (!(l->options & LI_O_UNLIMITED) && global.cps_lim) { |
| int max = freq_ctr_remain(&global.conn_per_sec, global.cps_lim, 0); |
| |
| if (unlikely(!max)) { |
| /* frontend accept rate limit was reached */ |
| expire = tick_add(now_ms, next_event_delay(&global.conn_per_sec, global.cps_lim, 0)); |
| goto wait_expire; |
| } |
| |
| if (max_accept > max) |
| max_accept = max; |
| } |
| #ifdef USE_OPENSSL |
| if (!(l->options & LI_O_UNLIMITED) && global.ssl_lim && l->bind_conf && l->bind_conf->is_ssl) { |
| int max = freq_ctr_remain(&global.ssl_per_sec, global.ssl_lim, 0); |
| |
| if (unlikely(!max)) { |
| /* frontend accept rate limit was reached */ |
| expire = tick_add(now_ms, next_event_delay(&global.ssl_per_sec, global.ssl_lim, 0)); |
| goto wait_expire; |
| } |
| |
| if (max_accept > max) |
| max_accept = max; |
| } |
| #endif |
| if (p && p->fe_sps_lim) { |
| int max = freq_ctr_remain(&p->fe_sess_per_sec, p->fe_sps_lim, 0); |
| |
| if (unlikely(!max)) { |
| /* frontend accept rate limit was reached */ |
| limit_listener(l, &p->listener_queue); |
| task_schedule(p->task, tick_add(now_ms, next_event_delay(&p->fe_sess_per_sec, p->fe_sps_lim, 0))); |
| return; |
| } |
| |
| if (max_accept > max) |
| max_accept = max; |
| } |
| |
| /* Note: if we fail to allocate a connection because of configured |
| * limits, we'll schedule a new attempt worst 1 second later in the |
| * worst case. If we fail due to system limits or temporary resource |
| * shortage, we try again 100ms later in the worst case. |
| */ |
| while (max_accept--) { |
| struct sockaddr_storage addr; |
| socklen_t laddr = sizeof(addr); |
| |
| if (unlikely(actconn >= global.maxconn) && !(l->options & LI_O_UNLIMITED)) { |
| limit_listener(l, &global_listener_queue); |
| task_schedule(global_listener_queue_task, tick_add(now_ms, 1000)); /* try again in 1 second */ |
| return; |
| } |
| |
| if (unlikely(p && p->feconn >= p->maxconn)) { |
| limit_listener(l, &p->listener_queue); |
| return; |
| } |
| |
| #ifdef USE_ACCEPT4 |
| /* only call accept4() if it's known to be safe, otherwise |
| * fallback to the legacy accept() + fcntl(). |
| */ |
| if (unlikely(accept4_broken || |
| ((cfd = accept4(fd, (struct sockaddr *)&addr, &laddr, SOCK_NONBLOCK)) == -1 && |
| (errno == ENOSYS || errno == EINVAL || errno == EBADF) && |
| (accept4_broken = 1)))) |
| #endif |
| if ((cfd = accept(fd, (struct sockaddr *)&addr, &laddr)) != -1) |
| fcntl(cfd, F_SETFL, O_NONBLOCK); |
| |
| if (unlikely(cfd == -1)) { |
| switch (errno) { |
| case EAGAIN: |
| if (fdtab[fd].ev & FD_POLL_HUP) { |
| /* the listening socket might have been disabled in a shared |
| * process and we're a collateral victim. We'll just pause for |
| * a while in case it comes back. In the mean time, we need to |
| * clear this sticky flag. |
| */ |
| fdtab[fd].ev &= ~FD_POLL_HUP; |
| goto transient_error; |
| } |
| fd_cant_recv(fd); |
| return; /* nothing more to accept */ |
| case EINVAL: |
| /* might be trying to accept on a shut fd (eg: soft stop) */ |
| goto transient_error; |
| case EINTR: |
| case ECONNABORTED: |
| continue; |
| case ENFILE: |
| if (p) |
| send_log(p, LOG_EMERG, |
| "Proxy %s reached system FD limit at %d. Please check system tunables.\n", |
| p->id, maxfd); |
| goto transient_error; |
| case EMFILE: |
| if (p) |
| send_log(p, LOG_EMERG, |
| "Proxy %s reached process FD limit at %d. Please check 'ulimit-n' and restart.\n", |
| p->id, maxfd); |
| goto transient_error; |
| case ENOBUFS: |
| case ENOMEM: |
| if (p) |
| send_log(p, LOG_EMERG, |
| "Proxy %s reached system memory limit at %d sockets. Please check system tunables.\n", |
| p->id, maxfd); |
| goto transient_error; |
| default: |
| /* unexpected result, let's give up and let other tasks run */ |
| goto stop; |
| } |
| } |
| |
| if (unlikely(cfd >= global.maxsock)) { |
| send_log(p, LOG_EMERG, |
| "Proxy %s reached the configured maximum connection limit. Please check the global 'maxconn' value.\n", |
| p->id); |
| close(cfd); |
| limit_listener(l, &global_listener_queue); |
| task_schedule(global_listener_queue_task, tick_add(now_ms, 1000)); /* try again in 1 second */ |
| return; |
| } |
| |
| /* increase the per-process number of cumulated connections */ |
| if (!(l->options & LI_O_UNLIMITED)) { |
| update_freq_ctr(&global.conn_per_sec, 1); |
| if (global.conn_per_sec.curr_ctr > global.cps_max) |
| global.cps_max = global.conn_per_sec.curr_ctr; |
| actconn++; |
| } |
| |
| totalconn++; |
| l->nbconn++; |
| |
| if (l->counters) { |
| if (l->nbconn > l->counters->conn_max) |
| l->counters->conn_max = l->nbconn; |
| } |
| |
| ret = l->accept(l, cfd, &addr); |
| if (unlikely(ret <= 0)) { |
| /* The connection was closed by stream_accept(). Either |
| * we just have to ignore it (ret == 0) or it's a critical |
| * error due to a resource shortage, and we must stop the |
| * listener (ret < 0). |
| */ |
| if (!(l->options & LI_O_UNLIMITED)) |
| actconn--; |
| l->nbconn--; |
| if (ret == 0) /* successful termination */ |
| continue; |
| |
| goto transient_error; |
| } |
| |
| if (l->nbconn >= l->maxconn) { |
| listener_full(l); |
| return; |
| } |
| |
| /* increase the per-process number of cumulated connections */ |
| if (!(l->options & LI_O_UNLIMITED)) { |
| update_freq_ctr(&global.sess_per_sec, 1); |
| if (global.sess_per_sec.curr_ctr > global.sps_max) |
| global.sps_max = global.sess_per_sec.curr_ctr; |
| } |
| #ifdef USE_OPENSSL |
| if (!(l->options & LI_O_UNLIMITED) && l->bind_conf && l->bind_conf->is_ssl) { |
| |
| update_freq_ctr(&global.ssl_per_sec, 1); |
| if (global.ssl_per_sec.curr_ctr > global.ssl_max) |
| global.ssl_max = global.ssl_per_sec.curr_ctr; |
| } |
| #endif |
| |
| } /* end of while (max_accept--) */ |
| |
| /* we've exhausted max_accept, so there is no need to poll again */ |
| stop: |
| fd_done_recv(fd); |
| return; |
| |
| transient_error: |
| /* pause the listener and try again in 100 ms */ |
| expire = tick_add(now_ms, 100); |
| |
| wait_expire: |
| limit_listener(l, &global_listener_queue); |
| task_schedule(global_listener_queue_task, tick_first(expire, global_listener_queue_task->expire)); |
| return; |
| } |
| |
| /* Notify the listener that a connection initiated from it was released. This |
| * is used to keep the connection count consistent and to possibly re-open |
| * listening when it was limited. |
| */ |
| void listener_release(struct listener *l) |
| { |
| struct proxy *fe = l->bind_conf->frontend; |
| |
| if (!(l->options & LI_O_UNLIMITED)) |
| actconn--; |
| l->nbconn--; |
| if (l->state == LI_FULL) |
| resume_listener(l); |
| |
| /* Dequeues all of the listeners waiting for a resource */ |
| if (!LIST_ISEMPTY(&global_listener_queue)) |
| dequeue_all_listeners(&global_listener_queue); |
| |
| if (!LIST_ISEMPTY(&fe->listener_queue) && |
| (!fe->fe_sps_lim || freq_ctr_remain(&fe->fe_sess_per_sec, fe->fe_sps_lim, 0) > 0)) |
| dequeue_all_listeners(&fe->listener_queue); |
| } |
| |
| /* |
| * Registers the bind keyword list <kwl> as a list of valid keywords for next |
| * parsing sessions. |
| */ |
| void bind_register_keywords(struct bind_kw_list *kwl) |
| { |
| LIST_ADDQ(&bind_keywords.list, &kwl->list); |
| } |
| |
| /* Return a pointer to the bind keyword <kw>, or NULL if not found. If the |
| * keyword is found with a NULL ->parse() function, then an attempt is made to |
| * find one with a valid ->parse() function. This way it is possible to declare |
| * platform-dependant, known keywords as NULL, then only declare them as valid |
| * if some options are met. Note that if the requested keyword contains an |
| * opening parenthesis, everything from this point is ignored. |
| */ |
| struct bind_kw *bind_find_kw(const char *kw) |
| { |
| int index; |
| const char *kwend; |
| struct bind_kw_list *kwl; |
| struct bind_kw *ret = NULL; |
| |
| kwend = strchr(kw, '('); |
| if (!kwend) |
| kwend = kw + strlen(kw); |
| |
| list_for_each_entry(kwl, &bind_keywords.list, list) { |
| for (index = 0; kwl->kw[index].kw != NULL; index++) { |
| if ((strncmp(kwl->kw[index].kw, kw, kwend - kw) == 0) && |
| kwl->kw[index].kw[kwend-kw] == 0) { |
| if (kwl->kw[index].parse) |
| return &kwl->kw[index]; /* found it !*/ |
| else |
| ret = &kwl->kw[index]; /* may be OK */ |
| } |
| } |
| } |
| return ret; |
| } |
| |
| /* Dumps all registered "bind" keywords to the <out> string pointer. The |
| * unsupported keywords are only dumped if their supported form was not |
| * found. |
| */ |
| void bind_dump_kws(char **out) |
| { |
| struct bind_kw_list *kwl; |
| int index; |
| |
| *out = NULL; |
| list_for_each_entry(kwl, &bind_keywords.list, list) { |
| for (index = 0; kwl->kw[index].kw != NULL; index++) { |
| if (kwl->kw[index].parse || |
| bind_find_kw(kwl->kw[index].kw) == &kwl->kw[index]) { |
| memprintf(out, "%s[%4s] %s%s%s\n", *out ? *out : "", |
| kwl->scope, |
| kwl->kw[index].kw, |
| kwl->kw[index].skip ? " <arg>" : "", |
| kwl->kw[index].parse ? "" : " (not supported)"); |
| } |
| } |
| } |
| } |
| |
| /************************************************************************/ |
| /* All supported sample and ACL keywords must be declared here. */ |
| /************************************************************************/ |
| |
| /* set temp integer to the number of connexions to the same listening socket */ |
| static int |
| smp_fetch_dconn(const struct arg *args, struct sample *smp, const char *kw, void *private) |
| { |
| smp->data.type = SMP_T_SINT; |
| smp->data.u.sint = smp->sess->listener->nbconn; |
| return 1; |
| } |
| |
| /* set temp integer to the id of the socket (listener) */ |
| static int |
| smp_fetch_so_id(const struct arg *args, struct sample *smp, const char *kw, void *private) |
| { |
| smp->data.type = SMP_T_SINT; |
| smp->data.u.sint = smp->sess->listener->luid; |
| return 1; |
| } |
| |
| /* parse the "accept-proxy" bind keyword */ |
| static int bind_parse_accept_proxy(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err) |
| { |
| struct listener *l; |
| |
| list_for_each_entry(l, &conf->listeners, by_bind) |
| l->options |= LI_O_ACC_PROXY; |
| |
| return 0; |
| } |
| |
| /* parse the "accept-netscaler-cip" bind keyword */ |
| static int bind_parse_accept_netscaler_cip(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err) |
| { |
| struct listener *l; |
| uint32_t val; |
| |
| if (!*args[cur_arg + 1]) { |
| memprintf(err, "'%s' : missing value", args[cur_arg]); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| val = atol(args[cur_arg + 1]); |
| if (val <= 0) { |
| memprintf(err, "'%s' : invalid value %d, must be > 0", args[cur_arg], val); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| list_for_each_entry(l, &conf->listeners, by_bind) { |
| l->options |= LI_O_ACC_CIP; |
| conf->ns_cip_magic = val; |
| } |
| |
| return 0; |
| } |
| |
| /* parse the "backlog" bind keyword */ |
| static int bind_parse_backlog(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err) |
| { |
| struct listener *l; |
| int val; |
| |
| if (!*args[cur_arg + 1]) { |
| memprintf(err, "'%s' : missing value", args[cur_arg]); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| val = atol(args[cur_arg + 1]); |
| if (val <= 0) { |
| memprintf(err, "'%s' : invalid value %d, must be > 0", args[cur_arg], val); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| list_for_each_entry(l, &conf->listeners, by_bind) |
| l->backlog = val; |
| |
| return 0; |
| } |
| |
| /* parse the "id" bind keyword */ |
| static int bind_parse_id(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err) |
| { |
| struct eb32_node *node; |
| struct listener *l, *new; |
| char *error; |
| |
| if (conf->listeners.n != conf->listeners.p) { |
| memprintf(err, "'%s' can only be used with a single socket", args[cur_arg]); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| if (!*args[cur_arg + 1]) { |
| memprintf(err, "'%s' : expects an integer argument", args[cur_arg]); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| new = LIST_NEXT(&conf->listeners, struct listener *, by_bind); |
| new->luid = strtol(args[cur_arg + 1], &error, 10); |
| if (*error != '\0') { |
| memprintf(err, "'%s' : expects an integer argument, found '%s'", args[cur_arg], args[cur_arg + 1]); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| new->conf.id.key = new->luid; |
| |
| if (new->luid <= 0) { |
| memprintf(err, "'%s' : custom id has to be > 0", args[cur_arg]); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| node = eb32_lookup(&px->conf.used_listener_id, new->luid); |
| if (node) { |
| l = container_of(node, struct listener, conf.id); |
| memprintf(err, "'%s' : custom id %d already used at %s:%d ('bind %s')", |
| args[cur_arg], l->luid, l->bind_conf->file, l->bind_conf->line, |
| l->bind_conf->arg); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| eb32_insert(&px->conf.used_listener_id, &new->conf.id); |
| return 0; |
| } |
| |
| /* parse the "maxconn" bind keyword */ |
| static int bind_parse_maxconn(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err) |
| { |
| struct listener *l; |
| int val; |
| |
| if (!*args[cur_arg + 1]) { |
| memprintf(err, "'%s' : missing value", args[cur_arg]); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| val = atol(args[cur_arg + 1]); |
| if (val <= 0) { |
| memprintf(err, "'%s' : invalid value %d, must be > 0", args[cur_arg], val); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| list_for_each_entry(l, &conf->listeners, by_bind) |
| l->maxconn = val; |
| |
| return 0; |
| } |
| |
| /* parse the "name" bind keyword */ |
| static int bind_parse_name(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err) |
| { |
| struct listener *l; |
| |
| if (!*args[cur_arg + 1]) { |
| memprintf(err, "'%s' : missing name", args[cur_arg]); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| list_for_each_entry(l, &conf->listeners, by_bind) |
| l->name = strdup(args[cur_arg + 1]); |
| |
| return 0; |
| } |
| |
| /* parse the "nice" bind keyword */ |
| static int bind_parse_nice(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err) |
| { |
| struct listener *l; |
| int val; |
| |
| if (!*args[cur_arg + 1]) { |
| memprintf(err, "'%s' : missing value", args[cur_arg]); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| val = atol(args[cur_arg + 1]); |
| if (val < -1024 || val > 1024) { |
| memprintf(err, "'%s' : invalid value %d, allowed range is -1024..1024", args[cur_arg], val); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| list_for_each_entry(l, &conf->listeners, by_bind) |
| l->nice = val; |
| |
| return 0; |
| } |
| |
| /* parse the "process" bind keyword */ |
| static int bind_parse_process(char **args, int cur_arg, struct proxy *px, struct bind_conf *conf, char **err) |
| { |
| unsigned long set = 0; |
| unsigned int low, high; |
| |
| if (strcmp(args[cur_arg + 1], "all") == 0) { |
| set = 0; |
| } |
| else if (strcmp(args[cur_arg + 1], "odd") == 0) { |
| set |= ~0UL/3UL; /* 0x555....555 */ |
| } |
| else if (strcmp(args[cur_arg + 1], "even") == 0) { |
| set |= (~0UL/3UL) << 1; /* 0xAAA...AAA */ |
| } |
| else if (isdigit((int)*args[cur_arg + 1])) { |
| char *dash = strchr(args[cur_arg + 1], '-'); |
| |
| low = high = str2uic(args[cur_arg + 1]); |
| if (dash) |
| high = str2uic(dash + 1); |
| |
| if (high < low) { |
| unsigned int swap = low; |
| low = high; |
| high = swap; |
| } |
| |
| if (low < 1 || high > LONGBITS) { |
| memprintf(err, "'%s' : invalid range %d-%d, allowed range is 1..%d", args[cur_arg], low, high, LONGBITS); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| while (low <= high) |
| set |= 1UL << (low++ - 1); |
| } |
| else { |
| memprintf(err, "'%s' expects 'all', 'odd', 'even', or a process range with numbers from 1 to %d.", args[cur_arg], LONGBITS); |
| return ERR_ALERT | ERR_FATAL; |
| } |
| |
| conf->bind_proc = set; |
| return 0; |
| } |
| |
| /* Note: must not be declared <const> as its list will be overwritten. |
| * Please take care of keeping this list alphabetically sorted. |
| */ |
| static struct sample_fetch_kw_list smp_kws = {ILH, { |
| { "dst_conn", smp_fetch_dconn, 0, NULL, SMP_T_SINT, SMP_USE_FTEND, }, |
| { "so_id", smp_fetch_so_id, 0, NULL, SMP_T_SINT, SMP_USE_FTEND, }, |
| { /* END */ }, |
| }}; |
| |
| /* Note: must not be declared <const> as its list will be overwritten. |
| * Please take care of keeping this list alphabetically sorted. |
| */ |
| static struct acl_kw_list acl_kws = {ILH, { |
| { /* END */ }, |
| }}; |
| |
| /* Note: must not be declared <const> as its list will be overwritten. |
| * Please take care of keeping this list alphabetically sorted, doing so helps |
| * all code contributors. |
| * Optional keywords are also declared with a NULL ->parse() function so that |
| * the config parser can report an appropriate error when a known keyword was |
| * not enabled. |
| */ |
| static struct bind_kw_list bind_kws = { "ALL", { }, { |
| { "accept-netscaler-cip", bind_parse_accept_netscaler_cip, 1 }, /* enable NetScaler Client IP insertion protocol */ |
| { "accept-proxy", bind_parse_accept_proxy, 0 }, /* enable PROXY protocol */ |
| { "backlog", bind_parse_backlog, 1 }, /* set backlog of listening socket */ |
| { "id", bind_parse_id, 1 }, /* set id of listening socket */ |
| { "maxconn", bind_parse_maxconn, 1 }, /* set maxconn of listening socket */ |
| { "name", bind_parse_name, 1 }, /* set name of listening socket */ |
| { "nice", bind_parse_nice, 1 }, /* set nice of listening socket */ |
| { "process", bind_parse_process, 1 }, /* set list of allowed process for this socket */ |
| { /* END */ }, |
| }}; |
| |
| __attribute__((constructor)) |
| static void __listener_init(void) |
| { |
| sample_register_fetches(&smp_kws); |
| acl_register_keywords(&acl_kws); |
| bind_register_keywords(&bind_kws); |
| } |
| |
| /* |
| * Local variables: |
| * c-indent-level: 8 |
| * c-basic-offset: 8 |
| * End: |
| */ |