int main(int argc, char *argv[])
{
int rc;
int tmp;
struct mosquitto *mosq;
int port = atoi(argv[1]);
mosquitto_lib_init();
mosq = mosquitto_new("prop-test", true, NULL);
mosquitto_connect_callback_set(mosq, on_connect);
mosquitto_publish_callback_set(mosq, on_publish);
tmp = MQTT_PROTOCOL_V5;
mosquitto_opts_set(mosq, MOSQ_OPT_PROTOCOL_VERSION, &tmp);
rc = mosquitto_connect(mosq, "localhost", port, 60);
while(run == -1){
rc = mosquitto_loop(mosq, -1, 1);
}
mosquitto_lib_cleanup();
return run;
}
/* Register the callbacks that the mosquitto connection will use. */
static bool set_callbacks(struct mosquitto *m) {
mosquitto_connect_callback_set(m, on_connect);
mosquitto_publish_callback_set(m, on_publish);
mosquitto_subscribe_callback_set(m, on_subscribe);
mosquitto_message_callback_set(m, on_message);
return true;
}
int main(int argc, char *argv[])
{
int i;
char *host = "5.44.237.19";
int port = 1883;
int keepalive = 60;
bool clean_session = true;
struct mosquitto *mosq = NULL;
sprintf(mqMsg,"%s",argv[1]);
printf("And the word is >> %s <<\n", mqMsg);
mosquitto_lib_init();
mosq = mosquitto_new(NULL, clean_session, NULL);
if(!mosq){
fprintf(stderr, "Error: Out of memory.\n");
return 1;
}
mosquitto_log_callback_set(mosq, my_log_callback);
mosquitto_connect_callback_set(mosq, my_connect_callback);
mosquitto_message_callback_set(mosq, my_message_callback);
mosquitto_publish_callback_set(mosq, my_publish_callback);
if(mosquitto_connect(mosq, host, port, keepalive)){
fprintf(stderr, "Unable to connect.\n");
return 1;
}
mosquitto_loop_forever(mosq, -1, 1);
mosquitto_destroy(mosq);
mosquitto_lib_cleanup();
return 0;
}
int mqPub(void)
{
int i;
char *host = "5.44.237.19";
int port = 1883;
int keepalive = 60;
bool clean_session = true;
struct mosquitto *mosq = NULL;
mosquitto_lib_init();
mosq = mosquitto_new(NULL, clean_session, NULL);
if(!mosq){
fprintf(stderr, "Error: Out of memory.\n");
return 1;
}
mosquitto_log_callback_set(mosq, my_log_callback);
mosquitto_connect_callback_set(mosq, my_connect_callback);
mosquitto_message_callback_set(mosq, my_message_callback);
mosquitto_publish_callback_set(mosq, my_publish_callback);
if(mosquitto_connect(mosq, host, port, keepalive)){
fprintf(stderr, "Unable to connect.\n");
return 1;
}
mosquitto_loop_forever(mosq, -1, 1);
mosquitto_destroy(mosq);
mosquitto_lib_cleanup();
return 0;
}
/* {{{ Mosquitto\Client::onPublish() */
PHP_METHOD(Mosquitto_Client, onPublish)
{
mosquitto_client_object *object;
zend_fcall_info publish_callback = empty_fcall_info;
zend_fcall_info_cache publish_callback_cache = empty_fcall_info_cache;
PHP_MOSQUITTO_ERROR_HANDLING();
if (zend_parse_parameters(ZEND_NUM_ARGS() TSRMLS_CC, "f!",
&publish_callback, &publish_callback_cache) == FAILURE) {
PHP_MOSQUITTO_RESTORE_ERRORS();
return;
}
PHP_MOSQUITTO_RESTORE_ERRORS();
object = (mosquitto_client_object *) mosquitto_client_object_get(getThis() TSRMLS_CC);
if (!ZEND_FCI_INITIALIZED(publish_callback)) {
zend_throw_exception(mosquitto_ce_exception, "Need a valid callback", 0 TSRMLS_CC);
}
object->publish_callback = publish_callback;
object->publish_callback_cache = publish_callback_cache;
Z_ADDREF_P(publish_callback.function_name);
if (publish_callback.object_ptr != NULL) {
Z_ADDREF_P(publish_callback.object_ptr);
}
mosquitto_publish_callback_set(object->client, php_mosquitto_publish_callback);
}
int main(int argc, char *argv[])
{
struct mosquitto *mosq;
int i;
double dstart, dstop, diff;
FILE *fptr;
uint8_t *buf;
buf = malloc(MESSAGE_SIZE*MESSAGE_COUNT);
if(!buf){
printf("Error: Out of memory.\n");
return 1;
}
start.tv_sec = 0;
start.tv_usec = 0;
stop.tv_sec = 0;
stop.tv_usec = 0;
if(create_data()){
printf("Error: Unable to create random input data.\n");
return 1;
}
fptr = fopen("msgsps_pub.dat", "rb");
if(!fptr){
printf("Error: Unable to open random input data.\n");
return 1;
}
fread(buf, sizeof(uint8_t), MESSAGE_SIZE*MESSAGE_COUNT, fptr);
fclose(fptr);
mosquitto_lib_init();
mosq = mosquitto_new("perftest", true, NULL);
mosquitto_connect_callback_set(mosq, my_connect_callback);
mosquitto_disconnect_callback_set(mosq, my_disconnect_callback);
mosquitto_publish_callback_set(mosq, my_publish_callback);
mosquitto_connect(mosq, "127.0.0.1", 1884, 600);
i=0;
while(!mosquitto_loop(mosq, 1, 10) && run){
if(i<MESSAGE_COUNT){
mosquitto_publish(mosq, NULL, "perf/test", MESSAGE_SIZE, &buf[i*MESSAGE_SIZE], 0, false);
i++;
}
}
dstart = (double)start.tv_sec*1.0e6 + (double)start.tv_usec;
dstop = (double)stop.tv_sec*1.0e6 + (double)stop.tv_usec;
diff = (dstop-dstart)/1.0e6;
printf("Start: %g\nStop: %g\nDiff: %g\nMessages/s: %g\n", dstart, dstop, diff, (double)MESSAGE_COUNT/diff);
mosquitto_destroy(mosq);
mosquitto_lib_cleanup();
return 0;
}
mosquittopp::mosquittopp(const char *id)
{
mosq = mosquitto_new(id, this);
mosquitto_connect_callback_set(mosq, on_connect_wrapper);
mosquitto_disconnect_callback_set(mosq, on_disconnect_wrapper);
mosquitto_publish_callback_set(mosq, on_publish_wrapper);
mosquitto_message_callback_set(mosq, on_message_wrapper);
mosquitto_subscribe_callback_set(mosq, on_subscribe_wrapper);
mosquitto_unsubscribe_callback_set(mosq, on_unsubscribe_wrapper);
}
int MqttHal_Init()
{
int rc = 0;
//struct mosq_config mosq_cfg;
senhub_info_t *pshinfo;
g_SensorHubList = NULL;
//ADV_INFO("%s: \n", __func__);
rc = mqtt_client_config_load(&g_mosq_cfg, CLIENT_SUB, 1, NULL);
if(rc){
mqtt_client_config_cleanup(&g_mosq_cfg);
return 1;
}
mosquitto_lib_init();
if(mqtt_client_id_generate(&g_mosq_cfg, "advmqttcli")){
return 1;
}
g_mosq = mosquitto_new(g_mosq_cfg.id, g_mosq_cfg.clean_session, &g_mosq_cfg);
if(!g_mosq){
switch(errno){
case ENOMEM:
if(!g_mosq_cfg.quiet) fprintf(stderr, "Error: Out of memory.\n");
break;
case EINVAL:
if(!g_mosq_cfg.quiet) fprintf(stderr, "Error: Invalid id and/or clean_session.\n");
break;
}
mosquitto_lib_cleanup();
return 1;
}
if(mqtt_client_opts_set(g_mosq, &g_mosq_cfg)){
return 1;
}
mosquitto_connect_callback_set(g_mosq, MqttHal_Connect_Callback);
mosquitto_message_callback_set(g_mosq, MqttHal_Message_Callback);
mosquitto_disconnect_callback_set(g_mosq, MqttHal_Disconnect_Callback);
mosquitto_publish_callback_set(g_mosq, MqttHal_Publish_Callback);
// Create senhub root
pshinfo = malloc(sizeof(senhub_info_t));
memset(pshinfo, 0, sizeof(senhub_info_t));
sprintf(pshinfo->macAddress, "0000%s" , g_GWInfMAC);
pshinfo->jsonNode = NULL;
pshinfo->id = senhub_list_newId(g_SensorHubList);
//printf("%s: list add id=%d\n", __func__, pshinfo->id);
g_SensorHubList = SENHUB_LIST_ADD(g_SensorHubList, pshinfo);
return rc;
}
mosquittopp::mosquittopp(const char *id, bool clean_session)
{
m_mosq = mosquitto_new(id, clean_session, this);
mosquitto_connect_callback_set(m_mosq, on_connect_wrapper);
mosquitto_disconnect_callback_set(m_mosq, on_disconnect_wrapper);
mosquitto_publish_callback_set(m_mosq, on_publish_wrapper);
mosquitto_message_callback_set(m_mosq, on_message_wrapper);
mosquitto_subscribe_callback_set(m_mosq, on_subscribe_wrapper);
mosquitto_unsubscribe_callback_set(m_mosq, on_unsubscribe_wrapper);
mosquitto_log_callback_set(m_mosq, on_log_wrapper);
}
static
int
mosq_callback_set(lua_State *L)
{
mosq_t *ctx = mosq_get(L, 1);
enum callback_types callback_type = luaL_checkinteger(L, 2);
if (!lua_isfunction(L, 3)) {
return luaL_argerror(L, 3, "expecting a function");
}
int ref = luaL_ref(L, LUA_REGISTRYINDEX);
switch (callback_type) {
case CONNECT:
ctx->connect_ref = ref;
mosquitto_connect_callback_set(ctx->mosq, mosq_connect_f);
break;
case DISCONNECT:
ctx->disconnect_ref = ref;
mosquitto_disconnect_callback_set(ctx->mosq, mosq_disconnect_f);
break;
case PUBLISH:
ctx->publish_ref = ref;
mosquitto_publish_callback_set(ctx->mosq, mosq_publish_f);
break;
case MESSAGE:
ctx->message_ref = ref;
mosquitto_message_callback_set(ctx->mosq, mosq_message_f);
break;
case SUBSCRIBE:
ctx->subscribe_ref = ref;
mosquitto_subscribe_callback_set(ctx->mosq, mosq_subscribe_f);
break;
case UNSUBSCRIBE:
ctx->unsubscribe_ref = ref;
mosquitto_unsubscribe_callback_set(ctx->mosq, mosq_unsubscribe_f);
break;
default:
luaL_unref(L, LUA_REGISTRYINDEX, ref);
luaL_argerror(L, 2, "unknown mosquitto callback type");
break;
}
return make_mosq_status_result(L, MOSQ_ERR_SUCCESS);
}
bool connect() {
struct mosquitto *m = mosquitto_new(options.clientId, true, this);
this->connection = m;
mosquitto_connect_callback_set(m, on_connect);
mosquitto_publish_callback_set(m, on_publish);
mosquitto_subscribe_callback_set(m, on_subscribe);
mosquitto_message_callback_set(m, on_message);
const int res = mosquitto_connect(m, options.brokerHostname, options.brokerPort,
options.keepaliveSeconds);
return res == MOSQ_ERR_SUCCESS;
}
int mosquittopp::reinitialise(const char *id, bool clean_session)
{
int rc;
rc = mosquitto_reinitialise(m_mosq, id, clean_session, this);
if(rc == MOSQ_ERR_SUCCESS){
mosquitto_connect_callback_set(m_mosq, on_connect_wrapper);
mosquitto_disconnect_callback_set(m_mosq, on_disconnect_wrapper);
mosquitto_publish_callback_set(m_mosq, on_publish_wrapper);
mosquitto_message_callback_set(m_mosq, on_message_wrapper);
mosquitto_subscribe_callback_set(m_mosq, on_subscribe_wrapper);
mosquitto_unsubscribe_callback_set(m_mosq, on_unsubscribe_wrapper);
mosquitto_log_callback_set(m_mosq, on_log_wrapper);
}
return rc;
}
int main(int argc, char *argv[])
{
char *host = "5.44.237.19";
int port = 1883;
int keepalive = 60;
bool clean_session = true;
struct mosquitto *mosq = NULL;
/* argv[1] is the filename, argv[2] is the number of MAC addresses */
/* and argv[3] is the timestamp in secs since the epoch */
if(argc==4) {
macFile=argv[1];
messagesToSend=atoi(argv[2]);
if(messagesToSend==0) {
return 1;
}
timeNow=argv[3];
} else {
printf("USAGE: thingithonmq macFileName numRecords timeStamp\n Where timeStamp is in seconds, like date +\%s\n");
return 1;
}
/* Start up Mosquitto */
mosquitto_lib_init();
mosq = mosquitto_new(NULL, clean_session, NULL);
if(!mosq){
fprintf(stderr, "Error: Out of memory.\n");
return 1;
}
mosquitto_log_callback_set(mosq, my_log_callback);
mosquitto_connect_callback_set(mosq, my_connect_callback);
mosquitto_publish_callback_set(mosq, my_publish_callback);
if(mosquitto_connect(mosq, host, port, keepalive)){
fprintf(stderr, "Unable to connect.\n");
return 1;
}
mosquitto_loop_forever(mosq, -1, 1);
mosquitto_destroy(mosq);
mosquitto_lib_cleanup();
return 0;
}
int main(int argc, char *argv[])
{
int rc;
struct mosquitto *mosq;
mosquitto_lib_init();
mosq = mosquitto_new("publish-qos0-test", true, NULL);
mosquitto_connect_callback_set(mosq, on_connect);
mosquitto_publish_callback_set(mosq, on_publish);
rc = mosquitto_connect(mosq, "localhost", 1888, 60);
while(run == -1){
mosquitto_loop(mosq, -1, 1);
}
mosquitto_lib_cleanup();
return run;
}
int mqtt_open(mqtt_t inst, dispatch_t d, const char* name, const char* host, int port, int keepalive) {
int rval = -1;
int fd;
inst->mosq = mosquitto_new(name, true, inst);
if (inst->mosq == NULL) {
error("Cannot create mqtt");
} else {
inst->dispatch = d;
mosquitto_log_callback_set(inst->mosq, on_log);
mosquitto_connect_callback_set(inst->mosq, on_connect);
mosquitto_disconnect_callback_set(inst->mosq, on_disconnect);
mosquitto_publish_callback_set(inst->mosq, on_publish);
mosquitto_message_callback_set(inst->mosq, on_message);
mosquitto_subscribe_callback_set(inst->mosq, on_subscribe);
mosquitto_unsubscribe_callback_set(inst->mosq, on_unsubscribe);
inst->timer = dispatch_create_timer(d, 2000000, mqtt_misc, inst);
if (inst->timer == NULL) {
error("Cannot create timer");
}
rval = mosquitto_connect(inst->mosq, host, port, keepalive);
if (rval != MOSQ_ERR_SUCCESS) {
error("Cannot connect mqtt");
} else {
fd = mosquitto_socket(inst->mosq);
rval = dispatch_register(d, fd, mqtt_read, mqtt_write, NULL, mqtt_close, inst);
if (rval != 0) {
error("Cannot register mqtt");
}
}
}
return rval;
}
int main(int argc, char *argv[])
{
int rc;
struct mosquitto *mosq;
mosquitto_lib_init();
mosq = mosquitto_new("08-tls-psk-pub", true, NULL);
rc = mosquitto_tls_psk_set(mosq, "deadbeef", "psk-id", NULL);
if(rc) return rc;
mosquitto_connect_callback_set(mosq, on_connect);
mosquitto_disconnect_callback_set(mosq, on_disconnect);
mosquitto_publish_callback_set(mosq, on_publish);
rc = mosquitto_connect(mosq, "localhost", 1888, 60);
if(rc) return rc;
while(run == -1){
mosquitto_loop(mosq, -1, 1);
}
mosquitto_lib_cleanup();
return run;
}
static int ctx_callback_set(lua_State *L)
{
ctx_t *ctx = ctx_check(L, 1);
int callback_type;
if (lua_isstring(L, 2)) {
callback_type = callback_type_from_string(lua_tostring(L, 2));
} else {
callback_type = luaL_checkinteger(L, 2);
}
if (!lua_isfunction(L, 3)) {
return luaL_argerror(L, 3, "expecting a callback function");
}
/* pop the function from the stack and store it in the registry */
int ref = luaL_ref(L, LUA_REGISTRYINDEX);
switch (callback_type) {
case CONNECT:
ctx->on_connect = ref;
mosquitto_connect_callback_set(ctx->mosq, ctx_on_connect);
break;
case DISCONNECT:
ctx->on_disconnect = ref;
mosquitto_disconnect_callback_set(ctx->mosq, ctx_on_disconnect);
break;
case PUBLISH:
ctx->on_publish = ref;
mosquitto_publish_callback_set(ctx->mosq, ctx_on_publish);
break;
case MESSAGE:
/* store the reference into the context, to be retrieved by ctx_on_message */
ctx->on_message = ref;
/* register C callback in mosquitto */
mosquitto_message_callback_set(ctx->mosq, ctx_on_message);
break;
case SUBSCRIBE:
ctx->on_subscribe = ref;
mosquitto_subscribe_callback_set(ctx->mosq, ctx_on_subscribe);
break;
case UNSUBSCRIBE:
ctx->on_unsubscribe = ref;
mosquitto_unsubscribe_callback_set(ctx->mosq, ctx_on_unsubscribe);
break;
case LOG:
ctx->on_log = ref;
mosquitto_log_callback_set(ctx->mosq, ctx_on_log);
break;
default:
luaL_unref(L, LUA_REGISTRYINDEX, ref);
luaL_argerror(L, 2, "not a proper callback type");
break;
}
return mosq__pstatus(L, MOSQ_ERR_SUCCESS);
}
/*
* ##########################
* Main Function
* ##########################
*/
int main(int argc, char *argv[])
{
/* program setup variable */
struct mosquitto *mosq = NULL;
struct mqtt_userdata *ud = NULL;
unsigned int max_inflight = 20;
bool debug = false;
char buf[MQTT_BUFSIZE];
char err[MQTT_ERR_BUFSIZE];
/* client id */
char id[MQTT_ID_LEN];
char id_prefix[MQTT_ID_LEN];
char hostname[MQTT_HOSTNAME_BUFSIZE];
/* broker variable */
char host[MQTT_IP_LEN] = "127.0.0.1";
int port = 1883;
int keepalive = 60;
/* will information */
char *will_topic = NULL;
long will_payloadlen = 0;
char *will_payload = NULL;
int will_qos = 0;
bool will_retain = false;
/* temp variable */
int i;
int rc;
int rc2;
/* initialized program and user data structure */
ud = malloc(sizeof(struct mqtt_userdata));
memset(ud, 0, sizeof(struct mqtt_userdata));
ud->last_mid = -1;
ud->connected = true;
memset(id, '\0', sizeof(id));
memset(id_prefix, '\0', sizeof(id_prefix));
ud->qos = 2;
/* get option */
for(i=1; i<argc; i++){
if(!strcmp(argv[i], "-p") || !strcmp(argv[i], "--port")){
if(i==argc-1){
fprintf(stderr, "Error: -p argument given but no port specified.\n\n");
mqtt_print_usage();
return 1;
}else{
port = atoi(argv[i+1]);
if(port<1 || port>65535){
fprintf(stderr, "Error: Invalid port given: %d\n", port);
mqtt_print_usage();
return 1;
}
}
i++;
}else if(!strcmp(argv[i], "-d") || !strcmp(argv[i], "--debug")){
debug = true;
}else if(!strcmp(argv[i], "-f") || !strcmp(argv[i], "--file")){
if(ud->mode != MQTT_MSGMODE_NONE){
fprintf(stderr, "Error: Only one type of message can be sent at once.\n\n");
mqtt_print_usage();
return 1;
}else if(i==argc-1){
fprintf(stderr, "Error: -f argument given but no file specified.\n\n");
mqtt_print_usage();
return 1;
}else{
if(mqtt_load_file(argv[i+1], ud)) return 1;
}
i++;
}else if(!strcmp(argv[i], "--help")){
mqtt_print_usage();
return 0;
}else if(!strcmp(argv[i], "-h") || !strcmp(argv[i], "--host")){
if(i==argc-1){
fprintf(stderr, "Error: -h argument given but no host specified.\n\n");
mqtt_print_usage();
return 1;
}else{
if (strlen(argv[i+1]) >= MQTT_IP_LEN) {
fprintf(stderr, "Error: max length of ip is %d.\n\n", MQTT_IP_LEN);
mqtt_print_usage();
} else {
memset(host, '\0', sizeof(host));
strcpy(host, argv[i+1]);
}
}
i++;
}else if(!strcmp(argv[i], "-i") || !strcmp(argv[i], "--id")){
if(strlen(id_prefix) != 0){
fprintf(stderr, "Error: -i and -I argument cannot be used together.\n\n");
mqtt_print_usage();
return 1;
}
if(i==argc-1){
fprintf(stderr, "Error: -i argument given but no id specified.\n\n");
mqtt_print_usage();
return 1;
}else{
if (strlen(argv[i+1]) >= MOSQ_MQTT_ID_MAX_LENGTH) {
fprintf(stderr, "Error: max length of client id is %d.\n\n", MOSQ_MQTT_ID_MAX_LENGTH);
mqtt_print_usage();
} else {
strcpy(id, argv[i+1]);
}
}
i++;
}else if(!strcmp(argv[i], "-I") || !strcmp(argv[i], "--id-prefix")){
if(strlen(id) != 0){
fprintf(stderr, "Error: -i and -I argument cannot be used together.\n\n");
mqtt_print_usage();
return 1;
}
if(i==argc-1){
fprintf(stderr, "Error: -I argument given but no id prefix specified.\n\n");
mqtt_print_usage();
return 1;
}else{
if (strlen(argv[i+1]) >= MOSQ_MQTT_ID_MAX_LENGTH) {
fprintf(stderr, "Error: max length of client id is %d.\n\n", MOSQ_MQTT_ID_MAX_LENGTH);
mqtt_print_usage();
} else {
strcpy(id_prefix, argv[i+1]);
}
}
i++;
}else if(!strcmp(argv[i], "-l") || !strcmp(argv[i], "--stdin-line")){
if(ud->mode != MQTT_MSGMODE_NONE){
fprintf(stderr, "Error: Only one type of message can be sent at once.\n\n");
mqtt_print_usage();
return 1;
}else{
ud->mode = MQTT_MSGMODE_STDIN_LINE;
}
}else if(!strcmp(argv[i], "-m") || !strcmp(argv[i], "--message")){
if(ud->mode != MQTT_MSGMODE_NONE){
fprintf(stderr, "Error: Only one type of message can be sent at once.\n\n");
mqtt_print_usage();
return 1;
}else if(i==argc-1){
fprintf(stderr, "Error: -m argument given but no message specified.\n\n");
mqtt_print_usage();
return 1;
}else{
ud->message = argv[i+1];
ud->msglen = strlen(ud->message);
ud->mode = MQTT_MSGMODE_CMD;
}
i++;
}else if(!strcmp(argv[i], "-M")){
if(i==argc-1){
fprintf(stderr, "Error: -M argument given but max_inflight not specified.\n\n");
mqtt_print_usage();
return 1;
}else{
max_inflight = atoi(argv[i+1]);
}
i++;
}else if(!strcmp(argv[i], "-n") || !strcmp(argv[i], "--null-message")){
if(ud->mode != MQTT_MSGMODE_NONE){
fprintf(stderr, "Error: Only one type of message can be sent at once.\n\n");
mqtt_print_usage();
return 1;
}else{
ud->mode = MQTT_MSGMODE_NULL;
}
}else if(!strcmp(argv[i], "-q") || !strcmp(argv[i], "--qos")){
if(i==argc-1){
fprintf(stderr, "Error: -q argument given but no QoS specified.\n\n");
mqtt_print_usage();
return 1;
}else{
ud->qos = atoi(argv[i+1]);
if(ud->qos<0 || ud->qos>2){
fprintf(stderr, "Error: Invalid QoS given: %d\n", ud->qos);
mqtt_print_usage();
return 1;
}
}
i++;
}else if(!strcmp(argv[i], "--quiet")){
ud->quiet = true;
}else if(!strcmp(argv[i], "-r") || !strcmp(argv[i], "--retain")){
ud->retain = 1;
}else if(!strcmp(argv[i], "-s") || !strcmp(argv[i], "--stdin-file")){
if(ud->mode != MQTT_MSGMODE_NONE){
fprintf(stderr, "Error: Only one type of message can be sent at once.\n\n");
mqtt_print_usage();
return 1;
}else{
if(mqtt_load_stdin(ud)) return 1;
}
}else if(!strcmp(argv[i], "-t") || !strcmp(argv[i], "--topic")){
if(i==argc-1){
fprintf(stderr, "Error: -t argument given but no topic specified.\n\n");
mqtt_print_usage();
return 1;
}else{
ud->topic = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "-u") || !strcmp(argv[i], "--username")){
if(i==argc-1){
fprintf(stderr, "Error: -u argument given but no username specified.\n\n");
mqtt_print_usage();
return 1;
}else{
ud->username = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "-P") || !strcmp(argv[i], "--pw")){
if(i==argc-1){
fprintf(stderr, "Error: -P argument given but no password specified.\n\n");
mqtt_print_usage();
return 1;
}else{
ud->password = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "--will-payload")){
if(i==argc-1){
fprintf(stderr, "Error: --will-payload argument given but no will payload specified.\n\n");
mqtt_print_usage();
return 1;
}else{
will_payload = argv[i+1];
will_payloadlen = strlen(will_payload);
}
i++;
}else if(!strcmp(argv[i], "--will-qos")){
if(i==argc-1){
fprintf(stderr, "Error: --will-qos argument given but no will QoS specified.\n\n");
mqtt_print_usage();
return 1;
}else{
will_qos = atoi(argv[i+1]);
if(will_qos < 0 || will_qos > 2){
fprintf(stderr, "Error: Invalid will QoS %d.\n\n", will_qos);
return 1;
}
}
i++;
}else if(!strcmp(argv[i], "--will-retain")){
will_retain = true;
}else if(!strcmp(argv[i], "--will-topic")){
if(i==argc-1){
fprintf(stderr, "Error: --will-topic argument given but no will topic specified.\n\n");
mqtt_print_usage();
return 1;
}else{
will_topic = argv[i+1];
}
i++;
}else{
fprintf(stderr, "Error: Unknown option '%s'.\n",argv[i]);
mqtt_print_usage();
return 1;
}
}
/* verify necessary variable */
if(!ud->topic || ud->mode == MQTT_MSGMODE_NONE){
fprintf(stderr, "Error: Both topic and message must be supplied.\n");
mqtt_print_usage();
return 1;
}
if(will_payload && !will_topic){
fprintf(stderr, "Error: Will payload given, but no will topic given.\n");
mqtt_print_usage();
return 1;
}
if(will_retain && !will_topic){
fprintf(stderr, "Error: Will retain given, but no will topic given.\n");
mqtt_print_usage();
return 1;
}
if(ud->password && !ud->username){
if(!ud->quiet) fprintf(stderr, "Warning: Not using password since username not set.\n");
}
/* init mosquitto library */
mosquitto_lib_init();
/* setup client id */
if(strlen(id_prefix) != 0){
snprintf(id, sizeof(id), "%s%d", id_prefix, getpid());
}else if(strlen(id) == 0){
memset(hostname, '\0', sizeof(hostname));
gethostname(hostname, sizeof(hostname));
snprintf(id, sizeof(id), "mosqub/%d-%s", getpid(), hostname);
}
if(strlen(id) > MOSQ_MQTT_ID_MAX_LENGTH){
/* Enforce maximum client id length of 23 characters */
id[MOSQ_MQTT_ID_MAX_LENGTH] = '\0';
}
/* start mosquitto */
mosq = mosquitto_new(id, true, ud);
if(!mosq){
if(!ud->quiet) fprintf(stderr, "Error: %s\n", strerror(errno));
mosquitto_lib_cleanup();
return 1;
}
/* setup mosquitto */
if(debug){
mosquitto_log_callback_set(mosq, mqtt_log_callback);
}
if(will_topic && mosquitto_will_set(mosq, will_topic, will_payloadlen, will_payload, will_qos, will_retain)){
if(!ud->quiet) fprintf(stderr, "Error: Problem setting will.\n");
mosquitto_lib_cleanup();
return 1;
}
if(ud->username && mosquitto_username_pw_set(mosq, ud->username, ud->password)){
if(!ud->quiet) fprintf(stderr, "Error: Problem setting username and password.\n");
mosquitto_lib_cleanup();
return 1;
}
mosquitto_max_inflight_messages_set(mosq, max_inflight);
mosquitto_connect_callback_set(mosq, mqtt_connect_callback);
mosquitto_disconnect_callback_set(mosq, mqtt_disconnect_callback);
mosquitto_publish_callback_set(mosq, mqtt_publish_callback);
/* connect mosquitto */
rc = mosquitto_connect(mosq, host, port, keepalive);
if(rc){
if(!ud->quiet){
if(rc == MOSQ_ERR_ERRNO){
#ifndef WIN32
strerror_r(errno, err, sizeof(err));
#else
FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, NULL, errno, 0, (LPTSTR)&err, sizeof(err), NULL);
#endif
fprintf(stderr, "Error: %s\n", err);
}else{
fprintf(stderr, "Unable to connect (%d: %s).\n", rc, mosquitto_strerror(rc));
}
}
mosquitto_lib_cleanup();
return rc;
}
/* publish mosquitto mqtt message BASED ON DIFFERENT message mode */
if(ud->mode == MQTT_MSGMODE_STDIN_LINE){
mosquitto_loop_start(mosq);
}
do{
if(ud->mode == MQTT_MSGMODE_STDIN_LINE){
if(ud->status == MQTT_STATUS_CONNACK_RECVD){
if(fgets(buf, sizeof(buf), stdin)){
buf[strlen(buf)-1] = '\0';
rc2 = mosquitto_publish(mosq, &ud->mid_sent, ud->topic, strlen(buf), buf, ud->qos, ud->retain);
if(rc2){
if(!ud->quiet) fprintf(stderr, "Error: Publish returned %d, disconnecting.\n", rc2);
mosquitto_disconnect(mosq);
}
}else if(feof(stdin)){
ud->last_mid = ud->mid_sent;
ud->status = MQTT_STATUS_WAITING;
}
}else if(ud->status == MQTT_STATUS_WAITING){
#ifdef WIN32
Sleep(1000);
#else
usleep(1000000);
#endif
}
rc = MOSQ_ERR_SUCCESS;
}else{
rc = mosquitto_loop(mosq, -1, 1);
}
}while(rc == MOSQ_ERR_SUCCESS && ud->connected);
if(ud->mode == MQTT_MSGMODE_STDIN_LINE){
mosquitto_loop_stop(mosq, false);
}
/* free mosquitto */
mosquitto_destroy(mosq);
mosquitto_lib_cleanup();
mqtt_userdata_free(ud);
return rc;
}
int main(int argc, char **argv)
{
struct mosquitto *mosq;
mosquitto_lib_init();
mosq = mosquitto_new("id", true, NULL);
mosquitto_publish_callback_set(mosq, on_publish);
mosquitto_connect(mosq, "localhost", 1883, 0);
RF24 radio(RPI_V2_GPIO_P1_22, BCM2835_SPI_CS0, BCM2835_SPI_SPEED_4MHZ);
const uint8_t rxpipe[5] = {'N','o','d','e','0'};
uint8_t payload[32];
// char humidity[4];
// char temperature[4];
radio.begin();
radio.enableDynamicPayloads();
radio.setRetries(5,15);
radio.printDetails();
radio.openReadingPipe(1,rxpipe);
radio.startListening();
while (1) {
if ( radio.available() )
{
// Dump the payloads until we've gotten everything
uint8_t len;
while (radio.available())
{
// Fetch the payload, and see if this was the last one.
len = radio.getDynamicPayloadSize();
radio.read( payload, len );
// Put a zero at the end for easy printing
// sprintf(humidity, "%d", payload[0]);
// sprintf(temperature, "%d", payload[1]);
// Spew it
printf("Got payload size=%i value=%d,%d\n\r",len,payload[0],payload[1]);
// mosquitto_publish(mosq, NULL, "nodes/01/humidity", strlen(humidity), humidity, 0, false);
// mosquitto_publish(mosq, NULL, "nodes/01/temperature", strlen(temperature), temperature, 0, false);
mosquitto_publish(mosq, NULL, "nodes/01/humidity", 1, &payload[0], 0, false);
mosquitto_publish(mosq, NULL, "nodes/01/temperature", 1, &payload[1], 0, false);
mosquitto_loop(mosq, -1, 1); // this calls mosquitto_loop() in a loop, it will exit once the client disconnects cleanly
}
}
sleep(1);
}
return 0;
}
int main(int argc, char **argv)
{
struct tm gmt_init, local_init;
char ch, *progname = *argv, *nodename, *topic, *prefix = DEFAULT_PREFIX;;
int usage = 0, interval = 10, rc;
struct utsname uts;
char clientid[30];
char *host = "localhost", *ca_file;
int port = 1883, keepalive = 60;
int do_tls = FALSE, tls_insecure = FALSE;
int do_psk = FALSE;
char *psk_key = NULL, *psk_identity = NULL;
struct udata udata;
udata.mid = 17;
while ((ch = getopt(argc, argv, "i:t:h:p:C:LUK:I:")) != EOF) {
switch (ch) {
case 'C':
ca_file = optarg;
do_tls = TRUE;
break;
case 'h':
host = optarg;
break;
case 'i':
interval = atoi(optarg);
interval = (interval < 1) ? 1 : interval;
break;
case 's':
tls_insecure = TRUE;
break;
case 'p':
port = atoi(optarg);
break;
case 't':
prefix = optarg;
break;
case 'L':
do_local = !do_local;
break;
case 'U':
do_utc = !do_utc;
break;
case 'I':
psk_identity = optarg;
do_psk = TRUE;
break;
case 'K':
psk_key = optarg;
do_psk = TRUE;
break;
default:
usage = 1;
break;
}
}
if (do_tls && do_psk)
usage = 1;
if (do_psk && (psk_key == NULL || psk_identity == NULL))
usage = 1;
if (usage) {
fprintf(stderr, "Usage: %s [-h host] [-i interval] [-p port] [-t prefix] [-C CA-cert] [-L] [-U] [-K psk-key] [-I psk-identity] [-s]\n", progname);
exit(1);
}
/* Find nodename; chop at first '.' */
if (uname(&uts) == 0) {
char *p;
nodename = strdup(uts.nodename);
if ((p = strchr(nodename, '.')) != NULL)
*p = 0;
} else {
nodename = strdup("unknown");
}
mosquitto_lib_init();
sprintf(clientid, "mqtt-tics-%d", getpid());
m = mosquitto_new(clientid, TRUE, (void *)&udata);
if (!m) {
fprintf(stderr, "Out of memory.\n");
exit(1);
}
if (do_psk) {
rc = mosquitto_tls_psk_set(m, psk_key, psk_identity,NULL);
if (rc != MOSQ_ERR_SUCCESS) {
fprintf(stderr, "Cannot set TLS PSK: %s\n",
mosquitto_strerror(rc));
exit(3);
}
} else if (do_tls) {
rc = mosquitto_tls_set(m, ca_file, NULL, NULL, NULL, NULL);
if (rc != MOSQ_ERR_SUCCESS) {
fprintf(stderr, "Cannot set TLS PSK: %s\n",
mosquitto_strerror(rc));
exit(3);
}
/* FIXME */
// mosquitto_tls_opts_set(m, SSL_VERIFY_PEER, "tlsv1", NULL);
if (tls_insecure) {
#if LIBMOSQUITTO_VERSION_NUMBER >= 1002000
/* mosquitto_tls_insecure_set() requires libmosquitto 1.2. */
mosquitto_tls_insecure_set(m, TRUE);
#endif
}
}
mosquitto_publish_callback_set(m, cb_pub);
mosquitto_disconnect_callback_set(m, cb_disconnect);
if ((rc = mosquitto_connect(m, host, port, keepalive)) != MOSQ_ERR_SUCCESS) {
fprintf(stderr, "Unable to connect to %s:%d: %s\n", host, port,
mosquitto_strerror(rc));
perror("");
exit(2);
}
signal(SIGINT, catcher);
mosquitto_loop_start(m);
if ((topic = malloc(strlen(prefix) + (3 * strlen(nodename)) + 12)) == NULL) {
fprintf(stderr, "ENOMEM\n");
goto abort;
}
sprintf(topic, prefix, nodename, nodename, nodename);
memset(&gmt_init, 0, sizeof(struct tm));
memset(&local_init, 0, sizeof(struct tm));
time(&start_tics);
while (1) {
pubtime(topic, &gmt_init, &local_init);
sleep(interval);
}
free(topic);
abort:
free(nodename);
mosquitto_disconnect(m);
mosquitto_loop_stop(m, false);
mosquitto_lib_cleanup();
return 0;
}
int main(int argc, char *argv[])
{
char *id = NULL;
char *id_prefix = NULL;
int i;
char *host = "localhost";
int port = 1883;
int keepalive = 60;
char buf[1024];
bool debug = false;
struct mosquitto *mosq = NULL;
int rc;
int rc2;
char hostname[256];
char err[1024];
int len;
char *will_payload = NULL;
long will_payloadlen = 0;
int will_qos = 0;
bool will_retain = false;
char *will_topic = NULL;
char *cafile = NULL;
char *capath = NULL;
char *certfile = NULL;
char *keyfile = NULL;
char *psk = NULL;
char *psk_identity = NULL;
for(i=1; i<argc; i++){
if(!strcmp(argv[i], "-p") || !strcmp(argv[i], "--port")){
if(i==argc-1){
fprintf(stderr, "Error: -p argument given but no port specified.\n\n");
print_usage();
return 1;
}else{
port = atoi(argv[i+1]);
if(port<1 || port>65535){
fprintf(stderr, "Error: Invalid port given: %d\n", port);
print_usage();
return 1;
}
}
i++;
}else if(!strcmp(argv[i], "--cafile")){
if(i==argc-1){
fprintf(stderr, "Error: --cafile argument given but no file specified.\n\n");
print_usage();
return 1;
}else{
cafile = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "--capath")){
if(i==argc-1){
fprintf(stderr, "Error: --capath argument given but no directory specified.\n\n");
print_usage();
return 1;
}else{
capath = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "--cert")){
if(i==argc-1){
fprintf(stderr, "Error: --cert argument given but no file specified.\n\n");
print_usage();
return 1;
}else{
certfile = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "-d") || !strcmp(argv[i], "--debug")){
debug = true;
}else if(!strcmp(argv[i], "-f") || !strcmp(argv[i], "--file")){
if(mode != MSGMODE_NONE){
fprintf(stderr, "Error: Only one type of message can be sent at once.\n\n");
print_usage();
return 1;
}else if(i==argc-1){
fprintf(stderr, "Error: -f argument given but no file specified.\n\n");
print_usage();
return 1;
}else{
if(load_file(argv[i+1])) return 1;
}
i++;
}else if(!strcmp(argv[i], "--help")){
print_usage();
return 0;
}else if(!strcmp(argv[i], "-h") || !strcmp(argv[i], "--host")){
if(i==argc-1){
fprintf(stderr, "Error: -h argument given but no host specified.\n\n");
print_usage();
return 1;
}else{
host = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "-i") || !strcmp(argv[i], "--id")){
if(id_prefix){
fprintf(stderr, "Error: -i and -I argument cannot be used together.\n\n");
print_usage();
return 1;
}
if(i==argc-1){
fprintf(stderr, "Error: -i argument given but no id specified.\n\n");
print_usage();
return 1;
}else{
id = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "-I") || !strcmp(argv[i], "--id-prefix")){
if(id){
fprintf(stderr, "Error: -i and -I argument cannot be used together.\n\n");
print_usage();
return 1;
}
if(i==argc-1){
fprintf(stderr, "Error: -I argument given but no id prefix specified.\n\n");
print_usage();
return 1;
}else{
id_prefix = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "--key")){
if(i==argc-1){
fprintf(stderr, "Error: --key argument given but no file specified.\n\n");
print_usage();
return 1;
}else{
keyfile = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "-l") || !strcmp(argv[i], "--stdin-line")){
if(mode != MSGMODE_NONE){
fprintf(stderr, "Error: Only one type of message can be sent at once.\n\n");
print_usage();
return 1;
}else{
mode = MSGMODE_STDIN_LINE;
}
}else if(!strcmp(argv[i], "-m") || !strcmp(argv[i], "--message")){
if(mode != MSGMODE_NONE){
fprintf(stderr, "Error: Only one type of message can be sent at once.\n\n");
print_usage();
return 1;
}else if(i==argc-1){
fprintf(stderr, "Error: -m argument given but no message specified.\n\n");
print_usage();
return 1;
}else{
message = argv[i+1];
msglen = strlen(message);
mode = MSGMODE_CMD;
}
i++;
}else if(!strcmp(argv[i], "-n") || !strcmp(argv[i], "--null-message")){
if(mode != MSGMODE_NONE){
fprintf(stderr, "Error: Only one type of message can be sent at once.\n\n");
print_usage();
return 1;
}else{
mode = MSGMODE_NULL;
}
}else if(!strcmp(argv[i], "--psk")){
if(i==argc-1){
fprintf(stderr, "Error: --psk argument given but no key specified.\n\n");
print_usage();
return 1;
}else{
psk = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "--psk-identity")){
if(i==argc-1){
fprintf(stderr, "Error: --psk-identity argument given but no identity specified.\n\n");
print_usage();
return 1;
}else{
psk_identity = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "-q") || !strcmp(argv[i], "--qos")){
if(i==argc-1){
fprintf(stderr, "Error: -q argument given but no QoS specified.\n\n");
print_usage();
return 1;
}else{
qos = atoi(argv[i+1]);
if(qos<0 || qos>2){
fprintf(stderr, "Error: Invalid QoS given: %d\n", qos);
print_usage();
return 1;
}
}
i++;
}else if(!strcmp(argv[i], "--quiet")){
quiet = true;
}else if(!strcmp(argv[i], "-r") || !strcmp(argv[i], "--retain")){
retain = 1;
}else if(!strcmp(argv[i], "-s") || !strcmp(argv[i], "--stdin-file")){
if(mode != MSGMODE_NONE){
fprintf(stderr, "Error: Only one type of message can be sent at once.\n\n");
print_usage();
return 1;
}else{
if(load_stdin()) return 1;
}
}else if(!strcmp(argv[i], "-t") || !strcmp(argv[i], "--topic")){
if(i==argc-1){
fprintf(stderr, "Error: -t argument given but no topic specified.\n\n");
print_usage();
return 1;
}else{
topic = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "-u") || !strcmp(argv[i], "--username")){
if(i==argc-1){
fprintf(stderr, "Error: -u argument given but no username specified.\n\n");
print_usage();
return 1;
}else{
username = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "-P") || !strcmp(argv[i], "--pw")){
if(i==argc-1){
fprintf(stderr, "Error: -P argument given but no password specified.\n\n");
print_usage();
return 1;
}else{
password = argv[i+1];
}
i++;
}else if(!strcmp(argv[i], "--will-payload")){
if(i==argc-1){
fprintf(stderr, "Error: --will-payload argument given but no will payload specified.\n\n");
print_usage();
return 1;
}else{
will_payload = argv[i+1];
will_payloadlen = strlen(will_payload);
}
i++;
}else if(!strcmp(argv[i], "--will-qos")){
if(i==argc-1){
fprintf(stderr, "Error: --will-qos argument given but no will QoS specified.\n\n");
print_usage();
return 1;
}else{
will_qos = atoi(argv[i+1]);
if(will_qos < 0 || will_qos > 2){
fprintf(stderr, "Error: Invalid will QoS %d.\n\n", will_qos);
return 1;
}
}
i++;
}else if(!strcmp(argv[i], "--will-retain")){
will_retain = true;
}else if(!strcmp(argv[i], "--will-topic")){
if(i==argc-1){
fprintf(stderr, "Error: --will-topic argument given but no will topic specified.\n\n");
print_usage();
return 1;
}else{
will_topic = argv[i+1];
}
i++;
}else{
fprintf(stderr, "Error: Unknown option '%s'.\n",argv[i]);
print_usage();
return 1;
}
}
if(!topic || mode == MSGMODE_NONE){
fprintf(stderr, "Error: Both topic and message must be supplied.\n");
print_usage();
return 1;
}
if(will_payload && !will_topic){
fprintf(stderr, "Error: Will payload given, but no will topic given.\n");
print_usage();
return 1;
}
if(will_retain && !will_topic){
fprintf(stderr, "Error: Will retain given, but no will topic given.\n");
print_usage();
return 1;
}
if(password && !username){
if(!quiet) fprintf(stderr, "Warning: Not using password since username not set.\n");
}
if((certfile && !keyfile) || (keyfile && !certfile)){
fprintf(stderr, "Error: Both certfile and keyfile must be provided if one of them is.\n");
print_usage();
return 1;
}
if((cafile || capath) && psk){
if(!quiet) fprintf(stderr, "Error: Only one of --psk or --cafile/--capath may be used at once.\n");
return 1;
}
if(psk && !psk_identity){
if(!quiet) fprintf(stderr, "Error: --psk-identity required if --psk used.\n");
return 1;
}
mosquitto_lib_init();
if(id_prefix){
id = malloc(strlen(id_prefix)+10);
if(!id){
if(!quiet) fprintf(stderr, "Error: Out of memory.\n");
mosquitto_lib_cleanup();
return 1;
}
snprintf(id, strlen(id_prefix)+10, "%s%d", id_prefix, getpid());
}else if(!id){
hostname[0] = '\0';
gethostname(hostname, 256);
hostname[255] = '\0';
len = strlen("mosqpub/-") + 6 + strlen(hostname);
id = malloc(len);
if(!id){
if(!quiet) fprintf(stderr, "Error: Out of memory.\n");
mosquitto_lib_cleanup();
return 1;
}
snprintf(id, len, "mosqpub/%d-%s", getpid(), hostname);
if(strlen(id) > MOSQ_MQTT_ID_MAX_LENGTH){
/* Enforce maximum client id length of 23 characters */
id[MOSQ_MQTT_ID_MAX_LENGTH] = '\0';
}
}
mosq = mosquitto_new(id, true, NULL);
if(!mosq){
switch(errno){
case ENOMEM:
if(!quiet) fprintf(stderr, "Error: Out of memory.\n");
break;
case EINVAL:
if(!quiet) fprintf(stderr, "Error: Invalid id.\n");
break;
}
mosquitto_lib_cleanup();
return 1;
}
if(debug){
mosquitto_log_callback_set(mosq, my_log_callback);
}
if(will_topic && mosquitto_will_set(mosq, will_topic, will_payloadlen, will_payload, will_qos, will_retain)){
if(!quiet) fprintf(stderr, "Error: Problem setting will.\n");
mosquitto_lib_cleanup();
return 1;
}
if(username && mosquitto_username_pw_set(mosq, username, password)){
if(!quiet) fprintf(stderr, "Error: Problem setting username and password.\n");
mosquitto_lib_cleanup();
return 1;
}
if((cafile || capath) && mosquitto_tls_set(mosq, cafile, capath, certfile, keyfile, NULL)){
if(!quiet) fprintf(stderr, "Error: Problem setting TLS options.\n");
mosquitto_lib_cleanup();
return 1;
}
if(psk && mosquitto_tls_psk_set(mosq, psk, psk_identity, NULL)){
if(!quiet) fprintf(stderr, "Error: Problem setting TLS-PSK options.\n");
mosquitto_lib_cleanup();
return 1;
}
mosquitto_connect_callback_set(mosq, my_connect_callback);
mosquitto_disconnect_callback_set(mosq, my_disconnect_callback);
mosquitto_publish_callback_set(mosq, my_publish_callback);
rc = mosquitto_connect(mosq, host, port, keepalive);
if(rc){
if(!quiet){
if(rc == MOSQ_ERR_ERRNO){
#ifndef WIN32
strerror_r(errno, err, 1024);
#else
FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, NULL, errno, 0, (LPTSTR)&err, 1024, NULL);
#endif
fprintf(stderr, "Error: %s\n", err);
}else{
fprintf(stderr, "Unable to connect (%d).\n", rc);
}
}
mosquitto_lib_cleanup();
return rc;
}
if(mode == MSGMODE_STDIN_LINE){
mosquitto_loop_start(mosq);
}
do{
if(mode == MSGMODE_STDIN_LINE){
if(status == STATUS_CONNACK_RECVD){
if(fgets(buf, 1024, stdin)){
buf[strlen(buf)-1] = '\0';
rc2 = mosquitto_publish(mosq, &mid_sent, topic, strlen(buf), buf, qos, retain);
if(rc2){
if(!quiet) fprintf(stderr, "Error: Publish returned %d, disconnecting.\n", rc2);
mosquitto_disconnect(mosq);
}
}else if(feof(stdin)){
last_mid = mid_sent;
status = STATUS_WAITING;
}
}else if(status == STATUS_WAITING){
#ifdef WIN32
Sleep(1000);
#else
usleep(1000000);
#endif
}
rc = MOSQ_ERR_SUCCESS;
}else{
rc = mosquitto_loop(mosq, -1, 1);
}
}while(rc == MOSQ_ERR_SUCCESS && connected);
if(mode == MSGMODE_STDIN_LINE){
mosquitto_loop_stop(mosq, false);
}
if(message && mode == MSGMODE_FILE){
free(message);
}
mosquitto_destroy(mosq);
mosquitto_lib_cleanup();
if(rc){
if(rc == MOSQ_ERR_ERRNO){
fprintf(stderr, "Error: %s\n", strerror(errno));
}else{
fprintf(stderr, "Error: %s\n", mosquitto_strerror(rc));
}
}
return rc;
}
int main(int argc, char *argv[])
{
struct mosq_config cfg;
char buf[1024];
struct mosquitto *mosq = NULL;
int rc;
int rc2;
rc = client_config_load(&cfg, CLIENT_PUB, argc, argv);
if(rc){
client_config_cleanup(&cfg);
if(rc == 2){
/* --help */
print_usage();
}else{
fprintf(stderr, "\nUse 'mosquitto_pub --help' to see usage.\n");
}
return 1;
}
topic = cfg.topic;
message = cfg.message;
msglen = cfg.msglen;
qos = cfg.qos;
retain = cfg.retain;
mode = cfg.pub_mode;
username = cfg.username;
password = cfg.password;
quiet = cfg.quiet;
if(cfg.pub_mode == MSGMODE_STDIN_FILE){
if(load_stdin()){
fprintf(stderr, "Error loading input from stdin.\n");
return 1;
}
}else if(cfg.file_input){
if(load_file(cfg.file_input)){
fprintf(stderr, "Error loading input file \"%s\".\n", cfg.file_input);
return 1;
}
}
if(!topic || mode == MSGMODE_NONE){
fprintf(stderr, "Error: Both topic and message must be supplied.\n");
print_usage();
return 1;
}
mosquitto_lib_init();
if(client_id_generate(&cfg, "mosqpub")){
return 1;
}
mosq = mosquitto_new(cfg.id, true, NULL);
if(!mosq){
switch(errno){
case ENOMEM:
if(!quiet) fprintf(stderr, "Error: Out of memory.\n");
break;
case EINVAL:
if(!quiet) fprintf(stderr, "Error: Invalid id.\n");
break;
}
mosquitto_lib_cleanup();
return 1;
}
if(cfg.debug){
mosquitto_log_callback_set(mosq, my_log_callback);
}
mosquitto_connect_callback_set(mosq, my_connect_callback);
mosquitto_disconnect_callback_set(mosq, my_disconnect_callback);
mosquitto_publish_callback_set(mosq, my_publish_callback);
if(client_opts_set(mosq, &cfg)){
return 1;
}
rc = client_connect(mosq, &cfg);
if(rc) return rc;
if(mode == MSGMODE_STDIN_LINE){
mosquitto_loop_start(mosq);
}
do{
if(mode == MSGMODE_STDIN_LINE){
if(status == STATUS_CONNACK_RECVD){
if(fgets(buf, 1024, stdin)){
buf[strlen(buf)-1] = '\0';
rc2 = mosquitto_publish(mosq, &mid_sent, topic, strlen(buf), buf, qos, retain);
if(rc2){
if(!quiet) fprintf(stderr, "Error: Publish returned %d, disconnecting.\n", rc2);
mosquitto_disconnect(mosq);
}
}else if(feof(stdin)){
last_mid = mid_sent;
status = STATUS_WAITING;
}
}else if(status == STATUS_WAITING){
if(last_mid_sent == last_mid && disconnect_sent == false){
mosquitto_disconnect(mosq);
disconnect_sent = true;
}
#ifdef WIN32
Sleep(100);
#else
usleep(100000);
#endif
}
rc = MOSQ_ERR_SUCCESS;
}else{
rc = mosquitto_loop(mosq, -1, 1);
}
}while(rc == MOSQ_ERR_SUCCESS && connected);
if(mode == MSGMODE_STDIN_LINE){
mosquitto_loop_stop(mosq, false);
}
if(message && mode == MSGMODE_FILE){
free(message);
}
mosquitto_destroy(mosq);
mosquitto_lib_cleanup();
if(rc){
fprintf(stderr, "Error: %s\n", mosquitto_strerror(rc));
}
return rc;
}