version 1.1.1.1.2.11, 2011/11/24 01:13:28
|
version 1.1.1.1.2.29, 2011/12/14 15:09:31
|
Line 1
|
Line 1
|
#include "global.h" |
#include "global.h" |
#include "rtlm.h" |
#include "rtlm.h" |
|
#include "mqtt.h" |
|
#include "client.h" |
|
|
|
|
|
io_enableDEBUG; |
|
|
sl_config cfg; |
sl_config cfg; |
|
sqlite3 *acc, *pub; |
|
FILE *logg; |
|
extern char compiled[], compiledby[], compilehost[]; |
|
static char szCfgName[MAXPATHLEN]; |
|
intptr_t Kill; |
|
|
|
struct tagArgs *args; |
|
|
|
|
|
static void |
|
Usage(void) |
|
{ |
|
printf( " -= MQTT Client =- Publisher/Subscriber from ELWIX\n" |
|
"=== %s@%s === Compiled: %s ===\n\n" |
|
" Syntax: mqtt [options] <connect_to_broker[:port]> <ConnectID> [value_for_publish]\n\n" |
|
"\t-t <topic>\t\tPublish topic\n" |
|
"\t-s <topic[|QoS]>\tSubscribe for this topic, if wish add different |QoS to topic\n" |
|
"\t-q <QoS>\t\tQoS level (0-at most 1, 1-at least 1, 2-exactly 1)\n" |
|
"\t-d\t\t\tSend duplicate message\n" |
|
"\t-r\t\t\tRetain message from broker\n" |
|
"\t-C\t\t\tNot clear before connect!\n" |
|
"\t-c <config>\t\tService config\n" |
|
"\t-f\t\t\t'value_for_publish' is file instead text\n" |
|
"\t-p <port>\t\tDifferent port for connect (default: 1883)\n" |
|
"\t-T <timeout>\t\tKeep alive timeout in seconds\n" |
|
"\t-U <username>\t\tUsername\n" |
|
"\t-P <password>\t\tPassword\n" |
|
"\t-W <topic>\t\tWill Topic\n" |
|
"\t-M <message>\t\tWill Message\n" |
|
"\t-D\t\t\tDaemon mode\n" |
|
"\t-v\t\t\tVerbose (more -vvv, more verbose)\n" |
|
"\t-h\t\t\tHelp! This screen\n\n", |
|
compiledby, compilehost, compiled); |
|
} |
|
|
|
static void |
|
cleanArgs(struct tagArgs * __restrict args) |
|
{ |
|
mqtt_msgFree(&args->msg, 42); |
|
AIT_FREE_VAL(&args->Will.Msg); |
|
AIT_FREE_VAL(&args->Will.Topic); |
|
AIT_FREE_VAL(&args->User); |
|
AIT_FREE_VAL(&args->Pass); |
|
AIT_FREE_VAL(&args->Publish); |
|
AIT_FREE_VAL(&args->Value); |
|
AIT_FREE_VAL(&args->ConnID); |
|
io_freeVars(&args->Subscribes); |
|
} |
|
|
|
|
int |
int |
main(int argc, char **argv) |
main(int argc, char **argv) |
{ |
{ |
sqlite3 *acc = NULL, *pub = NULL; | char ch, batch = 1; |
FILE *logg = NULL; | ait_val_t *v, val; |
int ret = 0; | u_short port = atoi(MQTT_PORT); |
| int sock; |
|
|
if (LoadConfig("/etc/mqtt.conf", &cfg)) { | if (!(args = malloc(sizeof(struct tagArgs)))) { |
printf("Error:: Load config #%d - %s\n", cfg_GetErrno(), cfg_GetError()); | printf("Error:: in arguments #%d - %s\n", errno, strerror(errno)); |
return 1; |
return 1; |
|
} else |
|
memset(args, 0, sizeof(struct tagArgs)); |
|
if (!(args->Subscribes = io_allocVars(1))) { |
|
printf("Error:: in subscribes array #%d - %s\n", io_GetErrno(), io_GetError()); |
|
free(args); |
|
return 1; |
|
} else |
|
args->free = cleanArgs; |
|
|
|
if (!(args->msg = mqtt_msgAlloc(USHRT_MAX))) { |
|
printf("Error:: in mqtt buffer #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); |
|
args->free(args); |
|
free(args); |
|
return 1; |
} |
} |
if (!mqttLoadRTLM(&cfg, 0)) { | |
printf("Error:: Can't load RTL ACC module\n"); | AIT_SET_STR(&args->ConnID, ""); |
UnloadConfig(&cfg); | AIT_SET_STR(&args->User, ""); |
return 2; | AIT_SET_STR(&args->Pass, ""); |
| |
| args->ka = MQTT_KEEPALIVE; |
| strlcpy(szCfgName, DEFAULT_CONFIG, sizeof szCfgName); |
| while ((ch = getopt(argc, argv, "T:U:P:p:t:s:q:drc:W:M:fDvh")) != -1) |
| switch (ch) { |
| case 'T': |
| args->ka = (u_short) strtol(optarg, NULL, 0); |
| break; |
| case 'M': |
| AIT_FREE_VAL(&args->Will.Msg); |
| AIT_SET_STR(&args->Will.Msg, optarg); |
| break; |
| case 'W': |
| AIT_FREE_VAL(&args->Will.Topic); |
| AIT_SET_STR(&args->Will.Topic, optarg); |
| break; |
| case 'U': |
| AIT_FREE_VAL(&args->User); |
| AIT_SET_STR(&args->User, optarg); |
| break; |
| case 'P': |
| AIT_FREE_VAL(&args->Pass); |
| AIT_SET_STR(&args->Pass, optarg); |
| break; |
| case 'p': |
| port = (u_short) strtol(optarg, NULL, 0); |
| break; |
| case 't': |
| AIT_FREE_VAL(&args->Publish); |
| AIT_SET_STR(&args->Publish, optarg); |
| break; |
| case 's': |
| v = io_allocVar(); |
| if (!v) { |
| printf("Error:: not enough memory #%d - %s\n", errno, strerror(errno)); |
| args->free(args); |
| free(args); |
| return 1; |
| } else |
| AIT_SET_STR(v, optarg); |
| io_arrayElem(args->Subscribes, io_arraySize(args->Subscribes), v); |
| break; |
| case 'q': |
| args->QoS = (char) strtol(optarg, NULL, 0); |
| if (args->QoS > MQTT_QOS_EXACTLY) { |
| printf("Error:: invalid QoS level %d\n", args->QoS); |
| args->free(args); |
| free(args); |
| return 1; |
| } |
| break; |
| case 'd': |
| args->Dup++; |
| break; |
| case 'r': |
| args->Retain++; |
| break; |
| case 'C': |
| args->notClear++; |
| break; |
| case 'f': |
| args->isFile++; |
| break; |
| case 'c': |
| strlcpy(szCfgName, optarg, sizeof szCfgName); |
| break; |
| case 'D': |
| batch = 0; |
| break; |
| case 'v': |
| io_incDebug; |
| break; |
| case 'h': |
| default: |
| args->free(args); |
| free(args); |
| Usage(); |
| return 1; |
| } |
| argc -= optind; |
| argv += optind; |
| if (argc < 2) { |
| printf("Error:: host for connect not found or connection id not supplied!\n"); |
| args->free(args); |
| free(args); |
| Usage(); |
| return 1; |
| } else { |
| AIT_FREE_VAL(&args->ConnID); |
| AIT_SET_STR(&args->ConnID, argv[1]); |
} |
} |
if (!mqttLoadRTLM(&cfg, 1)) { | if (argc > 2) |
printf("Error:: Can't load RTL PUB module\n"); | AIT_SET_STR(&args->Value, argv[2]); |
mqttUnloadRTLM(0); | if (!io_gethostbyname(*argv, port, &args->addr)) { |
UnloadConfig(&cfg); | printf("Error:: host not connect #%d - %s\n", io_GetErrno(), io_GetError()); |
return 2; | args->free(args); |
| free(args); |
| Usage(); |
| return 1; |
} |
} |
if (!mqttLoadRTLM(&cfg, 2)) { | ioVERBOSE(1) printf("Connecting to %s:%d ...\n", io_n2addr(&args->addr, &val), io_n2port(&args->addr)); |
printf("Error:: Can't load RTL LOG module\n"); | |
mqttUnloadRTLM(1); | if (LoadConfig(szCfgName, &cfg)) { |
mqttUnloadRTLM(0); | printf("Error:: can't load #%d - %s\n", cfg_GetErrno(), cfg_GetError()); |
| args->free(args); |
| free(args); |
| return 1; |
| } |
| |
| if ((sock = InitClient()) == -1) { |
UnloadConfig(&cfg); |
UnloadConfig(&cfg); |
|
args->free(args); |
|
free(args); |
return 2; |
return 2; |
} |
} |
|
|
acc = call.OpenACC(&cfg); | printf("Connected ... "); |
if (!acc) | switch (try2Connect(sock)) { |
goto end; | case -1: |
pub = call.OpenPUB(&cfg); | printf("FAILED!\n"); |
if (!pub) | close(sock); |
goto end; | UnloadConfig(&cfg); |
logg = call.OpenLOG(&cfg); | args->free(args); |
if (!logg) | free(args); |
goto end; | return 3; |
| case MQTT_RETCODE_ACCEPTED: |
if (mqttMkDir(&cfg)) { | printf("OK\n"); |
printf("Error:: in statedir #%d - %s\n", errno, strerror(errno)); | break; |
goto end; | case MQTT_RETCODE_REFUSE_VER: |
| printf("Incorrect version\n"); |
| break; |
| case MQTT_RETCODE_REFUSE_ID: |
| printf("Incorrect connectID\n"); |
| break; |
| case MQTT_RETCODE_REFUSE_UNAVAIL: |
| printf("Service unavailable\n"); |
| break; |
| case MQTT_RETCODE_REFUSE_USERPASS: |
| printf("Refuse user/pass\n"); |
| break; |
| case MQTT_RETCODE_DENIED: |
| printf("DENIED.\n"); |
| break; |
} |
} |
|
|
call.LOG(logg, "success!\n"); | shutdown(sock, SHUT_RDWR); |
if ((ret = call.LoginACC(&cfg, acc, "misho", "test123")) == -1) { | close(sock); |
printf("Error:: Authentication problem\n"); | |
goto end; | |
} else | |
call.LOG(logg, "Login: %s\n", ret ? "ALLOW" : "DENIED"); | |
|
|
end: |
|
call.CloseLOG(logg); |
|
call.ClosePUB(pub); |
|
call.CloseACC(acc); |
|
mqttUnloadRTLM(2); |
|
mqttUnloadRTLM(1); |
|
mqttUnloadRTLM(0); |
|
UnloadConfig(&cfg); |
UnloadConfig(&cfg); |
|
args->free(args); |
|
free(args); |
return 0; |
return 0; |
} |
} |