|
|
| version 1.2.2.2, 2012/04/24 13:54:50 | version 1.2.2.5, 2012/05/01 22:30:31 |
|---|---|
| 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-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-q <QoS>\t\tQoS level (0-at most 1, 1-at least 1, 2-exactly 1)\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) |
| { | { |
| int siz; | |
| u_char *qoses; | |
| u_short mid[2]; | |
| srandomdev(); | |
| mid[0] = random() % USHRT_MAX; | |
| siz = mqtt_msgSUBSCRIBE(args->msg, args->subscr, mid[0], args->Dup, args->QoS); | |
| if (siz == -1) { | |
| printf("Error:: in msgSUBSCRIBE #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); | |
| return -1; | |
| } | |
| siz = SendTo(sock, siz); | |
| if (siz == -1) | |
| return -1; | |
| siz = RecvFrom(sock); | |
| if (siz == -1) | |
| return -1; | |
| siz = mqtt_readSUBACK(args->msg, &mid[1], &qoses); | |
| if (siz == -1) { | |
| printf("Error:: in readSUBACK #%d - %s\n", mqtt_GetErrno(), mqtt_GetError()); | |
| return -1; | |
| } | |
| if (mid[1] != mid[0]) { | |
| printf("Error:: received different connection ID %d != %d\n", mid[1], mid[0]); | |
| return -1; | |
| } | |
| free(qoses); | |
| return 0; | return 0; |
| } | } |
| Line 59 Subscribe(int sock, FILE *lf) | Line 89 Subscribe(int sock, FILE *lf) |
| int | int |
| main(int argc, char **argv) | main(int argc, char **argv) |
| { | { |
| char ch, batch = 1; | char ch, 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 102 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 115 main(int argc, char **argv) | Line 146 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 158 main(int argc, char **argv) | Line 201 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 210 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 225 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->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 261 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); |
| 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); |