From a78815c3b6f8aa727cdccc7baace02fee95f429c Mon Sep 17 00:00:00 2001
From: DanyLE <xsang.le@gmail.com>
Date: Mon, 9 Jan 2023 19:51:21 +0100
Subject: [PATCH] Allow enable/disable plugin on each port

---
 antd-config.ini | 14 ++++----------
 http_server.c   | 42 ++++++++++++++++++++++++++++++++++++++----
 lib/handle.h    |  1 +
 3 files changed, 43 insertions(+), 14 deletions(-)

diff --git a/antd-config.ini b/antd-config.ini
index 49333e0..e97f8e7 100644
--- a/antd-config.ini
+++ b/antd-config.ini
@@ -54,6 +54,8 @@ ssl.enable=0
 
 [PORT:80]
 htdocs=/opt/www/htdocs
+; enable specific plugin
+plugins=lua,tunnel
 ; enable or disable SSL
 ssl.enable=0
 ; ^\/os\/+(.*)$ = /proxy/http://localhost:443/test.html?<query>
@@ -97,16 +99,6 @@ application/x-font-woff=woff,woff2
 application/x-font-otf=otf
 audio/mpeg=mp3,mpeg
 
-[FILEHANDLER]
-; specify a plugin for handling
-; a file type
-; lua page script
-ls = lua
-; pure lua script
-lua = lua
-; php and o ther scripting languages can be
-; handled by the cgi plugin
-; php = cgi
 
 ; Example of plugin configurations
 [PLUGIN:php]
@@ -115,6 +107,8 @@ lua = lua
 name = fastcgi
 ; run this plugin at startup
 autoload = true
+; file handle
+file_type = php,pp
 ; pluggin specific configurations here, for example
 socket = /var/php.sock
 bin = /usr/bin/phpfcgi
diff --git a/http_server.c b/http_server.c
index 642235b..63abec9 100644
--- a/http_server.c
+++ b/http_server.c
@@ -94,6 +94,8 @@ void destroy_config()
             {
                 if (cnf->htdocs != NULL)
                     free(cnf->htdocs);
+                if(cnf->plugins)
+                    free(cnf->plugins);
                 if (cnf->sock > 0)
                 {
                     close(cnf->sock);
@@ -242,10 +244,6 @@ static int config_handler(void *conf, const char *section, const char *name,
         pconfig->ssl_cipher = strdup(value);
     }
 #endif
-    else if (strcmp(section, "FILEHANDLER") == 0)
-    {
-        dput(pconfig->handlers, name, strdup(value));
-    }
     else if (strcmp(section, "MIMES") == 0)
     {
         dput(pconfig->mimes, name, strdup(value));
@@ -259,6 +257,7 @@ static int config_handler(void *conf, const char *section, const char *name,
         {
             p = (port_config_t *)malloc(sizeof(port_config_t));
             p->htdocs = NULL;
+            p->plugins = NULL;
             p->sock = -1;
             p->rules = dict_n(1);
             dput(pconfig->ports, buf, p);
@@ -281,6 +280,10 @@ static int config_handler(void *conf, const char *section, const char *name,
                 LOG("Server root is %s", p->htdocs);
             }
         }
+        else if(strcmp(name, "plugins") == 0)
+        {
+            p->plugins = strdup(value);
+        }
         else if (strcmp(name, "ssl.enable") == 0)
         {
             p->usessl = atoi(value);
@@ -326,6 +329,22 @@ static void init_plugins()
             for_each_assoc(it2, config)
             {
                 LOG("Plugin %s: [%s] -> [%s]", it->key, it2->key, (char*) it2->value);
+                if(strncmp(it2->key, "file_type", 9) == 0 && it2->value)
+                {
+                    char* file_type = strdup((char*) it2->value);
+                    char* token;
+                    char *stringp = file_type;
+                    while((token = strsep(&stringp,",")))
+                    {
+                        trim(token, ' ');
+                        if(strlen(token) > 0) 
+                        {
+                            dput(server_config.handlers,token, strdup((char*)it->key));
+                            LOG("Plugin %s: support %s file", it->key, token);
+                        }
+                    }
+                    free(file_type);
+                }
             }
             value = (char*)dvalue(config,"autoload");
             if( value && (strncmp(value,"1", 1) == 0  || strncmp(value, "true", 3) == 0 ) )
@@ -1669,9 +1688,24 @@ void *execute_plugin(void *data, const char *pname)
     plugin_header_t *meta = NULL;
     struct plugin_entry *plugin;
     char *error;
+    char pattern[256];
+
     antd_request_t *rq = (antd_request_t *)data;
     antd_task_t *task = antd_create_task(NULL, (void *)rq, NULL, rq->client->last_io);
     antd_task_bind_event(task, rq->client->sock, 0, TASK_EVT_ON_WRITABLE | TASK_EVT_ON_READABLE);
+
+    snprintf(pattern, sizeof(pattern), "\\b%s\\b", pname);
+    char *port_s = (char *)dvalue(rq->request, "SERVER_PORT");
+    port_config_t *pcnf = (port_config_t *)dvalue(server_config.ports, port_s);
+
+    // check if plugin is enabled on this port
+    if(!pcnf->plugins || !regex_match(pattern, pcnf->plugins , 0,NULL))
+    {
+        LOG("No plugin matched in [%s] using pattern [%s]", pcnf->plugins, pattern);
+        antd_error(rq->client, 403, "Access forbidden");
+        return task;
+    }
+
     // LOG("Plugin name '%s'", pname);
     rq->client->state = ANTD_CLIENT_PLUGIN_EXEC;
     // load the plugin
diff --git a/lib/handle.h b/lib/handle.h
index 1eaaf24..7f34d39 100644
--- a/lib/handle.h
+++ b/lib/handle.h
@@ -42,6 +42,7 @@ typedef struct
     unsigned int port;
     int usessl;
     char *htdocs;
+    char* plugins;
     int sock;
     dictionary_t rules;
 } port_config_t;