Skip to content
Snippets Groups Projects
Commit cd27a671 authored by Rob Swindell's avatar Rob Swindell :speech_balloon:
Browse files

Add JavaScript MQTT class for communicating with MQTT broker(s) via JS

Right now, only supports publish, subscribe to come next.

Only included with the mosquitto lib support is enabled in the build.
parent 94b5d624
No related branches found
No related tags found
1 merge request!463MRC mods by Codefenix (2024-10-20)
Pipeline #4333 failed
/* Synchronet JavaScript "MQTT" Class */
/****************************************************************************
* @format.tab-size 4 (Plain Text/Source Code File Header) *
* @format.use-tabs true (see http://www.synchro.net/ptsc_hdr.html) *
* *
* Copyright Rob Swindell - http://www.synchro.net/copyright.html *
* *
* This program is free software; you can redistribute it and/or *
* modify it under the terms of the GNU General Public License *
* as published by the Free Software Foundation; either version 2 *
* of the License, or (at your option) any later version. *
* See the GNU General Public License for more details: gpl.txt or *
* http://www.fsf.org/copyleft/gpl.html *
* *
* For Synchronet coding style and modification guidelines, see *
* http://www.synchro.net/source.html *
* *
* Note: If this box doesn't appear square, then you need to fix your tabs. *
****************************************************************************/
#if defined USE_MOSQUITTO
#include "sbbs.h"
#include "js_request.h"
typedef struct
{
mqtt_handle_t handle;
struct mqtt_cfg cfg;
} private_t;
static void js_finalize_mqtt(JSContext* cx, JSObject* obj)
{
private_t* p;
if((p = (private_t*)JS_GetPrivate(cx,obj)) == NULL)
return;
if(p->handle != NULL)
mosquitto_destroy(p->handle);
free(p);
JS_SetPrivate(cx, obj, NULL);
}
extern JSClass js_mqtt_class;
static JSBool js_disconnect(JSContext* cx, uintN argc, jsval *arglist)
{
JSObject* obj = JS_THIS_OBJECT(cx, arglist);
private_t* p;
jsrefcount rc;
JS_SET_RVAL(cx, arglist, JSVAL_FALSE);
if((p = (private_t*)js_GetClassPrivate(cx, obj, &js_mqtt_class)) == NULL) {
return JS_FALSE;
}
if(p->handle == NULL)
return JS_TRUE;
rc = JS_SUSPENDREQUEST(cx);
int result = mosquitto_disconnect(p->handle);
JS_SET_RVAL(cx, arglist, BOOLEAN_TO_JSVAL(result == MOSQ_ERR_SUCCESS));
JS_RESUMEREQUEST(cx, rc);
return JS_TRUE;
}
static int pw_callback(char* buf, int size, int rwflag, void* userdata)
{
private_t* p = (private_t*)userdata;
strncpy(buf, p->cfg.tls.keypass, size);
return strlen(p->cfg.tls.keypass);
}
static JSBool js_connect(JSContext* cx, uintN argc, jsval *arglist)
{
JSObject* obj = JS_THIS_OBJECT(cx, arglist);
jsval* argv = JS_ARGV(cx, arglist);
private_t* p;
jsrefcount rc;
char broker_addr[sizeof p->cfg.broker_addr];
uint16_t broker_port;
char username[sizeof p->cfg.username];
char password[sizeof p->cfg.password];
JS_SET_RVAL(cx, arglist, JSVAL_VOID);
if((p = (private_t*)js_GetClassPrivate(cx, obj, &js_mqtt_class)) == NULL)
return JS_FALSE;
rc = JS_SUSPENDREQUEST(cx);
SAFECOPY(broker_addr, p->cfg.broker_addr);
broker_port = p->cfg.broker_port;
SAFECOPY(username, p->cfg.username);
SAFECOPY(password, p->cfg.password);
uintN argn = 0;
if(argn < argc && JSVAL_IS_STRING(argv[argn])) {
JSVALUE_TO_STRBUF(cx, argv[argn], broker_addr, sizeof broker_addr, NULL);
++argn;
}
if(argn < argc && JSVAL_IS_NUMBER(argv[argn])) {
broker_port = JSVAL_TO_INT(argv[argn]);
++argn;
}
if(argn < argc && JSVAL_IS_STRING(argv[argn])) {
JSVALUE_TO_STRBUF(cx, argv[argn], username, sizeof username, NULL);
++argn;
}
if(argn < argc && JSVAL_IS_STRING(argv[argn])) {
JSVALUE_TO_STRBUF(cx, argv[argn], password, sizeof password, NULL);
++argn;
}
mosquitto_int_option(p->handle, MOSQ_OPT_PROTOCOL_VERSION, p->cfg.protocol_version);
mosquitto_username_pw_set(p->handle, *username ? username : NULL, *password ? password : NULL);
if(p->cfg.tls.mode == MQTT_TLS_CERT) {
char* certfile = NULL;
char* keyfile = NULL;
if(p->cfg.tls.certfile[0] && p->cfg.tls.keyfile[0]) {
certfile = p->cfg.tls.certfile;
keyfile = p->cfg.tls.keyfile;
}
int result = mosquitto_tls_set(p->handle,
p->cfg.tls.cafile,
NULL, // capath
certfile,
keyfile,
pw_callback);
if(result != MOSQ_ERR_SUCCESS)
return result;
}
else if(p->cfg.tls.mode == MQTT_TLS_PSK) {
int result = mosquitto_tls_psk_set(p->handle,
p->cfg.tls.psk,
p->cfg.tls.identity,
NULL // ciphers (default)
);
if(result != MOSQ_ERR_SUCCESS)
return result;
}
int result = mosquitto_connect_bind(p->handle,
broker_addr,
broker_port,
p->cfg.keepalive,
/* bind_address */NULL);
JS_SET_RVAL(cx, arglist, BOOLEAN_TO_JSVAL(result == MOSQ_ERR_SUCCESS));
JS_RESUMEREQUEST(cx, rc);
return JS_TRUE;
}
static JSBool js_publish(JSContext* cx, uintN argc, jsval *arglist)
{
JSObject* obj = JS_THIS_OBJECT(cx, arglist);
jsval* argv = JS_ARGV(cx, arglist);
char* topic = NULL;
char* data = NULL;
size_t len = 0;
private_t* p;
jsrefcount rc;
bool retain = false;
JS_SET_RVAL(cx, arglist, JSVAL_VOID);
if((p = (private_t*)js_GetClassPrivate(cx, obj, &js_mqtt_class)) == NULL)
return JS_FALSE;
int qos = p->cfg.publish_qos;
if(!js_argc(cx, argc, 2))
return JS_FALSE;
JS_SET_RVAL(cx, arglist, JSVAL_FALSE);
if(p->handle == NULL)
return JS_TRUE;
uintN argn = 0;
if(argn < argc && JSVAL_IS_BOOLEAN(argv[argn])) {
retain = JSVAL_TO_BOOLEAN(argv[argn]);
++argn;
}
if(argn < argc && JSVAL_IS_NUMBER(argv[argn])) {
qos = JSVAL_TO_INT(argv[argn]);
++argn;
}
JSVALUE_TO_MSTRING(cx, argv[argn], topic, NULL);
HANDLE_PENDING(cx, topic);
++argn;
JSVALUE_TO_MSTRING(cx, argv[argn], data, &len);
HANDLE_PENDING(cx, data);
++argn;
rc = JS_SUSPENDREQUEST(cx);
int result = mosquitto_publish_v5(p->handle,
/* mid: */NULL,
/* topic: */topic,
/* payloadlen */len,
/* payload */data,
qos,
retain,
/* properties */NULL);
JS_SET_RVAL(cx, arglist, BOOLEAN_TO_JSVAL(result == MOSQ_ERR_SUCCESS));
free(data);
free(topic);
JS_RESUMEREQUEST(cx, rc);
return JS_TRUE;
}
/* Properites */
enum {
MQTT_PROP_BROKER_ADDR
,MQTT_PROP_BROKER_PORT
,MQTT_PROP_USERNAME
,MQTT_PROP_PASSWORD
,MQTT_PROP_KEEPALIVE
,MQTT_PROP_PROT_VER
,MQTT_PROP_PUB_QOS
,MQTT_PROP_TLS_MODE
,MQTT_PROP_TLS_CAFILE
,MQTT_PROP_TLS_CERTFILE
,MQTT_PROP_TLS_KEYFILE
,MQTT_PROP_TLS_KEYPASS
,MQTT_PROP_TLS_PSK
,MQTT_PROP_TLS_IDENTITY
};
#ifdef BUILD_JSDOCS
static char* com_prop_desc[] = {
"IP address or hostname of MQTT broker to connect to, by default"
,"TCP port number of MQTT broker to connect to, by default"
,"Username to use when authenticating with MQTT broker, by default"
,"Password to use when authenticating with MQTT broker, by default"
,"Seconds of time to keep inactive connection alive"
,"Protocol version number (3 = 3.1.0, 4 = 3.1.1, 5 = 5.0)"
,"Quality Of Service (QOS) value to use when publishing, by default"
,"TLS (encryption) mode"
,"TLS Certificate Authority (CA) certificate (file path)"
,"TLS Client certificate (file path)"
,"Private key file"
,"Private key file password"
,"TLS Pre-Shared-Key"
,"TLS PSK Identity"
,NULL
};
#endif
static JSBool js_mqtt_set(JSContext* cx, JSObject* obj, jsid id, JSBool strict, jsval *vp)
{
jsval idval;
jsint tiny;
private_t* p;
int32 i;
if((p = (private_t*)JS_GetPrivate(cx,obj)) == NULL) {
// Prototype access
return JS_TRUE;
}
JS_IdToValue(cx, id, &idval);
tiny = JSVAL_TO_INT(idval);
switch(tiny) {
case MQTT_PROP_BROKER_ADDR:
JSVALUE_TO_STRBUF(cx, *vp, p->cfg.broker_addr, sizeof p->cfg.broker_addr, NULL);
break;
case MQTT_PROP_BROKER_PORT:
if(!JS_ValueToInt32(cx, *vp, &i))
return JS_FALSE;
p->cfg.broker_port = i;
break;
case MQTT_PROP_USERNAME:
JSVALUE_TO_STRBUF(cx, *vp, p->cfg.username, sizeof p->cfg.username, NULL);
break;
case MQTT_PROP_PASSWORD:
JSVALUE_TO_STRBUF(cx, *vp, p->cfg.password, sizeof p->cfg.password, NULL);
break;
case MQTT_PROP_KEEPALIVE:
if(!JS_ValueToInt32(cx, *vp, &i))
return JS_FALSE;
p->cfg.keepalive = i;
break;
case MQTT_PROP_PROT_VER:
if(!JS_ValueToInt32(cx, *vp, &i))
return JS_FALSE;
p->cfg.protocol_version = i;
break;
case MQTT_PROP_PUB_QOS:
if(!JS_ValueToInt32(cx, *vp, &i))
return JS_FALSE;
p->cfg.publish_qos = i;
break;
case MQTT_PROP_TLS_MODE:
if(!JS_ValueToInt32(cx, *vp, &i))
return JS_FALSE;
p->cfg.tls.mode = i;
break;
case MQTT_PROP_TLS_CAFILE:
JSVALUE_TO_STRBUF(cx, *vp, p->cfg.tls.cafile, sizeof p->cfg.tls.cafile, NULL);
break;
case MQTT_PROP_TLS_CERTFILE:
JSVALUE_TO_STRBUF(cx, *vp, p->cfg.tls.certfile, sizeof p->cfg.tls.certfile, NULL);
break;
case MQTT_PROP_TLS_KEYFILE:
JSVALUE_TO_STRBUF(cx, *vp, p->cfg.tls.keyfile, sizeof p->cfg.tls.keyfile, NULL);
break;
case MQTT_PROP_TLS_KEYPASS:
JSVALUE_TO_STRBUF(cx, *vp, p->cfg.tls.keypass, sizeof p->cfg.tls.keypass, NULL);
break;
case MQTT_PROP_TLS_PSK:
JSVALUE_TO_STRBUF(cx, *vp, p->cfg.tls.psk, sizeof p->cfg.tls.psk, NULL);
break;
case MQTT_PROP_TLS_IDENTITY:
JSVALUE_TO_STRBUF(cx, *vp, p->cfg.tls.identity, sizeof p->cfg.tls.identity, NULL);
break;
}
return JS_TRUE;
}
static JSBool js_mqtt_get(JSContext* cx, JSObject* obj, jsid id, jsval *vp)
{
jsval idval;
jsint tiny;
private_t* p;
JSString* js_str;
jsrefcount rc;
if((p = (private_t*)JS_GetPrivate(cx,obj)) == NULL) {
// Protoype access
return JS_TRUE;
}
JS_IdToValue(cx, id, &idval);
tiny = JSVAL_TO_INT(idval);
rc = JS_SUSPENDREQUEST(cx);
switch(tiny) {
case MQTT_PROP_BROKER_ADDR:
JS_RESUMEREQUEST(cx, rc);
if((js_str = JS_NewStringCopyZ(cx, p->cfg.broker_addr)) == NULL)
return JS_FALSE;
*vp = STRING_TO_JSVAL(js_str);
rc = JS_SUSPENDREQUEST(cx);
break;
case MQTT_PROP_BROKER_PORT:
*vp = INT_TO_JSVAL(p->cfg.broker_port);
break;
case MQTT_PROP_USERNAME:
JS_RESUMEREQUEST(cx, rc);
if((js_str = JS_NewStringCopyZ(cx, p->cfg.username)) == NULL)
return JS_FALSE;
*vp = STRING_TO_JSVAL(js_str);
rc = JS_SUSPENDREQUEST(cx);
break;
case MQTT_PROP_PASSWORD:
JS_RESUMEREQUEST(cx, rc);
if((js_str = JS_NewStringCopyZ(cx, p->cfg.password)) == NULL)
return JS_FALSE;
*vp = STRING_TO_JSVAL(js_str);
rc = JS_SUSPENDREQUEST(cx);
break;
case MQTT_PROP_KEEPALIVE:
*vp = INT_TO_JSVAL(p->cfg.keepalive);
break;
case MQTT_PROP_PROT_VER:
*vp = INT_TO_JSVAL(p->cfg.protocol_version);
break;
case MQTT_PROP_PUB_QOS:
*vp = INT_TO_JSVAL(p->cfg.publish_qos);
break;
case MQTT_PROP_TLS_MODE:
*vp = INT_TO_JSVAL(p->cfg.tls.mode);
break;
case MQTT_PROP_TLS_CAFILE:
JS_RESUMEREQUEST(cx, rc);
if((js_str = JS_NewStringCopyZ(cx, p->cfg.tls.cafile)) == NULL)
return JS_FALSE;
*vp = STRING_TO_JSVAL(js_str);
rc = JS_SUSPENDREQUEST(cx);
break;
case MQTT_PROP_TLS_CERTFILE:
JS_RESUMEREQUEST(cx, rc);
if((js_str = JS_NewStringCopyZ(cx, p->cfg.tls.certfile)) == NULL)
return JS_FALSE;
*vp = STRING_TO_JSVAL(js_str);
rc = JS_SUSPENDREQUEST(cx);
break;
case MQTT_PROP_TLS_KEYFILE:
JS_RESUMEREQUEST(cx, rc);
if((js_str = JS_NewStringCopyZ(cx, p->cfg.tls.keyfile)) == NULL)
return JS_FALSE;
*vp = STRING_TO_JSVAL(js_str);
rc = JS_SUSPENDREQUEST(cx);
break;
case MQTT_PROP_TLS_KEYPASS:
JS_RESUMEREQUEST(cx, rc);
if((js_str = JS_NewStringCopyZ(cx, p->cfg.tls.keypass)) == NULL)
return JS_FALSE;
*vp = STRING_TO_JSVAL(js_str);
rc = JS_SUSPENDREQUEST(cx);
break;
case MQTT_PROP_TLS_PSK:
JS_RESUMEREQUEST(cx, rc);
if((js_str = JS_NewStringCopyZ(cx, p->cfg.tls.psk)) == NULL)
return JS_FALSE;
*vp = STRING_TO_JSVAL(js_str);
rc = JS_SUSPENDREQUEST(cx);
break;
case MQTT_PROP_TLS_IDENTITY:
JS_RESUMEREQUEST(cx, rc);
if((js_str = JS_NewStringCopyZ(cx, p->cfg.tls.identity)) == NULL)
return JS_FALSE;
*vp = STRING_TO_JSVAL(js_str);
rc = JS_SUSPENDREQUEST(cx);
break;
}
JS_RESUMEREQUEST(cx, rc);
return JS_TRUE;
}
#define MQTT_PROP_FLAGS JSPROP_ENUMERATE
static jsSyncPropertySpec js_mqtt_properties[] = {
/* name ,tinyid ,flags, ver */
{ "broker_addr" ,MQTT_PROP_BROKER_ADDR ,MQTT_PROP_FLAGS, 320 },
{ "broker_port" ,MQTT_PROP_BROKER_PORT ,MQTT_PROP_FLAGS, 320 },
{ "username" ,MQTT_PROP_USERNAME ,MQTT_PROP_FLAGS, 320 },
{ "password" ,MQTT_PROP_PASSWORD ,MQTT_PROP_FLAGS, 320 },
{ "keepalive" ,MQTT_PROP_KEEPALIVE ,MQTT_PROP_FLAGS, 320 },
{ "protocol_version" ,MQTT_PROP_PROT_VER ,MQTT_PROP_FLAGS, 320 },
{ "publish_qos" ,MQTT_PROP_PUB_QOS ,MQTT_PROP_FLAGS, 320 },
{ "tls_mode" ,MQTT_PROP_TLS_MODE ,MQTT_PROP_FLAGS, 320 },
{ "tls_ca_cert" ,MQTT_PROP_TLS_CAFILE ,MQTT_PROP_FLAGS, 320 },
{ "tls_client_cert" ,MQTT_PROP_TLS_CERTFILE ,MQTT_PROP_FLAGS, 320 },
{ "tls_private_key" ,MQTT_PROP_TLS_KEYFILE ,MQTT_PROP_FLAGS, 320 },
{ "tls_key_password" ,MQTT_PROP_TLS_KEYPASS ,MQTT_PROP_FLAGS, 320 },
{ "tls_psk" ,MQTT_PROP_TLS_PSK ,MQTT_PROP_FLAGS, 320 },
{ "tls_psk_identity" ,MQTT_PROP_TLS_IDENTITY ,MQTT_PROP_FLAGS, 320 },
{0}
};
static jsSyncMethodSpec js_mqtt_functions[] = {
{"connect", js_connect, 0, JSTYPE_BOOLEAN, JSDOCSTR("[string broker_address] [,number broker_port] [,string username] [,string password]")
,JSDOCSTR("Connect to an MQTT broker")
,320
},
{"disconnect", js_disconnect, 0, JSTYPE_VOID, JSDOCSTR("")
,JSDOCSTR("Close an open connection to the MQTT broker")
,320
},
{"publish", js_publish, 4, JSTYPE_BOOLEAN, JSDOCSTR("[bool retain=false,] [number qos,] topic, data")
,JSDOCSTR("Publish a string to specified topic")
,320
},
{0}
};
static JSBool js_mqtt_resolve(JSContext* cx, JSObject* obj, jsid id)
{
char* name=NULL;
JSBool ret;
if(id != JSID_VOID && id != JSID_EMPTY) {
jsval idval;
JS_IdToValue(cx, id, &idval);
if(JSVAL_IS_STRING(idval)) {
JSSTRING_TO_MSTRING(cx, JSVAL_TO_STRING(idval), name, NULL);
HANDLE_PENDING(cx, name);
}
}
ret = js_SyncResolve(cx, obj, name, js_mqtt_properties, js_mqtt_functions, NULL, 0);
free(name);
return ret;
}
static JSBool js_mqtt_enumerate(JSContext* cx, JSObject* obj)
{
return js_mqtt_resolve(cx, obj, JSID_VOID);
}
JSClass js_mqtt_class = {
"MQTT" /* name */
,JSCLASS_HAS_PRIVATE /* flags */
,JS_PropertyStub /* addProperty */
,JS_PropertyStub /* delProperty */
,js_mqtt_get /* getProperty */
,js_mqtt_set /* setProperty */
,js_mqtt_enumerate /* enumerate */
,js_mqtt_resolve /* resolve */
,JS_ConvertStub /* convert */
,js_finalize_mqtt /* finalize */
};
static JSBool js_mqtt_constructor(JSContext* cx, uintN argc, jsval *arglist)
{
JSObject* obj;
jsval* argv = JS_ARGV(cx, arglist);
private_t* p;
char* client_id = NULL;
scfg_t* scfg = JS_GetRuntimePrivate(JS_GetRuntime(cx));
if(scfg == NULL) {
JS_ReportError(cx, "JS_GetRuntimePrivate returned NULL");
return JS_FALSE;
}
obj = JS_NewObject(cx, &js_mqtt_class, NULL, NULL);
JS_SET_RVAL(cx, arglist, OBJECT_TO_JSVAL(obj));
if(argc > 0) {
JSSTRING_TO_MSTRING(cx, JSVAL_TO_STRING(argv[0]), client_id, NULL);
HANDLE_PENDING(cx, client_id);
}
if((p = (private_t*)malloc(sizeof(private_t))) == NULL) {
JS_ReportError(cx,"malloc failed");
free(client_id);
return JS_FALSE;
}
memset(p, 0, sizeof *p);
p->cfg = scfg->mqtt;
p->handle = mosquitto_new(client_id, /* clean_session: */true, /* userdata: */p);
free(client_id);
if(!JS_SetPrivate(cx, obj, p)) {
JS_ReportError(cx,"JS_SetPrivate failed");
free(p);
return JS_FALSE;
}
#ifdef BUILD_JSDOCS
js_DescribeSyncObject(cx,obj,"Class used for MQTT communications",320);
js_DescribeSyncConstructor(cx,obj,"To create a new MQTT object: "
"var mqtt = new MQTT([client_id])</tt><br>"
);
js_CreateArrayOfStrings(cx, obj, "_property_desc_list", com_prop_desc, JSPROP_READONLY);
#endif
return JS_TRUE;
}
JSObject* js_CreateMQTTClass(JSContext* cx, JSObject* parent)
{
return JS_InitClass(cx, parent, NULL
,&js_mqtt_class
,js_mqtt_constructor
,0 /* number of constructor args */
,NULL /* props, specified in constructor */
,NULL /* funcs, specified in constructor */
,NULL,NULL);
}
#endif // USE_MOSQUITTO
......@@ -1516,6 +1516,10 @@ extern "C" BOOL js_CreateCommonObjects(JSContext* js_cx
if(js_CreateCryptCertClass(js_cx, *glob)==NULL)
break;
#if defined USE_MOSQUITTO
if(js_CreateMQTTClass(js_cx, *glob) == NULL)
break;
#endif
/* Area Objects */
if(!js_CreateUserObjects(js_cx, *glob, cfg, /* user: */NULL, client, startup == NULL ? NULL :startup->web_file_vpath_prefix, /* subscan: */NULL))
break;
......
......@@ -51,6 +51,7 @@ OBJS = $(LOAD_CFG_OBJS) \
$(MTOBJODIR)/js_internal$(OFILE)\
$(MTOBJODIR)/js_msg_area$(OFILE)\
$(MTOBJODIR)/js_msgbase$(OFILE)\
$(MTOBJODIR)/js_mqtt$(OFILE)\
$(MTOBJODIR)/js_filebase$(OFILE)\
$(MTOBJODIR)/js_queue$(OFILE)\
$(MTOBJODIR)/js_request$(OFILE)\
......
......@@ -1507,6 +1507,7 @@ extern "C" {
/* js_cryptcert.c */
DLLEXPORT JSObject* js_CreateCryptCertClass(JSContext* cx, JSObject* parent);
DLLEXPORT JSObject* js_CreateMQTTClass(JSContext*, JSObject* parent);
#endif
#ifdef SBBS /* These aren't exported */
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment