1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
|
/*
Copyright (c) 2020 Roger Light <roger@atchoo.org>
All rights reserved. This program and the accompanying materials
are made available under the terms of the Eclipse Public License 2.0
and Eclipse Distribution License v1.0 which accompany this distribution.
The Eclipse Public License is available at
https://www.eclipse.org/legal/epl-2.0/
and the Eclipse Distribution License is available at
http://www.eclipse.org/org/documents/edl-v10.php.
SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause
Contributors:
Roger Light - initial implementation and documentation.
*/
#include "config.h"
#include <cjson/cJSON.h>
#include <stdio.h>
#include <uthash.h>
#include "mosquitto.h"
#include "mosquitto_broker.h"
#include "json_help.h"
#include "dynamic_security.h"
/* ################################################################
* #
* # Plugin global variables
* #
* ################################################################ */
/* ################################################################
* #
* # Function declarations
* #
* ################################################################ */
/* ################################################################
* #
* # Local variables
* #
* ################################################################ */
/* ################################################################
* #
* # Utility functions
* #
* ################################################################ */
static int dynsec_clientlist__cmp(void *a, void *b)
{
struct dynsec__clientlist *clientlist_a = a;
struct dynsec__clientlist *clientlist_b = b;
return strcmp(clientlist_a->client->username, clientlist_b->client->username);
}
void dynsec_clientlist__kick_all(struct dynsec__clientlist *base_clientlist)
{
struct dynsec__clientlist *clientlist, *clientlist_tmp;
HASH_ITER(hh, base_clientlist, clientlist, clientlist_tmp){
mosquitto_kick_client_by_username(clientlist->client->username, false);
}
}
cJSON *dynsec_clientlist__all_to_json(struct dynsec__clientlist *base_clientlist)
{
struct dynsec__clientlist *clientlist, *clientlist_tmp;
cJSON *j_clients, *j_client;
j_clients = cJSON_CreateArray();
if(j_clients == NULL) return NULL;
HASH_ITER(hh, base_clientlist, clientlist, clientlist_tmp){
j_client = cJSON_CreateObject();
if(j_client == NULL){
cJSON_Delete(j_clients);
return NULL;
}
cJSON_AddItemToArray(j_clients, j_client);
if(cJSON_AddStringToObject(j_client, "username", clientlist->client->username) == NULL
|| (clientlist->priority != -1 && cJSON_AddIntToObject(j_client, "priority", clientlist->priority) == NULL)
){
cJSON_Delete(j_clients);
return NULL;
}
}
return j_clients;
}
int dynsec_clientlist__add(struct dynsec__clientlist **base_clientlist, struct dynsec__client *client, int priority)
{
struct dynsec__clientlist *clientlist;
HASH_FIND(hh, *base_clientlist, client->username, strlen(client->username), clientlist);
if(clientlist != NULL){
/* Client is already in the group */
return MOSQ_ERR_SUCCESS;
}
clientlist = mosquitto_malloc(sizeof(struct dynsec__clientlist));
if(clientlist == NULL){
return MOSQ_ERR_NOMEM;
}
clientlist->client = client;
clientlist->priority = priority;
HASH_ADD_KEYPTR_INORDER(hh, *base_clientlist, client->username, strlen(client->username), clientlist, dynsec_clientlist__cmp);
return MOSQ_ERR_SUCCESS;
}
void dynsec_clientlist__cleanup(struct dynsec__clientlist **base_clientlist)
{
struct dynsec__clientlist *clientlist, *clientlist_tmp;
HASH_ITER(hh, *base_clientlist, clientlist, clientlist_tmp){
HASH_DELETE(hh, *base_clientlist, clientlist);
mosquitto_free(clientlist);
}
}
void dynsec_clientlist__remove(struct dynsec__clientlist **base_clientlist, struct dynsec__client *client)
{
struct dynsec__clientlist *clientlist;
HASH_FIND(hh, *base_clientlist, client->username, strlen(client->username), clientlist);
if(clientlist){
HASH_DELETE(hh, *base_clientlist, clientlist);
mosquitto_free(clientlist);
}
}
|
