|
|
| version 1.1.1.1.2.7, 2011/11/23 01:07:18 | version 1.1.1.1.2.28, 2011/12/14 12:45:30 |
|---|---|
| Line 1 | Line 1 |
| #include "global.h" | #include "global.h" |
| #include "rtlm.h" | #include "rtlm.h" |
| #include "mqtt.h" | |
| #include "client.h" | |
| sl_config cfg; | 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] <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; |
| ait_val_t *v, val; | |
| 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 module\n"); | AIT_SET_STR(&args->ConnID, ""); |
| return 2; | AIT_SET_STR(&args->User, ""); |
| 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': | |
| Verbose++; | |
| 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 (argc > 2) | |
| AIT_SET_STR(&args->Value, argv[2]); | |
| 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; | |
| } | |
| VERB(1) printf("Connecting to %s:%d ...\n", io_n2addr(&args->addr, &val), io_n2port(&args->addr)); | |
| acc = mqttOpenDB(&cfg, 0); | if (LoadConfig(szCfgName, &cfg)) { |
| if (!acc) | printf("Error:: can't load #%d - %s\n", cfg_GetErrno(), cfg_GetError()); |
| goto end; | args->free(args); |
| pub = mqttOpenDB(&cfg, 1); | free(args); |
| if (!pub) | return 1; |
| goto end; | } |
| if (mqttMkDir(&cfg)) { | if ((sock = InitClient()) == -1) { |
| printf("Error:: in statedir #%d - %s\n", errno, strerror(errno)); | UnloadConfig(&cfg); |
| goto end; | args->free(args); |
| free(args); | |
| return 2; | |
| } | } |
| printf("success!\n"); | printf("Connected ... "); |
| switch (try2Connect(sock)) { | |
| case -1: | |
| printf("FAILED!\n"); | |
| close(sock); | |
| UnloadConfig(&cfg); | |
| args->free(args); | |
| free(args); | |
| return 3; | |
| case MQTT_RETCODE_ACCEPTED: | |
| printf("OK\n"); | |
| break; | |
| 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; | |
| } | |
| end: | shutdown(sock, SHUT_RDWR); |
| mqttCloseDB(pub); | close(sock); |
| mqttCloseDB(acc); | |
| mqttUnloadRTLM(0); | |
| UnloadConfig(&cfg); | UnloadConfig(&cfg); |
| args->free(args); | |
| free(args); | |
| return 0; | return 0; |
| } | } |