Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 1 | /* |
| 2 | * include/proto/applet.h |
| 3 | * This file contains applet function prototypes |
| 4 | * |
| 5 | * Copyright (C) 2000-2015 Willy Tarreau - w@1wt.eu |
| 6 | * |
| 7 | * This library is free software; you can redistribute it and/or |
| 8 | * modify it under the terms of the GNU Lesser General Public |
| 9 | * License as published by the Free Software Foundation, version 2.1 |
| 10 | * exclusively. |
| 11 | * |
| 12 | * This library is distributed in the hope that it will be useful, |
| 13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 15 | * Lesser General Public License for more details. |
| 16 | * |
| 17 | * You should have received a copy of the GNU Lesser General Public |
| 18 | * License along with this library; if not, write to the Free Software |
| 19 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 20 | */ |
| 21 | |
| 22 | #ifndef _PROTO_APPLET_H |
| 23 | #define _PROTO_APPLET_H |
| 24 | |
| 25 | #include <stdlib.h> |
| 26 | |
| 27 | #include <common/config.h> |
Willy Tarreau | 81f38d6 | 2015-04-13 17:11:11 +0200 | [diff] [blame] | 28 | #include <common/mini-clist.h> |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 29 | #include <types/applet.h> |
| 30 | #include <proto/connection.h> |
| 31 | |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 32 | extern unsigned int nb_applets; |
| 33 | extern unsigned int applets_active_queue; |
| 34 | |
Willy Tarreau | 64bca9d | 2015-09-25 17:39:23 +0200 | [diff] [blame] | 35 | extern struct list applet_active_queue; |
Willy Tarreau | 81f38d6 | 2015-04-13 17:11:11 +0200 | [diff] [blame] | 36 | |
Willy Tarreau | 3c595ac | 2015-04-19 09:59:31 +0200 | [diff] [blame] | 37 | void applet_run_active(); |
| 38 | |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 39 | |
| 40 | static int inline appctx_res_wakeup(struct appctx *appctx); |
| 41 | |
| 42 | |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 43 | /* Initializes all required fields for a new appctx. Note that it does the |
| 44 | * minimum acceptable initialization for an appctx. This means only the |
| 45 | * 3 integer states st0, st1, st2 are zeroed. |
| 46 | */ |
| 47 | static inline void appctx_init(struct appctx *appctx) |
| 48 | { |
| 49 | appctx->st0 = appctx->st1 = appctx->st2 = 0; |
Thierry FOURNIER / OZON.IO | 6a22dcb | 2016-11-12 10:51:33 +0100 | [diff] [blame] | 50 | appctx->io_release = NULL; |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 51 | } |
| 52 | |
| 53 | /* Tries to allocate a new appctx and initialize its main fields. The appctx |
| 54 | * is returned on success, NULL on failure. The appctx must be released using |
| 55 | * pool_free2(connection) or appctx_free(), since it's allocated from the |
| 56 | * connection pool. <applet> is assigned as the applet, but it can be NULL. |
| 57 | */ |
Willy Tarreau | 3057645 | 2015-04-13 13:50:30 +0200 | [diff] [blame] | 58 | static inline struct appctx *appctx_new(struct applet *applet) |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 59 | { |
| 60 | struct appctx *appctx; |
| 61 | |
| 62 | appctx = pool_alloc2(pool2_connection); |
| 63 | if (likely(appctx != NULL)) { |
| 64 | appctx->obj_type = OBJ_TYPE_APPCTX; |
| 65 | appctx->applet = applet; |
| 66 | appctx_init(appctx); |
Willy Tarreau | 81f38d6 | 2015-04-13 17:11:11 +0200 | [diff] [blame] | 67 | LIST_INIT(&appctx->runq); |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 68 | LIST_INIT(&appctx->buffer_wait.list); |
| 69 | appctx->buffer_wait.target = appctx; |
| 70 | appctx->buffer_wait.wakeup_cb = (int (*)(void *))appctx_res_wakeup; |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 71 | nb_applets++; |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 72 | } |
| 73 | return appctx; |
| 74 | } |
| 75 | |
| 76 | /* Releases an appctx previously allocated by appctx_new(). Note that |
| 77 | * we share the connection pool. |
| 78 | */ |
| 79 | static inline void appctx_free(struct appctx *appctx) |
| 80 | { |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 81 | if (!LIST_ISEMPTY(&appctx->runq)) { |
Willy Tarreau | 81f38d6 | 2015-04-13 17:11:11 +0200 | [diff] [blame] | 82 | LIST_DEL(&appctx->runq); |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 83 | applets_active_queue--; |
| 84 | } |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 85 | if (!LIST_ISEMPTY(&appctx->buffer_wait.list)) { |
| 86 | LIST_DEL(&appctx->buffer_wait.list); |
| 87 | LIST_INIT(&appctx->buffer_wait.list); |
| 88 | } |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 89 | pool_free2(pool2_connection, appctx); |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 90 | nb_applets--; |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 91 | } |
| 92 | |
Willy Tarreau | 81f38d6 | 2015-04-13 17:11:11 +0200 | [diff] [blame] | 93 | /* wakes up an applet when conditions have changed */ |
| 94 | static inline void appctx_wakeup(struct appctx *appctx) |
| 95 | { |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 96 | if (LIST_ISEMPTY(&appctx->runq)) { |
Willy Tarreau | 64bca9d | 2015-09-25 17:39:23 +0200 | [diff] [blame] | 97 | LIST_ADDQ(&applet_active_queue, &appctx->runq); |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 98 | applets_active_queue++; |
| 99 | } |
Willy Tarreau | 81f38d6 | 2015-04-13 17:11:11 +0200 | [diff] [blame] | 100 | } |
| 101 | |
Willy Tarreau | 3c595ac | 2015-04-19 09:59:31 +0200 | [diff] [blame] | 102 | /* removes an applet from the list of active applets */ |
| 103 | static inline void appctx_pause(struct appctx *appctx) |
| 104 | { |
| 105 | if (!LIST_ISEMPTY(&appctx->runq)) { |
| 106 | LIST_DEL(&appctx->runq); |
| 107 | LIST_INIT(&appctx->runq); |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 108 | applets_active_queue--; |
Willy Tarreau | 3c595ac | 2015-04-19 09:59:31 +0200 | [diff] [blame] | 109 | } |
| 110 | } |
| 111 | |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 112 | /* Callback used to wake up an applet when a buffer is available. The applet |
| 113 | * <appctx> is woken up is if it is not already in the list of "active" |
| 114 | * applets. This functions returns 1 is the stream is woken up, otherwise it |
| 115 | * returns 0. */ |
| 116 | static inline int appctx_res_wakeup(struct appctx *appctx) |
| 117 | { |
| 118 | if (!LIST_ISEMPTY(&appctx->runq)) |
| 119 | return 0; |
| 120 | appctx_wakeup(appctx); |
| 121 | return 1; |
| 122 | } |
| 123 | |
| 124 | |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 125 | #endif /* _PROTO_APPLET_H */ |
| 126 | |
| 127 | /* |
| 128 | * Local variables: |
| 129 | * c-indent-level: 8 |
| 130 | * c-basic-offset: 8 |
| 131 | * End: |
| 132 | */ |