blob: d664fd9b246c604cbc6b18bfed7fb5b2bed001e7 [file] [log] [blame]
Willy Tarreau8a8d83b2015-04-13 13:24:54 +02001/*
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 Tarreau81f38d62015-04-13 17:11:11 +020028#include <common/mini-clist.h>
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020029#include <types/applet.h>
30#include <proto/connection.h>
Olivier Houchard673867c2018-05-25 16:58:52 +020031#include <proto/task.h>
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020032
Christopher Faulet1cbe74c2016-12-06 09:13:22 +010033extern unsigned int nb_applets;
Willy Tarreau81f38d62015-04-13 17:11:11 +020034
Olivier Houchard673867c2018-05-25 16:58:52 +020035struct task *task_run_applet(struct task *t, void *context, unsigned short state);
Willy Tarreau3c595ac2015-04-19 09:59:31 +020036
Willy Tarreau21028b52018-11-06 17:32:37 +010037int appctx_buf_available(void *arg);
Christopher Fauleta73e59b2016-12-09 17:30:18 +010038
Christopher Fauleta73e59b2016-12-09 17:30:18 +010039
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020040/* Initializes all required fields for a new appctx. Note that it does the
41 * minimum acceptable initialization for an appctx. This means only the
Aurélien Nephtaliabbf6072018-04-18 13:26:46 +020042 * 3 integer states st0, st1, st2 and the chunk used to gather unfinished
43 * commands are zeroed
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020044 */
Emeric Brun1138fd02017-06-19 12:38:55 +020045static inline void appctx_init(struct appctx *appctx, unsigned long thread_mask)
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020046{
47 appctx->st0 = appctx->st1 = appctx->st2 = 0;
Aurélien Nephtaliabbf6072018-04-18 13:26:46 +020048 appctx->chunk = NULL;
Thierry FOURNIER / OZON.IO6a22dcb2016-11-12 10:51:33 +010049 appctx->io_release = NULL;
Willy Tarreauf65610a2017-10-31 16:06:06 +010050 appctx->thread_mask = thread_mask;
Olivier Houchard673867c2018-05-25 16:58:52 +020051 appctx->state = 0;
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020052}
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
Willy Tarreaubafbe012017-11-24 17:34:44 +010056 * pool_free(connection) or appctx_free(), since it's allocated from the
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020057 * connection pool. <applet> is assigned as the applet, but it can be NULL.
58 */
Emeric Brun1138fd02017-06-19 12:38:55 +020059static inline struct appctx *appctx_new(struct applet *applet, unsigned long thread_mask)
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020060{
61 struct appctx *appctx;
62
Willy Tarreaubafbe012017-11-24 17:34:44 +010063 appctx = pool_alloc(pool_head_connection);
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020064 if (likely(appctx != NULL)) {
65 appctx->obj_type = OBJ_TYPE_APPCTX;
66 appctx->applet = applet;
Emeric Brun1138fd02017-06-19 12:38:55 +020067 appctx_init(appctx, thread_mask);
Olivier Houchard673867c2018-05-25 16:58:52 +020068 appctx->t = task_new(thread_mask);
69 if (unlikely(appctx->t == NULL)) {
70 pool_free(pool_head_connection, appctx);
71 return NULL;
72 }
73 appctx->t->process = task_run_applet;
74 appctx->t->context = appctx;
Christopher Fauleta73e59b2016-12-09 17:30:18 +010075 LIST_INIT(&appctx->buffer_wait.list);
76 appctx->buffer_wait.target = appctx;
Willy Tarreau21028b52018-11-06 17:32:37 +010077 appctx->buffer_wait.wakeup_cb = appctx_buf_available;
Emeric Brun1138fd02017-06-19 12:38:55 +020078 HA_ATOMIC_ADD(&nb_applets, 1);
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020079 }
80 return appctx;
81}
82
83/* Releases an appctx previously allocated by appctx_new(). Note that
84 * we share the connection pool.
85 */
Emeric Brunc7306062017-06-26 16:36:53 +020086static inline void __appctx_free(struct appctx *appctx)
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020087{
Olivier Houchard673867c2018-05-25 16:58:52 +020088 task_delete(appctx->t);
89 task_free(appctx->t);
Christopher Fauleta73e59b2016-12-09 17:30:18 +010090 if (!LIST_ISEMPTY(&appctx->buffer_wait.list)) {
Christopher Faulet2a944ee2017-11-07 10:42:54 +010091 HA_SPIN_LOCK(BUF_WQ_LOCK, &buffer_wq_lock);
Christopher Fauleta73e59b2016-12-09 17:30:18 +010092 LIST_DEL(&appctx->buffer_wait.list);
93 LIST_INIT(&appctx->buffer_wait.list);
Christopher Faulet2a944ee2017-11-07 10:42:54 +010094 HA_SPIN_UNLOCK(BUF_WQ_LOCK, &buffer_wq_lock);
Christopher Fauleta73e59b2016-12-09 17:30:18 +010095 }
Emeric Brun1138fd02017-06-19 12:38:55 +020096
Willy Tarreaubafbe012017-11-24 17:34:44 +010097 pool_free(pool_head_connection, appctx);
Emeric Brun1138fd02017-06-19 12:38:55 +020098 HA_ATOMIC_SUB(&nb_applets, 1);
Willy Tarreau8a8d83b2015-04-13 13:24:54 +020099}
Olivier Houchard673867c2018-05-25 16:58:52 +0200100
Emeric Brunc7306062017-06-26 16:36:53 +0200101static inline void appctx_free(struct appctx *appctx)
102{
Olivier Houchard673867c2018-05-25 16:58:52 +0200103 /* The task is supposed to be run on this thread, so we can just
104 * check if it's running already (or about to run) or not
105 */
106 if (!(appctx->t->state & TASK_RUNNING))
107 __appctx_free(appctx);
108 else {
109 /* if it's running, or about to run, defer the freeing
110 * until the callback is called.
111 */
Emeric Brunc7306062017-06-26 16:36:53 +0200112 appctx->state |= APPLET_WANT_DIE;
Olivier Houchard673867c2018-05-25 16:58:52 +0200113 task_wakeup(appctx->t, TASK_WOKEN_OTHER);
Emeric Brunc7306062017-06-26 16:36:53 +0200114 }
Emeric Brunc7306062017-06-26 16:36:53 +0200115}
Willy Tarreau8a8d83b2015-04-13 13:24:54 +0200116
Willy Tarreau81f38d62015-04-13 17:11:11 +0200117/* wakes up an applet when conditions have changed */
Emeric Brunc7306062017-06-26 16:36:53 +0200118static inline void appctx_wakeup(struct appctx *appctx)
Willy Tarreau3c595ac2015-04-19 09:59:31 +0200119{
Olivier Houchard673867c2018-05-25 16:58:52 +0200120 task_wakeup(appctx->t, TASK_WOKEN_OTHER);
Christopher Fauleta73e59b2016-12-09 17:30:18 +0100121}
122
Willy Tarreau8a8d83b2015-04-13 13:24:54 +0200123#endif /* _PROTO_APPLET_H */
124
125/*
126 * Local variables:
127 * c-indent-level: 8
128 * c-basic-offset: 8
129 * End:
130 */