|
|
| version 1.2.2.2, 2012/04/24 13:54:50 | version 1.2.2.7, 2012/05/08 13:04:02 |
|---|---|
| Line 17 Usage(void) | Line 17 Usage(void) |
| { | { |
| printf( " -= MQTT Subscriber Client =- Subscriber from ELWIX\n" | printf( " -= MQTT Subscriber Client =- Subscriber from ELWIX\n" |
| "=== %s@%s === Compiled: %s ===\n\n" | "=== %s@%s === Compiled: %s ===\n\n" |
| " Syntax: mqtt_subs [options] <connect_to_broker[:port]> <ConnectID> <topic> [exec_script <value>]\n\n" | " Syntax: mqtt_subs [options] <connect_to_broker[:port]> <ConnectID> [exec_script <value>]\n\n" |
| "\t-l <value2file>\t\tSave received values to file\n" | "\t-l <value2file>\t\tSave received values to file\n" |
| "\t-u\t\tUnsubscribe given topic(s)\n" | |
| "\t-s <topic[|QoS]>\tSubscribe for this topic, if wish add different |QoS to 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\n" | "\t-d\t\t\tSend duplicate message\n\n" |
| "\t-C\t\t\tNot clear before connect!\n" | "\t-C\t\t\tNot clear before connect!\n" |
| "\t-p <port>\t\tDifferent port for connect (default: 1883)\n" | "\t-p <port>\t\tDifferent port for connect (default: 1883)\n" |
| Line 39 static void | Line 39 static void |
| cleanArgs(struct tagArgs * __restrict args) | cleanArgs(struct tagArgs * __restrict args) |
| { | { |
| mqtt_msgFree(&args->msg, 42); | mqtt_msgFree(&args->msg, 42); |
| mqtt_subFree(&args->subscr); | |
| AIT_FREE_VAL(&args->Will.Msg); | AIT_FREE_VAL(&args->Will.Msg); |
| AIT_FREE_VAL(&args->Will.Topic); | AIT_FREE_VAL(&args->Will.Topic); |
| AIT_FREE_VAL(&args->User); | AIT_FREE_VAL(&args->User); |
| Line 46 cleanArgs(struct tagArgs * __restrict args) | Line 47 cleanArgs(struct tagArgs * __restrict args) |
| AIT_FREE_VAL(&args->Publish); | AIT_FREE_VAL(&args->Publish); |
| AIT_FREE_VAL(&args->Value); | AIT_FREE_VAL(&args->Value); |
| AIT_FREE_VAL(&args->ConnID); | AIT_FREE_VAL(&args->ConnID); |
| io_freeVars(&args->Subscribes); | |
| } | } |
| static int | static int |
| Subscribe(int sock, FILE *lf) | Subscribe(int sock, FILE *lf) |
| { | { |
| u_char *qoses, *qos; | |
| u_short mid; | |
| mqtt_subscr_t *sub; | |
| srandomdev(); | |
| mid = random() % USHRT_MAX; | |
| printf(" > Execute SUBSCRIBE request #%d ... ", mid); | |
| qoses = mqtt_cli_Subscribe(args->cli, args->subscr, mid, args->Dup, MQTT_QOS_ACK); | |
| if (!qoses) { | |
| printf("Error:: Subscribe #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); | |
| return -1; | |
| } else | |
| printf("OK\n"); | |
| for (sub = args->subscr, qos = qoses; sub->sub_topic.msg_base; sub++, qos++) | |
| printf(" + Topic %s with QoS %d subscribed %s\n", (char*) | |
| sub->sub_topic.msg_base, sub->sub_ret, *qos ? "done" : "failed"); | |
| free(qoses); | |
| return 0; | return 0; |
| } | } |
| static int | |
| Unsubscribe(int sock) | |
| { | |
| u_short mid; | |
| srandomdev(); | |
| mid = random() % USHRT_MAX; | |
| printf(" > Execute UNSUBSCRIBE request #%d ... ", mid); | |
| if (mqtt_cli_Unsubscribe(args->cli, args->subscr, mid, args->Dup, MQTT_QOS_ACK)) { | |
| printf("Error:: Unsubscribe #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); | |
| return -1; | |
| } else | |
| printf("OK\n"); | |
| return 0; | |
| } | |
| int | int |
| main(int argc, char **argv) | main(int argc, char **argv) |
| { | { |
| char ch, batch = 1; | char ch, un = 0, idx = 0, batch = 1; |
| ait_val_t *v, val; | ait_val_t val; |
| u_short port = atoi(MQTT_PORT); | u_short port = atoi(MQTT_PORT); |
| int sock, ret = 0; | mqtt_subscr_t *sub; |
| char szLogName[MAXPATHLEN] = { 0 }; | int ret = 0; |
| char *str, szStr[STRSIZ], szLogName[MAXPATHLEN] = { 0 }; | |
| FILE *lf; | FILE *lf; |
| if (!(args = malloc(sizeof(struct tagArgs)))) { | if (!(args = malloc(sizeof(struct tagArgs)))) { |
| Line 71 main(int argc, char **argv) | Line 110 main(int argc, char **argv) |
| return 1; | return 1; |
| } else | } else |
| memset(args, 0, sizeof(struct tagArgs)); | memset(args, 0, sizeof(struct tagArgs)); |
| if (!(args->Subscribes = io_allocVars(1))) { | if (!(args->subscr = mqtt_subAlloc(idx))) { |
| printf("Error:: in subscribes array #%d - %s\n", io_GetErrno(), io_GetError()); | printf("Error:: in subscribes array #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); |
| free(args); | free(args); |
| return 1; | return 1; |
| } else | } else |
| Line 90 main(int argc, char **argv) | Line 129 main(int argc, char **argv) |
| AIT_SET_STR(&args->Pass, ""); | AIT_SET_STR(&args->Pass, ""); |
| args->ka = MQTT_KEEPALIVE; | args->ka = MQTT_KEEPALIVE; |
| while ((ch = getopt(argc, argv, "T:U:P:p:s:q:dl:W:M:CDvh")) != -1) | while ((ch = getopt(argc, argv, "T:U:P:p:s:q:dl:W:M:CDvuh")) != -1) |
| switch (ch) { | switch (ch) { |
| case 'T': | case 'T': |
| args->ka = (u_short) strtol(optarg, NULL, 0); | args->ka = (u_short) strtol(optarg, NULL, 0); |
| Line 115 main(int argc, char **argv) | Line 154 main(int argc, char **argv) |
| port = (u_short) strtol(optarg, NULL, 0); | port = (u_short) strtol(optarg, NULL, 0); |
| break; | break; |
| case 's': | case 's': |
| v = io_allocVar(); | sub = mqtt_subRealloc(&args->subscr, idx + 1); |
| if (!v) { | if (!sub) { |
| printf("Error:: not enough memory #%d - %s\n", errno, strerror(errno)); | printf("Error:: #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); |
| args->free(args); | args->free(args); |
| free(args); | free(args); |
| return 1; | return 1; |
| } else | } else |
| AIT_SET_STR(v, optarg); | sub += idx++; |
| io_arrayElem(args->Subscribes, io_arraySize(args->Subscribes), v); | |
| strlcpy(szStr, optarg, sizeof szStr); | |
| if ((str = strchr(szStr, '|'))) { | |
| *str++ = 0; | |
| *str -= 0x30; | |
| if (*str < 0 || *str > MQTT_QOS_RESERVED) | |
| sub->sub_ret = (u_char) args->QoS; | |
| else | |
| sub->sub_ret = (u_char) *str; | |
| } else | |
| sub->sub_ret = (u_char) args->QoS; | |
| sub->sub_topic.msg_base = strdup(szStr); | |
| sub->sub_topic.msg_len = strlen(szStr); | |
| break; | break; |
| case 'q': | case 'q': |
| args->QoS = (char) strtol(optarg, NULL, 0); | args->QoS = (char) strtol(optarg, NULL, 0); |
| Line 149 main(int argc, char **argv) | Line 200 main(int argc, char **argv) |
| case 'v': | case 'v': |
| io_incDebug; | io_incDebug; |
| break; | break; |
| case 'u': | |
| un = 1; | |
| break; | |
| case 'h': | case 'h': |
| default: | default: |
| args->free(args); | args->free(args); |
| Line 158 main(int argc, char **argv) | Line 212 main(int argc, char **argv) |
| } | } |
| argc -= optind; | argc -= optind; |
| argv += optind; | argv += optind; |
| if (argc < 3) { | if (argc < 2) { |
| printf("Error:: host for connect not found, connection id or topic not supplied!\n\n"); | printf("Error:: host for connect not found, connection id or topic not supplied!\n\n"); |
| args->free(args); | args->free(args); |
| free(args); | free(args); |
| Line 167 main(int argc, char **argv) | Line 221 main(int argc, char **argv) |
| } else { | } else { |
| AIT_FREE_VAL(&args->ConnID); | AIT_FREE_VAL(&args->ConnID); |
| AIT_SET_STR(&args->ConnID, argv[1]); | AIT_SET_STR(&args->ConnID, argv[1]); |
| AIT_FREE_VAL(&args->Publish); | |
| AIT_SET_STR(&args->Publish, argv[2]); | |
| } | } |
| if (argc > 3) { | if (argc > 2) { |
| AIT_FREE_VAL(&args->Value); | AIT_FREE_VAL(&args->Value); |
| AIT_SET_STR(&args->Value, argv[3]); | AIT_SET_STR(&args->Value, argv[2]); |
| } | } |
| if (!io_gethostbyname(*argv, port, &args->addr)) { | if (!io_gethostbyname(*argv, port, &args->addr)) { |
| printf("Error:: host not valid #%d - %s\n", io_GetErrno(), io_GetError()); | printf("Error:: host not valid #%d - %s\n", io_GetErrno(), io_GetError()); |
| Line 184 main(int argc, char **argv) | Line 236 main(int argc, char **argv) |
| printf("Connecting to %s:%d ... ", io_n2addr(&args->addr, &val), io_n2port(&args->addr)); | printf("Connecting to %s:%d ... ", io_n2addr(&args->addr, &val), io_n2port(&args->addr)); |
| AIT_FREE_VAL(&val); | AIT_FREE_VAL(&val); |
| if ((sock = InitClient()) == -1) { | if (!(args->cli = mqtt_cli_Open(&args->addr.sa, args->ka))) { |
| args->free(args); | args->free(args); |
| free(args); | free(args); |
| return 2; | return 2; |
| } | } |
| switch ((ret = ConnectClient(sock))) { | switch ((ret = ConnectClient(args->cli->sock))) { |
| case -1: | case -1: |
| printf(">> FAILED!\n"); | printf(">> FAILED!\n"); |
| break; | break; |
| Line 220 main(int argc, char **argv) | Line 272 main(int argc, char **argv) |
| else | else |
| lf = stdout; | lf = stdout; |
| if (lf) { | if (lf) { |
| ret = Subscribe(sock, lf); | ret = Subscribe(args->cli->sock, lf); |
| if (un) | |
| Unsubscribe(args->cli->sock); | |
| fclose(lf); | fclose(lf); |
| } else | } else |
| printf("Error:: in subscribe file #%d - %s\n", errno, strerror(errno)); | printf("Error:: in subscribe file #%d - %s\n", errno, strerror(errno)); |
| CloseClient(sock); | } else |
| } else { | |
| close(sock); | |
| ret = 3; | ret = 3; |
| } | |
| mqtt_cli_Close(&args->cli); | |
| args->free(args); | args->free(args); |
| free(args); | free(args); |