--- mqtt/src/Attic/mqtt.c 2011/11/24 16:00:31 1.1.1.1.2.15 +++ mqtt/src/Attic/mqtt.c 2011/12/13 09:51:35 1.1.1.1.2.24 @@ -1,87 +1,202 @@ #include "global.h" #include "rtlm.h" +#include "mqtt.h" +#include "client.h" sl_config cfg; +sqlite3 *acc, *pub; +FILE *logg; +extern char compiled[], compiledby[], compilehost[]; +static char szCfgName[MAXPATHLEN]; +int Verbose, 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] [value_for_publish]\n\n" + "\t-t \t\tPublish topic\n" + "\t-s \tSubscribe for this topic, if wish add different |QoS to topic\n" + "\t-q \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\tService config\n" + "\t-f\t\t\t'value_for_publish' is file instead text\n" + "\t-p \t\tDifferent port for connect (default: 1883)\n" + "\t-U \t\tUsername\n" + "\t-P \t\tPassword\n" + "\t-W \t\tWill Topic\n" + "\t-M \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); + io_freeVars(&args->Subscribes); +} + + int main(int argc, char **argv) { - sqlite3 *acc = NULL, *pub = NULL; - FILE *logg = NULL; - int ret = 0; + char ch, batch = 1, szStr[STRSIZ] = { 0 }; + ait_val_t *v, val; + u_short port = atoi(MQTT_PORT); + int sock; - if (LoadConfig("/etc/mqtt.conf", &cfg)) { - printf("Error:: Load config #%d - %s\n", cfg_GetErrno(), cfg_GetError()); + if (!(args = malloc(sizeof(struct tagArgs)))) { + printf("Error:: in arguments #%d - %s\n", errno, strerror(errno)); 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"); - UnloadConfig(&cfg); - return 2; + + strlcpy(szCfgName, DEFAULT_CONFIG, sizeof szCfgName); + while ((ch = getopt(argc, argv, "U:P:p:t:s:q:drc:W:M:fDvh")) != -1) + switch (ch) { + 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 = malloc(sizeof(ait_val_t)); + 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), (void**) &v); + break; + case 'q': + args->QoS = (char) strtol(optarg, NULL, 0); + if (args->QoS < MQTT_QOS_ONCE || 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 'f': + args->isFile++; + break; + case 'c': + strlcpy(szCfgName, optarg, sizeof szCfgName); + break; + case 'D': + batch = 0; + break; + case 'v': + Verbose++; + break; + case 'h': + default: + args->free(args); + free(args); + Usage(); + return 1; + } + argc -= optind; + argv += optind; + if (!argc) { + printf("Error:: host for connect not found!\n"); + args->free(args); + free(args); + Usage(); + return 1; } - if (!mqttLoadRTLM(&cfg, 1)) { - printf("Error:: Can't load RTL PUB module\n"); - mqttUnloadRTLM(0); - UnloadConfig(&cfg); - return 2; + if (argc > 1) + AIT_SET_STR(&args->Value, argv[1]); + if (!io_gethostbyname(*argv, port, &args->addr)) { + printf("Error:: host not connect #%d - %s\n", io_GetErrno(), io_GetError()); + args->free(args); + free(args); + Usage(); + return 1; } - if (!mqttLoadRTLM(&cfg, 2)) { - printf("Error:: Can't load RTL LOG module\n"); - mqttUnloadRTLM(1); - mqttUnloadRTLM(0); + VERB(1) printf("Connecting to %s:%d ...\n", io_n2addr(&args->addr, &val), io_n2port(&args->addr)); + + if (LoadConfig(szCfgName, &cfg)) { + 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); + args->free(args); + free(args); return 2; } - acc = call.OpenACC(&cfg); - if (!acc) - goto end; - pub = call.OpenPUB(&cfg); - if (!pub) - goto end; - logg = call.OpenLOG(&cfg); - if (!logg) - goto end; - - if (mqttMkDir(&cfg)) { - printf("Error:: in statedir #%d - %s\n", errno, strerror(errno)); - goto end; + if (SendConnect(sock) == -1) { + close(sock); + UnloadConfig(&cfg); + args->free(args); + free(args); + return 3; } - call.LOG(logg, "success!\n"); - if ((ret = call.LoginACC(&cfg, acc, "misho", "test123")) == -1) { - printf("Error:: Authentication problem\n"); - goto end; - } else - call.LOG(logg, "Login: %s\n", ret ? "ALLOW" : "DENIED"); + shutdown(sock, SHUT_RDWR); + close(sock); - if ((ret = call.InitSessPUB(&cfg, pub, "misho", "127.0.0.1", 12345)) == -1) { - printf("Error:: Session init problem\n"); - goto end; - } else - call.LOG(logg, "InitSess: %d\n", ret); - - if ((ret = call.ChkSessPUB(&cfg, pub, "misho", "127.0._.%")) == -1) { - printf("Error:: Session check problem\n"); - goto end; - } else - call.LOG(logg, "ChkSess: %d\n", ret); - - if ((ret = call.FiniSessPUB(&cfg, pub, "misho", "127.0._.1%")) == -1) { - printf("Error:: Session fini problem\n"); - goto end; - } else - call.LOG(logg, "FiniSess: %d\n", ret); - -end: - call.CloseLOG(logg); - call.ClosePUB(pub); - call.CloseACC(acc); - mqttUnloadRTLM(2); - mqttUnloadRTLM(1); - mqttUnloadRTLM(0); UnloadConfig(&cfg); + args->free(args); + free(args); return 0; }