You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
107 lines
2.4 KiB
C
107 lines
2.4 KiB
C
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <mosquitto.h>
|
|
#include <mosquitto_broker.h>
|
|
#include <mosquitto_plugin.h>
|
|
|
|
static int tick_callback(int event, void *event_data, void *user_data);
|
|
static int unpwd_check_callback(int event, void *event_data, void *user_data);
|
|
|
|
static mosquitto_plugin_id_t *plg_id;
|
|
|
|
static char *username = NULL;
|
|
static char *password = NULL;
|
|
static char *clientid = NULL;
|
|
static int auth_delay = -1;
|
|
|
|
MOSQUITTO_PLUGIN_DECLARE_VERSION(5);
|
|
|
|
int mosquitto_plugin_init(mosquitto_plugin_id_t *identifier, void **user_data, struct mosquitto_opt *auth_opts, int auth_opt_count)
|
|
{
|
|
(void)user_data;
|
|
(void)auth_opts;
|
|
(void)auth_opt_count;
|
|
|
|
plg_id = identifier;
|
|
|
|
mosquitto_callback_register(plg_id, MOSQ_EVT_TICK, tick_callback, NULL, NULL);
|
|
mosquitto_callback_register(plg_id, MOSQ_EVT_BASIC_AUTH, unpwd_check_callback, NULL, NULL);
|
|
|
|
return MOSQ_ERR_SUCCESS;
|
|
}
|
|
|
|
int mosquitto_plugin_cleanup(void *user_data, struct mosquitto_opt *auth_opts, int auth_opt_count)
|
|
{
|
|
(void)user_data;
|
|
(void)auth_opts;
|
|
(void)auth_opt_count;
|
|
|
|
free(username);
|
|
free(password);
|
|
free(clientid);
|
|
|
|
mosquitto_callback_unregister(plg_id, MOSQ_EVT_BASIC_AUTH, unpwd_check_callback, NULL);
|
|
mosquitto_callback_unregister(plg_id, MOSQ_EVT_TICK, tick_callback, NULL);
|
|
|
|
return MOSQ_ERR_SUCCESS;
|
|
}
|
|
|
|
static int tick_callback(int event, void *event_data, void *user_data)
|
|
{
|
|
struct mosquitto_evt_tick *ed = event_data;
|
|
|
|
(void)user_data;
|
|
|
|
if(event != MOSQ_EVT_TICK){
|
|
abort();
|
|
}
|
|
|
|
if(auth_delay == 0){
|
|
if(clientid && username && password
|
|
&& !strcmp(username, "delayed-username") && !strcmp(password, "good")){
|
|
|
|
mosquitto_complete_basic_auth(clientid, MOSQ_ERR_SUCCESS);
|
|
}else{
|
|
mosquitto_complete_basic_auth(clientid, MOSQ_ERR_AUTH);
|
|
}
|
|
free(username);
|
|
free(password);
|
|
free(clientid);
|
|
username = NULL;
|
|
password = NULL;
|
|
clientid = NULL;
|
|
}else if(auth_delay > 0){
|
|
auth_delay--;
|
|
}
|
|
|
|
/* fast turn around for quick testing */
|
|
ed->next_ms = 100;
|
|
|
|
return MOSQ_ERR_SUCCESS;
|
|
}
|
|
|
|
static int unpwd_check_callback(int event, void *event_data, void *user_data)
|
|
{
|
|
struct mosquitto_evt_basic_auth *ed = event_data;
|
|
|
|
(void)event;
|
|
(void)user_data;
|
|
|
|
free(username);
|
|
free(password);
|
|
free(clientid);
|
|
|
|
if(ed->username){
|
|
username = strdup(ed->username);
|
|
}
|
|
if(ed->password){
|
|
password = strdup(ed->password);
|
|
}
|
|
clientid = strdup(mosquitto_client_id(ed->client));
|
|
/* Delay for arbitrary 10 ticks */
|
|
auth_delay = 10;
|
|
|
|
return MOSQ_ERR_AUTH_DELAYED;
|
|
}
|