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; |
Emeric Brun | c730606 | 2017-06-26 16:36:53 +0200 | [diff] [blame] | 51 | appctx->state = APPLET_SLEEPING; |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 52 | } |
| 53 | |
| 54 | /* Tries to allocate a new appctx and initialize its main fields. The appctx |
| 55 | * is returned on success, NULL on failure. The appctx must be released using |
| 56 | * pool_free2(connection) or appctx_free(), since it's allocated from the |
| 57 | * connection pool. <applet> is assigned as the applet, but it can be NULL. |
| 58 | */ |
Willy Tarreau | 3057645 | 2015-04-13 13:50:30 +0200 | [diff] [blame] | 59 | static inline struct appctx *appctx_new(struct applet *applet) |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 60 | { |
| 61 | struct appctx *appctx; |
| 62 | |
| 63 | appctx = pool_alloc2(pool2_connection); |
| 64 | if (likely(appctx != NULL)) { |
| 65 | appctx->obj_type = OBJ_TYPE_APPCTX; |
| 66 | appctx->applet = applet; |
| 67 | appctx_init(appctx); |
Willy Tarreau | 81f38d6 | 2015-04-13 17:11:11 +0200 | [diff] [blame] | 68 | LIST_INIT(&appctx->runq); |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 69 | LIST_INIT(&appctx->buffer_wait.list); |
| 70 | appctx->buffer_wait.target = appctx; |
| 71 | appctx->buffer_wait.wakeup_cb = (int (*)(void *))appctx_res_wakeup; |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 72 | nb_applets++; |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 73 | } |
| 74 | return appctx; |
| 75 | } |
| 76 | |
| 77 | /* Releases an appctx previously allocated by appctx_new(). Note that |
| 78 | * we share the connection pool. |
| 79 | */ |
Emeric Brun | c730606 | 2017-06-26 16:36:53 +0200 | [diff] [blame] | 80 | static inline void __appctx_free(struct appctx *appctx) |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 81 | { |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 82 | if (!LIST_ISEMPTY(&appctx->runq)) { |
Willy Tarreau | 81f38d6 | 2015-04-13 17:11:11 +0200 | [diff] [blame] | 83 | LIST_DEL(&appctx->runq); |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 84 | applets_active_queue--; |
| 85 | } |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 86 | if (!LIST_ISEMPTY(&appctx->buffer_wait.list)) { |
| 87 | LIST_DEL(&appctx->buffer_wait.list); |
| 88 | LIST_INIT(&appctx->buffer_wait.list); |
| 89 | } |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 90 | pool_free2(pool2_connection, appctx); |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 91 | nb_applets--; |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 92 | } |
Emeric Brun | c730606 | 2017-06-26 16:36:53 +0200 | [diff] [blame] | 93 | static inline void appctx_free(struct appctx *appctx) |
| 94 | { |
| 95 | if (appctx->state & APPLET_RUNNING) { |
| 96 | appctx->state |= APPLET_WANT_DIE; |
| 97 | return; |
| 98 | } |
| 99 | __appctx_free(appctx); |
| 100 | } |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 101 | |
Willy Tarreau | 81f38d6 | 2015-04-13 17:11:11 +0200 | [diff] [blame] | 102 | /* wakes up an applet when conditions have changed */ |
Emeric Brun | c730606 | 2017-06-26 16:36:53 +0200 | [diff] [blame] | 103 | static inline void __appctx_wakeup(struct appctx *appctx) |
Willy Tarreau | 81f38d6 | 2015-04-13 17:11:11 +0200 | [diff] [blame] | 104 | { |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 105 | if (LIST_ISEMPTY(&appctx->runq)) { |
Willy Tarreau | 64bca9d | 2015-09-25 17:39:23 +0200 | [diff] [blame] | 106 | LIST_ADDQ(&applet_active_queue, &appctx->runq); |
Christopher Faulet | 1cbe74c | 2016-12-06 09:13:22 +0100 | [diff] [blame] | 107 | applets_active_queue++; |
| 108 | } |
Willy Tarreau | 81f38d6 | 2015-04-13 17:11:11 +0200 | [diff] [blame] | 109 | } |
| 110 | |
Emeric Brun | c730606 | 2017-06-26 16:36:53 +0200 | [diff] [blame] | 111 | static inline void appctx_wakeup(struct appctx *appctx) |
Willy Tarreau | 3c595ac | 2015-04-19 09:59:31 +0200 | [diff] [blame] | 112 | { |
Emeric Brun | c730606 | 2017-06-26 16:36:53 +0200 | [diff] [blame] | 113 | if (appctx->state & APPLET_RUNNING) { |
| 114 | appctx->state |= APPLET_WOKEN_UP; |
| 115 | return; |
Willy Tarreau | 3c595ac | 2015-04-19 09:59:31 +0200 | [diff] [blame] | 116 | } |
Emeric Brun | c730606 | 2017-06-26 16:36:53 +0200 | [diff] [blame] | 117 | __appctx_wakeup(appctx); |
Willy Tarreau | 3c595ac | 2015-04-19 09:59:31 +0200 | [diff] [blame] | 118 | } |
| 119 | |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 120 | /* Callback used to wake up an applet when a buffer is available. The applet |
| 121 | * <appctx> is woken up is if it is not already in the list of "active" |
| 122 | * applets. This functions returns 1 is the stream is woken up, otherwise it |
Emeric Brun | c730606 | 2017-06-26 16:36:53 +0200 | [diff] [blame] | 123 | * returns 0. If task is running we request we check if woken was already |
| 124 | * requested */ |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 125 | static inline int appctx_res_wakeup(struct appctx *appctx) |
| 126 | { |
Emeric Brun | c730606 | 2017-06-26 16:36:53 +0200 | [diff] [blame] | 127 | if (appctx->state & APPLET_RUNNING) { |
| 128 | if (appctx->state & APPLET_WOKEN_UP) { |
| 129 | return 0; |
| 130 | } |
| 131 | appctx->state |= APPLET_WOKEN_UP; |
| 132 | return 1; |
| 133 | } |
| 134 | |
| 135 | if (!LIST_ISEMPTY(&appctx->runq)) { |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 136 | return 0; |
Emeric Brun | c730606 | 2017-06-26 16:36:53 +0200 | [diff] [blame] | 137 | } |
| 138 | __appctx_wakeup(appctx); |
Christopher Faulet | a73e59b | 2016-12-09 17:30:18 +0100 | [diff] [blame] | 139 | return 1; |
| 140 | } |
| 141 | |
| 142 | |
Willy Tarreau | 8a8d83b | 2015-04-13 13:24:54 +0200 | [diff] [blame] | 143 | #endif /* _PROTO_APPLET_H */ |
| 144 | |
| 145 | /* |
| 146 | * Local variables: |
| 147 | * c-indent-level: 8 |
| 148 | * c-basic-offset: 8 |
| 149 | * End: |
| 150 | */ |